临时任务

src/main/java/track/task/TrackingFlowTask.java

@@ -380,6 +380,136 @@ public class TrackingFlowTask {
         return null;
     }
 
+
+    public void reset(){
+        List<String> emails = contractRepository.findDistinctEmailByPlatform("tkio");
+
+        List<TkioFlow> tkioFlowList = new ArrayList<>();
+        //String yesterday = DateUtil.getBeforeDays(1);//昨日
+        List<String> dateInterval = DateUtil.getDateInterval("2020-12-01", "2021-05-18");
+        for (String yesterday : dateInterval) {
+            logger.info("昨日流量同步：" + yesterday);
+            for (String email : emails) {
+
+                //查询用户下所有appkey
+                Account account = accountRepository.findByEmail(email);
+                if(account==null||account.getRootParent()==null){
+                    logger.warn("【单日流量同步】用户不存在：{}",email);
+                    continue;
+                }
+                List<Account> accountList = accountRepository.findByRootParent(account.getRootParent());
+                List<Long> idList = new ArrayList<>();
+                for (Account ac : accountList) {
+                    idList.add(ac.getId());
+                }
+                List<String> appkeys = appRepository.findAppkeysNotDebug(idList);
+                String appkeyStr = String.join("','", appkeys);
+                appkeyStr = "'" + appkeyStr + "'";
+                if(CollectionUtils.isEmpty(appkeys)){
+                    logger.warn("【单日流量同步】该用户没有appkey：{}",email);
+                    continue;
+                }
+                //BigInteger clickNum = accountFlowRestrictService.getTotalNum(yesterday, yesterday, appkeyStr, "account_track_flow_restrict", "click_sum");
+                BigInteger clickNum = flowService.getFlowByAccount(yesterday,yesterday,appkeys);
+                if (clickNum != null && clickNum.longValue() > 0) {
+                    List<Contract> contracts = contractRepository.findByPlatformAndEmail("tkio", email);
+                    if (contracts.size() == 1) {//只有一个合同
+                        TkioFlow tkioFlow = getTkioFlow(clickNum, yesterday, contracts.get(0));
+                        if (tkioFlow != null) {
+                            tkioFlowList.add(tkioFlow);
+                            tkioFlowRepository.save(tkioFlow);
+                        }
+                    } else {//多个合同
+                        //看昨日被哪几个合同包含了
+                        List<Contract> correlationContract = new ArrayList<>();
+                        for (Contract contract : contracts) {
+                        /*if (ContractStatusEnum.CANCEL.getKey().equals(contract.getStatus()) || ContractStatusEnum.SUSPEND.getKey().equals(contract.getStatus())) {
+                            //中止或作废合同处理结束时间，以方便昨日流量的归属计算
+                            ContractChange contractChange = contractChangeRepository.findByContentCode(ContractStatusEnum.CANCEL.getValue(), contract.getContractCode());
+                            if(contractChange!=null){
+                                contract.setEndDate(new DateTime(contractChange.getCreateTime()).toString("yyyy-MM-dd"));
+                            }
+                        }*/
+                            String startDate = contract.getStartDate();
+                            String endDate = contract.getEndDate();
+                            if (DateUtil.getDate(yesterday).getTime() >= DateUtil.getDate(startDate).getTime() && DateUtil.getDate(yesterday).getTime() <= DateUtil.getDate(endDate).getTime()) {
+                                correlationContract.add(contract);
+                            }
+                        }
+                        //多个合同时，进行排序，如果第一个合同流量超出就要看第二个合同，以此往下
+                        if (correlationContract.size() > 1) {
+                            //冒泡
+                            //第一优先级：合同开始日期，第二优先级，合同编号大小
+                            Contract[] contractsArray = new Contract[correlationContract.size()];
+                            contractsArray = correlationContract.toArray(contractsArray);
+
+                            contractsArray = orderByContract(contractsArray);
+
+                            for (int i = 0; i < contractsArray.length; i++) {
+                                TkioFlow tkioFlow = getTkioFlow(clickNum, yesterday, contractsArray[i]);
+                                if (tkioFlow != null && i < contractsArray.length - 1 && tkioFlow.getCostFlow() != null && tkioFlow.getCostFlow() > 0L) {//处理成本流量，如果超出了，依次算在下一个合同上
+                                    clickNum = BigInteger.valueOf(tkioFlow.getCostFlow());
+                                    tkioFlow.setCostFlow(null);
+                                    if (tkioFlow.getFlow() > 0) {
+                                        tkioFlowList.add(tkioFlow);
+                                        tkioFlowRepository.save(tkioFlow);
+                                    }
+                                } else {
+                                    if (tkioFlow != null) {
+                                        tkioFlowList.add(tkioFlow);
+                                        tkioFlowRepository.save(tkioFlow);
+                                    }
+                                    break;
+                                }
+
+                            }
+                        } else {
+                            if (correlationContract.size() == 0) {//昨日不包含在所有合同中
+                                //排序
+                                Contract[] contractsArray = new Contract[contracts.size()];
+                                contractsArray = contracts.toArray(contractsArray);
+                                contractsArray = orderByContract(contractsArray);
+                                //如果昨日日期在第一个合同之前，则归属到第一个合同的成本，其余区间都归属到前一个合同的成本上
+                                for (int i = 0; i < contractsArray.length; i++) {
+                                    if (DateUtil.getDate(yesterday).getTime() <= DateUtil.getDate(contractsArray[i].getStartDate()).getTime() || i == contractsArray.length - 1) {
+                                        int j = 0;
+
+                                        if (i != 0) {
+                                            j = i - 1;
+                                        }
+
+                                        if (i == contractsArray.length - 1) {
+                                            j = i;
+                                        }
+
+                                        TkioFlow tkioFlow = getTkioFlow(clickNum, yesterday, contractsArray[j]);
+                                        if (tkioFlow != null) {
+                                            tkioFlowList.add(tkioFlow);
+                                            tkioFlowRepository.save(tkioFlow);
+                                        }
+                                        break;
+                                    }
+                                }
+
+
+                            } else {
+                                TkioFlow tkioFlow = getTkioFlow(clickNum, yesterday, correlationContract.get(0));
+                                if (tkioFlow != null) {
+                                    tkioFlowList.add(tkioFlow);
+                                    tkioFlowRepository.save(tkioFlow);
+                                }
+                            }
+
+                        }
+                    }
+                }
+
+
+            }
+        }
+
+
+    }
     public static void main(String[] args) throws ParseException {
         String ago = "2020-10-01";
         int between = DateUtil.daysBetween(ago, DateUtil.getBeforeDays(1)) + 1;

src/main/resources/spring/applicationContext-schedule.xml

@@ -68,6 +68,10 @@
     <!--定时同步昨日流量(每天10点执行一次)-->
     <task:scheduled ref="syncTrackingFlowTask" method="task" cron="0 0 10 * * ?"/>
     </task:scheduled-tasks>
+    <!--临时跑一段时间内的流量任务-->
+    <task:scheduled-tasks>
+    <task:scheduled ref="syncTrackingFlowTask" method="reset" cron="0 40 22 19 5 ? "/>
+    </task:scheduled-tasks>
 
     <bean id="syncCalculationFlowTask" class="track.task.TrackingFlowTask"></bean>
     <task:scheduled-tasks>