工具库-事件影响力更新调整

src/main/java/com/zhiwei/brandkbs2/controller/app/AppToolsetController.java

@@ -138,4 +138,10 @@ public class AppToolsetController {
     public ResponseResult getIfUnread(){
         return ResponseResult.success(toolsetService.getIfUnread());
     }
+
+    @ApiOperation("工具库-获取项目工具库功能显示列表")
+    @GetMapping("/show-list")
+    public ResponseResult getProjectToolsetShowList(){
+        return ResponseResult.success(toolsetService.getProjectToolsetShowList());
+    }
 }

src/main/java/com/zhiwei/brandkbs2/dao/BytedanceCustomEventUpdateTaskDataDao.java

+package com.zhiwei.brandkbs2.dao;
+
+import com.zhiwei.brandkbs2.pojo.BytedanceCustomEventUpdateTaskData;
+
+/**
+ * @ClassName: BytedanceCustomEventUpdateTaskDataDao
+ * @Description BytedanceCustomEventUpdateTaskDataDao
+ * @author: cjz
+ * @date: 2023-10-24 13:30
+ */
+public interface BytedanceCustomEventUpdateTaskDataDao extends BaseMongoDao<BytedanceCustomEventUpdateTaskData> {
+}

src/main/java/com/zhiwei/brandkbs2/dao/impl/BytedanceCustomEventUpdateTaskDataDaoImpl.java

+package com.zhiwei.brandkbs2.dao.impl;
+
+import com.zhiwei.brandkbs2.dao.BytedanceCustomEventUpdateTaskDataDao;
+import com.zhiwei.brandkbs2.pojo.BytedanceCustomEventUpdateTaskData;
+import org.springframework.stereotype.Component;
+
+/**
+ * @ClassName: BytedanceCustomEventUpdateTaskDataDaoImpl
+ * @Description BytedanceCustomEventUpdateTaskDataDaoImpl
+ * @author: cjz
+ * @date: 2023-10-24 13:31
+ */
+@Component("bytedanceCustomEventUpdateTaskDataDao")
+public class BytedanceCustomEventUpdateTaskDataDaoImpl extends BaseMongoDaoImpl<BytedanceCustomEventUpdateTaskData> implements BytedanceCustomEventUpdateTaskDataDao {
+
+    private static final String COLLECTION_NAME = "brandkbs_bytedance_custom_event_task_data";
+
+    public BytedanceCustomEventUpdateTaskDataDaoImpl() {
+        super(COLLECTION_NAME);
+    }
+}

src/main/java/com/zhiwei/brandkbs2/easyexcel/listener/BytedanceEventListener.java

@@ -3,6 +3,8 @@ package com.zhiwei.brandkbs2.easyexcel.listener;
 import com.alibaba.excel.context.AnalysisContext;
 import com.alibaba.excel.event.AnalysisEventListener;
 import com.zhiwei.brandkbs2.easyexcel.dto.UploadBytedanceEventDTO;
