|
@@ -21,6 +21,8 @@ import cn.com.ctop.oa.modules.service.IWechatCheckinDataService;
|
|
|
import cn.com.ctop.oa.modules.service.IWechatNoListService;
|
|
|
import cn.com.ctop.oa.modules.service.IWechatUserListService;
|
|
|
import cn.com.ctop.toutiao.modules.material.service.IByteDanceAdvertiserDataService;
|
|
|
+import cn.com.ctop.toutiao.modules.report.entity.BytedanceAccountReportTaskRecord;
|
|
|
+import cn.com.ctop.toutiao.modules.report.mapper.BytedanceReportMaterialDailyMapper;
|
|
|
import cn.com.ctop.toutiao.modules.report.service.*;
|
|
|
import com.alibaba.fastjson.JSONArray;
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
@@ -28,11 +30,13 @@ import org.apache.catalina.User;
|
|
|
import org.jeecg.common.util.DateUtils;
|
|
|
import org.junit.Test;
|
|
|
import org.junit.runner.RunWith;
|
|
|
+import org.openqa.selenium.By;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
|
import org.springframework.test.context.ActiveProfiles;
|
|
|
import org.springframework.test.context.junit4.SpringRunner;
|
|
|
|
|
|
+import javax.annotation.Resource;
|
|
|
import java.text.ParseException;
|
|
|
import java.text.SimpleDateFormat;
|
|
|
import java.util.Date;
|
|
@@ -388,16 +392,55 @@ public class SampleTest {
|
|
|
@Autowired
|
|
|
private IRuleKuaiShouPlanService ruleKuaiShouPlanService;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private IBytedanceAccountReportTaskRecordService reportTaskRecordService;
|
|
|
@Test
|
|
|
public void execute() {
|
|
|
- List<UserAllocation> allocations = allocationService.getByMediaId("3");
|
|
|
+ List<UserAllocation> allocations = allocationService.listByMediaId("3",0);
|
|
|
for (UserAllocation allocation:allocations) {
|
|
|
- CtopOauthToken token = tokenService.getTokenByAccountId(allocation.getAccountId());
|
|
|
-// load12Data(token);
|
|
|
- load12AccountData(token);
|
|
|
+ BytedanceAccountReportTaskRecord reportTaskRecord1 = new BytedanceAccountReportTaskRecord(allocation.getAccountId(),"2020-10-01","2020-10-31");
|
|
|
+ reportTaskRecordService.save(reportTaskRecord1);
|
|
|
+ BytedanceAccountReportTaskRecord reportTaskRecord2 = new BytedanceAccountReportTaskRecord(allocation.getAccountId(),"2020-11-01","2020-11-30");
|
|
|
+ reportTaskRecordService.save(reportTaskRecord2);
|
|
|
+ BytedanceAccountReportTaskRecord reportTaskRecord3 = new BytedanceAccountReportTaskRecord(allocation.getAccountId(),"2020-12-01","2020-12-31");
|
|
|
+ reportTaskRecordService.save(reportTaskRecord3);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void loadBDAccoutData() {
|
|
|
+// List<BytedanceAccountReportTaskRecord> records = reportTaskRecordService.listByParams(1,"ACCOUNT_DAILY");
|
|
|
+ List<BytedanceAccountReportTaskRecord> records = reportTaskRecordService.listByParams(100,"ACCOUNT_DAILY");
|
|
|
+ executorService = Executors.newFixedThreadPool(8);
|
|
|
+ countDownLatch = new CountDownLatch(records.size());
|
|
|
+ records.forEach(record -> executorService.submit(() -> {
|
|
|
+ try {
|
|
|
+// reportService.getAdvertiserReport(record, CtopAdConstant.BYTEDANCE_REPORT_TYPE_DAILY);
|
|
|
+// log.info("账户数据获取完成:account:{},时间:{}-{}",record.getAccountId(),record.getStartDate(),record.getEndDate());
|
|
|
+ bytedanceReportService.bytedanceMaterialReportV2(record);
|
|
|
+ log.info("素材数据获取完成:account:{},时间:{}-{}",record.getAccountId(),record.getStartDate(),record.getEndDate());
|
|
|
+ } catch (Exception e) {
|
|
|
+
|
|
|
+ } finally {
|
|
|
+ countDownLatch.countDown();
|
|
|
+ }
|
|
|
+ }));
|
|
|
+ try {
|
|
|
+ countDownLatch.await();
|
|
|
+ } catch (InterruptedException e) {
|
|
|
+ e.printStackTrace();
|
|
|
}
|
|
|
+ System.out.println("数据获取完成");
|
|
|
+ bytedanceReportMaterialDailyMapper.updateImageReportCode();
|
|
|
+ bytedanceReportMaterialDailyMapper.updateImageReportProjectId();
|
|
|
+ bytedanceReportMaterialDailyMapper.updateImageReportProjectName();
|
|
|
+ bytedanceReportMaterialDailyMapper.updateImageReportPlaneId();
|
|
|
+ bytedanceReportMaterialDailyMapper.updateImageReportPlaneName();
|
|
|
}
|
|
|
|
|
|
+ @Resource
|
|
|
+ private BytedanceReportMaterialDailyMapper bytedanceReportMaterialDailyMapper;
|
|
|
+
|
|
|
private void load12AccountData(CtopOauthToken token) {
|
|
|
reportService.getAdvertiserReport(token, DateUtils.parseDate("2020-10-01", "yyyy-MM-dd"), DateUtils.parseDate("2020-10-31", "yyyy-MM-dd"), CtopAdConstant.BYTEDANCE_REPORT_TYPE_DAILY);
|
|
|
reportService.getAdvertiserReport(token, DateUtils.parseDate("2020-11-01", "yyyy-MM-dd"), DateUtils.parseDate("2020-11-30", "yyyy-MM-dd"), CtopAdConstant.BYTEDANCE_REPORT_TYPE_DAILY);
|