|
@@ -1,8 +1,10 @@
|
|
package org.jeecg;
|
|
package org.jeecg;
|
|
|
|
|
|
|
|
+import cn.com.ctop.kuaishou.modules.batch.service.IKuaiShouReportTaskService;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.junit.Test;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.junit.runner.RunWith;
|
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
import org.springframework.test.context.junit4.SpringRunner;
|
|
import org.springframework.test.context.junit4.SpringRunner;
|
|
@@ -27,7 +29,8 @@ public class SampleTest {
|
|
try {
|
|
try {
|
|
|
|
|
|
|
|
|
|
- reportTaskService.getTaskList();
|
|
|
|
|
|
+ // reportTaskService.createTask(1022694L,"c2defedd56511f4a768ce8e34e0232ff");
|
|
|
|
+ reportTaskService.getTaskList();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -95,136 +98,9 @@ public class SampleTest {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
- @Autowired
|
|
|
|
- private ICtopOauthTokenService tokenService;
|
|
|
|
|
|
|
|
- @Test
|
|
|
|
- public void testBakToken() {
|
|
|
|
- Date getDate = DateUtils.addDay(new Date(), -1);
|
|
|
|
- QueryWrapper<UserAllocation> queryWrapper = new QueryWrapper<>();
|
|
|
|
- queryWrapper.eq("project_id", 215);
|
|
|
|
- List<UserAllocation> allocations = userAllocationService.list(queryWrapper);
|
|
|
|
- if (null != allocations && allocations.size() > 0) {
|
|
|
|
- for (UserAllocation allocation : allocations) {
|
|
|
|
- CtopOauthToken token = tokenService.getTokenByAccountId(allocation.getAccountId());
|
|
|
|
- reportService.getAdvertiserPlanReport(token, getDate, getDate, CtopAdConstant.BYTEDANCE_REPORT_TYPE_DAILY);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- formatZybData(215, getDate);
|
|
|
|
- }
|
|
|
|
|
|
|
|
- @Autowired
|
|
|
|
- private IUserAllocationService userAllocationService;
|
|
|
|
-
|
|
|
|
- private void formatZybData(Integer projectId, Date getDate) {
|
|
|
|
- //清洗关于作业帮数据
|
|
|
|
- //1:查询作业帮相关的用户数据
|
|
|
|
- QueryWrapper<UserAllocation> wrapper = new QueryWrapper<>();
|
|
|
|
- wrapper.eq("project_id", projectId);
|
|
|
|
- List<UserAllocation> allocations = userAllocationService.list(wrapper);
|
|
|
|
- if (null != allocations && allocations.size() > 0) {
|
|
|
|
- allocations.forEach(allocation -> {
|
|
|
|
- //根据accountId和时间,查询相关的日报表信息
|
|
|
|
- QueryWrapper<BytedancePlanDailyReport> queryWrapper = new QueryWrapper<>();
|
|
|
|
- SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");
|
|
|
|
- String dateString = dateFormat.format(getDate) + " 00:00:00";
|
|
|
|
- queryWrapper.eq("advertiser_id", allocation.getAccountId())
|
|
|
|
- .eq("stat_datetime", dateString)
|
|
|
|
- ;
|
|
|
|
- List<BytedancePlanDailyReport> bytedancePlanDailyReports = planDailyReportService.list(queryWrapper);
|
|
|
|
- if (null != bytedancePlanDailyReports && bytedancePlanDailyReports.size() > 0) {
|
|
|
|
- bytedancePlanDailyReports.forEach(report -> {
|
|
|
|
- String adName = report.getAdName();
|
|
|
|
- String[] tags = adName.split("-");
|
|
|
|
- report.setAdsense(tags[1]);
|
|
|
|
- report.setPlanCode(tags[2]);
|
|
|
|
- String materialName = tags[3];
|
|
|
|
- report.setMaterialName(materialName);
|
|
|
|
- if (materialName.contains("其他")) {
|
|
|
|
- report.setMaterialType("其他");
|
|
|
|
- } else if (materialName.contains("图片轮播")) {
|
|
|
|
- report.setMaterialType("图片轮播");
|
|
|
|
- } else if (materialName.contains("大字报")) {
|
|
|
|
- report.setMaterialType("大字报");
|
|
|
|
- } else if (materialName.contains("文字动画")) {
|
|
|
|
- report.setMaterialType("文字动画");
|
|
|
|
- } else if (materialName.contains("采访形式")) {
|
|
|
|
- report.setMaterialType("采访形式");
|
|
|
|
- } else if (materialName.contains("剧情")) {
|
|
|
|
- report.setMaterialType("剧情");
|
|
|
|
- } else if (materialName.contains("口播")) {
|
|
|
|
- report.setMaterialType("口播");
|
|
|
|
- } else {
|
|
|
|
- report.setMaterialType("其他");
|
|
|
|
- }
|
|
|
|
- planDailyReportService.updateById(report);
|
|
|
|
- });
|
|
|
|
- }
|
|
|
|
- });
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private IBytedancePlanDailyReportService planDailyReportService;
|
|
|
|
- @Autowired
|
|
|
|
- private IReportService reportService;
|
|
|
|
-
|
|
|
|
- /**
|
|
|
|
- * CSV文件生成方法
|
|
|
|
- *
|
|
|
|
- * @param head 文件头
|
|
|
|
- * @param dataList 数据列表
|
|
|
|
- * @param outPutPath 文件输出路径
|
|
|
|
- * @param filename 文件名
|
|
|
|
- * @return
|
|
|
|
- */
|
|
|
|
- public static File createCSVFile(String head, List<CtopOauthToken> dataList, String outPutPath, String filename) {
|
|
|
|
-
|
|
|
|
- File csvFile = null;
|
|
|
|
- BufferedWriter csvWtriter = null;
|
|
|
|
- try {
|
|
|
|
- csvFile = new File(outPutPath + File.separator + filename + ".csv");
|
|
|
|
- File parent = csvFile.getParentFile();
|
|
|
|
- if (parent != null && !parent.exists()) {
|
|
|
|
- parent.mkdirs();
|
|
|
|
- }
|
|
|
|
- csvFile.createNewFile();
|
|
|
|
|
|
|
|
- // GB2312使正确读取分隔符","
|
|
|
|
- csvWtriter = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(
|
|
|
|
- csvFile), "GB2312"), 1024);
|
|
|
|
- // 写入文件头部
|
|
|
|
- writeRow(head, csvWtriter);
|
|
|
|
-
|
|
|
|
- // 写入文件内容
|
|
|
|
- for (CtopOauthToken row : dataList) {
|
|
|
|
- writeRow(row.toString(), csvWtriter);
|
|
|
|
- }
|
|
|
|
- csvWtriter.flush();
|
|
|
|
- } catch (Exception e) {
|
|
|
|
- e.printStackTrace();
|
|
|
|
- } finally {
|
|
|
|
- try {
|
|
|
|
- csvWtriter.close();
|
|
|
|
- } catch (IOException e) {
|
|
|
|
- e.printStackTrace();
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- return csvFile;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- public static byte[] readInputStream(InputStream inputStream) throws IOException {
|
|
|
|
- byte[] buffer = new byte[1024];
|
|
|
|
- int len = 0;
|
|
|
|
- ByteArrayOutputStream bos = new ByteArrayOutputStream();
|
|
|
|
- while ((len = inputStream.read(buffer)) != -1) {
|
|
|
|
- bos.write(buffer, 0, len);
|
|
|
|
- }
|
|
|
|
- bos.close();
|
|
|
|
- return bos.toByteArray();
|
|
|
|
- }
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
}
|