+import com.zhiwei.brandkbs2.pojo.BytedanceCustomEventUpdateTaskData;
+import com.zhiwei.brandkbs2.util.Tools;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -17,22 +19,30 @@ import java.util.Objects;
  */
 public class BytedanceEventListener extends AnalysisEventListener<UploadBytedanceEventDTO> {
 
-    private Map<String, List<UploadBytedanceEventDTO>> map;
+    private String taskId;
 
+    private List<BytedanceCustomEventUpdateTaskData> data;
 
-    public BytedanceEventListener(Map<String, List<UploadBytedanceEventDTO>> map){
-        this.map = map;
+
+    public BytedanceEventListener(String taskId, List<BytedanceCustomEventUpdateTaskData> data){
+        this.taskId = taskId;
+        this.data = data;
     }
 
     @Override
-    public void invoke(UploadBytedanceEventDTO data, AnalysisContext context) {
-        if (map.containsKey(data.getEventName())){
-            map.get(data.getEventName()).add(data);
-        }else {
-            List<UploadBytedanceEventDTO> list = new ArrayList<>();
-            list.add(data);
-            map.put(data.getEventName(), list);
-        }
+    public void invoke(UploadBytedanceEventDTO dto, AnalysisContext context) {
+        BytedanceCustomEventUpdateTaskData taskData = new BytedanceCustomEventUpdateTaskData();
+        taskData.setTaskId(taskId);
+        taskData.setEventName(dto.getEventName());
+        taskData.setChannel(dto.getChannel());
+        taskData.setPlatform(dto.getPlatform());
+        taskData.setUrl(dto.getUrl());
+        taskData.setWeiboComment(dto.getWeiboComment());
+        taskData.setWeiboForward(dto.getWeiboForward());
+        taskData.setWeiboLike(dto.getWeiboLike());
+        taskData.setWechatRead(dto.getWechatRead());
+        taskData.setWechatReading(dto.getWechatReading());
+        data.add(taskData);
     }
 
     @Override

src/main/java/com/zhiwei/brandkbs2/pojo/BytedanceCustomEventUpdateTask.java

@@ -42,11 +42,11 @@ public class BytedanceCustomEventUpdateTask extends AbstractBaseMongo{
     /**
      * 事件数据
      */
-    private Map<String, List<UploadBytedanceEventDTO>> eventData;
+//    private Map<String, List<UploadBytedanceEventDTO>> eventData;
     /**
      * 报错数据
      */
-    private List<UploadBytedanceEventDTO> errorData;
+    private List<BytedanceCustomEventUpdateTaskData> errorData;
     /**
      * 竞品影响力指数
      */
@@ -136,7 +136,7 @@ public class BytedanceCustomEventUpdateTask extends AbstractBaseMongo{
         task.setTaskStatus(TaskStatus.CALCULATING.getStatus());
         task.setProcessStatus(ProcessStatus.CHANNEL_MATCH.getStatus());
         task.setExtraCompute(extraCompute);
-        task.setEventInfo(null);
+        task.setEventInfo(new ArrayList<>());
         task.setErrorData(new ArrayList<>());
         task.setCancel(false);
         task.setProjectId(UserThreadLocal.getProjectId());

src/main/java/com/zhiwei/brandkbs2/pojo/BytedanceCustomEventUpdateTaskData.java

+package com.zhiwei.brandkbs2.pojo;
+
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+/**
+ * @ClassName: BytedanceCustomEventInfluenceTask
+ * @Description 字节定制事件影响力计算rw
+ * @author: cjz
+ * @date: 2023-10-16 15:12
+ */
+@Getter
+@Setter
+@NoArgsConstructor
+public class BytedanceCustomEventUpdateTaskData extends AbstractBaseMongo {
+    private String taskId;
+
+    private String platform;
+
+    private String channel;
+
+    private String url;
+
+    private String eventName;
+
+    private Integer weiboForward;
+
+    private Integer weiboComment;
+
+    private Integer weiboLike;
+
+    private Integer wechatRead;
+
+    private Integer wechatReading;
+}

src/main/java/com/zhiwei/brandkbs2/pojo/Project.java

@@ -123,9 +123,7 @@ public class Project extends AbstractProject {
         projectVO.setContendList(this.getContendList());
         // 模块配置
         projectVO.setModuleShowList(this.getModuleShowList());
-        if (Objects.nonNull(this.getToolsetShowList())){
         projectVO.setToolsetShowList(this.getToolsetShowList());
-        }
         projectVO.setChannelFileUrl(this.getChannelFileUrl());
         projectVO.setNegativeChannelParams(this.getNegativeChannelParams());
         projectVO.setPositiveChannelParams(this.getPositiveChannelParams());

src/main/java/com/zhiwei/brandkbs2/pojo/vo/ProjectVO.java

@@ -100,7 +100,7 @@ public class ProjectVO {
      * 工具库功能入口展示列表（目前有：摘要提取，互动量更新，词云，字节影响力计算，其中前三者默认开启）
      */
     @ApiModelProperty("工具库功能入口展示列表（目前有：摘要提取，互动量更新，词云，字节影响力计算，其中前三者默认开启）")
-    private List<String> toolsetShowList = Arrays.asList("articleSummary", "interactionUpdate", "highWord");
+    private List<String> toolsetShowList;
     /**
      * 黑渠道对应组
      */

src/main/java/com/zhiwei/brandkbs2/service/ToolsetService.java

@@ -148,4 +148,10 @@ public interface ToolsetService {
      * @param taskId    任务id
      */
     void cancelTask(String taskId);
+
+    /**
+     * 获取项目工具库功能显示列表
+     * @return
+     */
+    List<String> getProjectToolsetShowList();
 }

src/main/java/com/zhiwei/brandkbs2/service/impl/ProjectServiceImpl.java

@@ -443,27 +443,27 @@ public class ProjectServiceImpl implements ProjectService {
     @Override
     public PageVO<JSONObject> getToolsetBytedanceCustomInfo(int page, int size, String type) {
         Query query = new Query();
+        long channelTotal = bytedanceCustomChannelDao.count(query);
+        long channelWeightTotal = bytedanceCustomPlatformWeightDao.count(query);
+        long channelInfluenceTotal = bytedanceCustomChannelInfluenceDao.count(query);
         query.limit(size);
         query.skip((long) (page - 1) * size);
         switch (type){
             case "重要媒体":
-                long total = bytedanceCustomChannelDao.count(query);
-                return PageVO.createPageVo(total, page, size, bytedanceCustomChannelDao.findList(query).stream().map(data ->{
+                return PageVO.createPageVo(channelTotal, page, size, bytedanceCustomChannelDao.findList(query).stream().map(data ->{
                     JSONObject jsonObject = new JSONObject();
                     jsonObject.put("channel", data.getChannel());
                     return jsonObject;
                 }).collect(Collectors.toList()));
             case "平台权重":
-                long total2 = bytedanceCustomPlatformWeightDao.count(query);
-                return PageVO.createPageVo(total2, page, size, bytedanceCustomPlatformWeightDao.findList(query).stream().map(data ->{
+                return PageVO.createPageVo(channelWeightTotal, page, size, bytedanceCustomPlatformWeightDao.findList(query).stream().map(data ->{
                     JSONObject jsonObject = new JSONObject();
                     jsonObject.put("type", data.getType());
                     jsonObject.put("weight", data.getWeight());
                     return jsonObject;
                 }).collect(Collectors.toList()));
             case "渠道影响力":
-                long total3 = bytedanceCustomChannelInfluenceDao.count(query);
-                return PageVO.createPageVo(total3, page, size, bytedanceCustomChannelInfluenceDao.findList(query).stream().map(data ->{
+                return PageVO.createPageVo(channelInfluenceTotal, page, size, bytedanceCustomChannelInfluenceDao.findList(query).stream().map(data ->{
                     JSONObject jsonObject = new JSONObject();
                     jsonObject.put("platform", data.getPlatform());
                     jsonObject.put("channel", data.getChannel());

src/main/resources/application-dev.properties

 application.name = brandkbs2
 server.port=8888
 spring.flyway.encoding=UTF-8
+#\u5355\u4E2A\u4E0A\u4F20\u6587\u4EF6\u5927\u5C0F\u8BBE\u7F6E
+spring.servlet.multipart.max-file-size=10MB
+#\u603B\u4E0A\u4F20\u6587\u4EF6\u5927\u5C0F\u8BBE\u7F6E
+spring.servlet.multipart.max-request-size=20MB
 server.servlet.context-path=/brandkbs
 #jwt
 jwt.key=Token

src/main/resources/application-local.properties

 application.name = brandkbs2
 server.port=8888
 spring.flyway.encoding=UTF-8
+#\u5355\u4E2A\u4E0A\u4F20\u6587\u4EF6\u5927\u5C0F\u8BBE\u7F6E
+spring.servlet.multipart.max-file-size=10MB
+#\u603B\u4E0A\u4F20\u6587\u4EF6\u5927\u5C0F\u8BBE\u7F6E
+spring.servlet.multipart.max-request-size=20MB
 server.servlet.context-path=/brandkbs
 #jwt
 jwt.key=Token
 jwt.hour=120
 #\u8DEF\u5F84\u5B58\u653E
-brandkbs.file.url=E:\\work3\\brandkbs3\\
-brandkbs.img.url=E:\\work\\brandkbs2\\img\\
-brandkbs.head.url=E:\\work\\brandkbs2\\head\\
+brandkbs.file.url=D:\\ExcelTest\\
+brandkbs.img.url=D:\\ExcelTest\\
+brandkbs.head.url=D:\\ExcelTest\\
 brandkbs.image.url=https://brandkbs.zhiweidata.com/brandkbs/images/
 #\u56FE\u7247\u8D44\u6E90\u8DEF\u5F84
 cbs.imagesPath=file:${brandkbs.img.url},file:${brandkbs.head.url}

src/main/resources/application-prod.properties

 application.name = brandkbs2
 server.port=8888
 spring.flyway.encoding=UTF-8
+#\u5355\u4E2A\u4E0A\u4F20\u6587\u4EF6\u5927\u5C0F\u8BBE\u7F6E
+spring.servlet.multipart.max-file-size=10MB
+#\u603B\u4E0A\u4F20\u6587\u4EF6\u5927\u5C0F\u8BBE\u7F6E
+spring.servlet.multipart.max-request-size=20MB
 server.servlet.context-path=/brandkbs
 #jwt
 jwt.key=Token