完成发布渠道评估实训开发

pom.xml

@@ -293,6 +293,17 @@
         </dependency>
 
 
+        <!--        //词云-->
+        <dependency>
+            <groupId>com.kennycason</groupId>
+            <artifactId>kumo-core</artifactId>
+            <version>1.28</version>
+        </dependency>
+        <dependency>
+            <groupId>com.kennycason</groupId>
+            <artifactId>kumo-tokenizers</artifactId>
+            <version>1.28</version>
+        </dependency>
 
 
 <!--        <dependency>-->

src/main/java/com/sztzjy/linkCommerce/controller/stu/CptfController.java

@@ -8,6 +8,7 @@ import com.sztzjy.linkCommerce.mapper.*;
 import com.sztzjy.linkCommerce.service.ScoreService;
 import com.sztzjy.linkCommerce.util.CompareListsUtil;
 import com.sztzjy.linkCommerce.util.ResultEntity;
+import com.sztzjy.linkCommerce.util.StringUtils;
 import com.sztzjy.linkCommerce.util.file.IFileUtil;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -55,106 +56,106 @@ public class CptfController {
     public String url;
 
 
-//    @PostMapping("/commitCpdjAnswer")
+    @PostMapping("/commitCpdjAnswer")
-//    @ApiOperation("发布渠道-提交")
+    @ApiOperation("发布渠道-提交")
-//    @AnonymousAccess
+    @AnonymousAccess
-//    public ResultEntity commitFbqdAnswer(@RequestBody FbqdStuAnswer stuAnswer) {
+    public ResultEntity commitFbqdAnswer(@RequestBody FbqdStuAnswer stuAnswer) {
-//        //先查询是否提交过
+        //先查询是否提交过
-//        StuGradeExample stuGradeExample = new StuGradeExample();
+        StuGradeExample stuGradeExample = new StuGradeExample();
-//        stuGradeExample.createCriteria().andUserIdEqualTo(stuAnswer.getUserId()).andProjectEqualTo(Constant.PRO_CPTFCS).andModuleEqualTo(Constant.MODULE_CPDJ);
+        stuGradeExample.createCriteria().andUserIdEqualTo(stuAnswer.getUserId()).andProjectEqualTo(Constant.PRO_CPTFCS).andModuleEqualTo(Constant.MODULE_CPDJ);
-//        List<StuGrade> stuGradeList = stuGradeMapper.selectByExampleWithBLOBs(stuGradeExample);
+        List<StuGrade> stuGradeList = stuGradeMapper.selectByExampleWithBLOBs(stuGradeExample);
-//        if(CollectionUtils.isEmpty(stuGradeList)){
+        if(CollectionUtils.isEmpty(stuGradeList)){
-//            FbqdAnswer answer = fbqdAnswerMapper.selectByExampleWithBLOBs(new FbqdAnswerExample()).get(0);
+            FbqdAnswer answer = fbqdAnswerMapper.selectByExampleWithBLOBs(new FbqdAnswerExample()).get(0);
-//            int count = 0;
+            int count = 0;
-//
+
-//            //1.数据标准化结果
+            //1.数据标准化结果
-//            String stuExposureData = stuAnswer.getExposureData();
+            String stuExposureData = stuAnswer.getExposureData();
-//
+
-//            String exposureData = answer.getExposureData();
+            String exposureData = answer.getExposureData();
-//            //数据标准化结果 校验
+            //数据标准化结果 校验
-//
+
-//            //错误次数
+            //错误次数
-//            int monthlyDataErrorCount =compareListsUtil.compareErrorCount(stuExposureData, exposureData).getErrorCount();
+            int monthlyDataErrorCount =compareListsUtil.compareErrorCount(stuExposureData, exposureData).getErrorCount();
-//            count = count + monthlyDataErrorCount;
+            count = count + monthlyDataErrorCount;
-//
+
-//
+
-//            //2.渠道中的比重
+            //2.渠道中的比重
-//            String stuAnswerClickData = stuAnswer.getClickData();
+            String stuAnswerClickData = stuAnswer.getClickData();
-//
+
-//            String clickData = answer.getClickData();
+            String clickData = answer.getClickData();
-//            //错误次数
+            //错误次数
-//            int cheapSumCount =compareListsUtil.compareErrorCount(stuAnswerClickData, clickData).getErrorCount();
+            int cheapSumCount =compareListsUtil.compareErrorCount(stuAnswerClickData, clickData).getErrorCount();
-//            count = count + cheapSumCount;
+            count = count + cheapSumCount;
-//
+
-//
+
-//
+
-//            //3.信息滴
+            //3.信息滴
-//            String stuAnswerAdCost = stuAnswer.getAdCost();
+            String stuAnswerAdCost = stuAnswer.getAdCost();
-//
+
-//            String adCost = answer.getAdCost();
+            String adCost = answer.getAdCost();
-//            //错误次数
+            //错误次数
-//            int cheapPercentageCount =compareListsUtil.compareErrorCount(stuAnswerAdCost, adCost).getErrorCount();
+            int cheapPercentageCount =compareListsUtil.compareErrorCount(stuAnswerAdCost, adCost).getErrorCount();
-//            count = count + cheapPercentageCount;
+            count = count + cheapPercentageCount;
-//
+
-//
+
-//            //每个指标的权重
+            //每个指标的权重
-//            String stuAnswerOrderAmount = stuAnswer.getOrderAmount();
+            String stuAnswerOrderAmount = stuAnswer.getOrderAmount();
-//
+
-//            String orderAmount = answer.getOrderAmount();
+            String orderAmount = answer.getOrderAmount();
-//            //错误次数
+            //错误次数
-//            int costEffectiveSumCount =compareListsUtil.compareErrorCount(stuAnswerOrderAmount, orderAmount).getErrorCount();
+            int costEffectiveSumCount =compareListsUtil.compareErrorCount(stuAnswerOrderAmount, orderAmount).getErrorCount();
-//            count = count + costEffectiveSumCount;
+            count = count + costEffectiveSumCount;
-//
+
-//
+
-//
+
-//            //综合评分
+            //综合评分
-//            String stuAnswerKeywords = stuAnswer.getKeywords();
+            String stuAnswerKeywords = stuAnswer.getKeywords();
-//
+
-//            String keywords = answer.getKeywords();
+            String keywords = answer.getKeywords();
-//            //错误次数
+            //错误次数
-//            int costEffectivePercentageCount =compareListsUtil.compareErrorCount(stuAnswerKeywords, keywords).getErrorCount();
+            int costEffectivePercentageCount =compareListsUtil.compareErrorCount(stuAnswerKeywords, keywords).getErrorCount();
-//            count = count + costEffectivePercentageCount;
+            count = count + costEffectivePercentageCount;
-//
+
-//
+
-//
+
-//            //主观题
+            //主观题
-//            int i = compareListsUtil.calculateLengthPenaltyEnhanced(stuAnswer.getScenePromotion());
+            int i = compareListsUtil.calculateLengthPenaltyEnhanced(stuAnswer.getScenePromotion());
-//            count = count + i;
+            count = count + i;
-//            stuAnswer.setId(IdUtil.simpleUUID());
+            stuAnswer.setId(IdUtil.simpleUUID());
-//
+
-//            fbqdStuAnswerMapper.insertSelective(stuAnswer);
+            fbqdStuAnswerMapper.insertSelective(stuAnswer);
-//
+
-//            Boolean aBoolean = scoreService.insertScore(stuAnswer.getUserId(), Constant.PRO_CPTFCS, Constant.MODULE_FBQDPG, BigDecimal.valueOf(100 - count), stuAnswer.getScenePromotion());
+            Boolean aBoolean = scoreService.insertScore(stuAnswer.getUserId(), Constant.PRO_CPTFCS, Constant.MODULE_FBQDPG, BigDecimal.valueOf(100 - count), stuAnswer.getScenePromotion());
-//            if(aBoolean){
+            if(aBoolean){
-//                return new ResultEntity<>(HttpStatus.OK, "提交成功");
+                return new ResultEntity<>(HttpStatus.OK, "提交成功");
-//            }else {
+            }else {
-//                return new ResultEntity<>(HttpStatus.BAD_REQUEST, "提交失败");
+                return new ResultEntity<>(HttpStatus.BAD_REQUEST, "提交失败");
-//            }
+            }
-//        }else {
+        }else {
-//            return new ResultEntity<>(HttpStatus.BAD_REQUEST, "已提交过");
+            return new ResultEntity<>(HttpStatus.BAD_REQUEST, "已提交过");
-//        }
+        }
-//    }
+    }
-//
+
-//
+
-//
+
-//    @ApiOperation("发布渠道-回显接口")
+    @ApiOperation("发布渠道-回显接口")
-//    @GetMapping("/selectByCpdjUserId")
+    @GetMapping("/selectByCpdjUserId")
-//    @AnonymousAccess
+    @AnonymousAccess
-//    public ResultEntity selectByCpdjUserId(@ApiParam("用户ID") String userId) {
+    public ResultEntity selectByCpdjUserId(@ApiParam("用户ID") String userId) {
-//
+
-//        FbqdStuAnswerExample cpdjStuAnswerExample = new FbqdStuAnswerExample();
+        FbqdStuAnswerExample cpdjStuAnswerExample = new FbqdStuAnswerExample();
-//        cpdjStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
+        cpdjStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
-//
+
-//        List<FbqdStuAnswer> stuAnswerList = fbqdStuAnswerMapper.selectByExampleWithBLOBs(cpdjStuAnswerExample);
+        List<FbqdStuAnswer> stuAnswerList = fbqdStuAnswerMapper.selectByExampleWithBLOBs(cpdjStuAnswerExample);
-//
+
-//        if (CollectionUtils.isEmpty(stuAnswerList)) {
+        if (CollectionUtils.isEmpty(stuAnswerList)) {
-//            return new ResultEntity<>(HttpStatus.OK);
+            return new ResultEntity<>(HttpStatus.OK);
-//        } else {
+        } else {
-//            return new ResultEntity<>(HttpStatus.OK, stuAnswerList.get(0));
+            return new ResultEntity<>(HttpStatus.OK, stuAnswerList.get(0));
-//        }
+        }
-//
+
-//    }
+    }
-//
+
 
 
 
@@ -186,20 +187,30 @@ public class CptfController {
                 return new ResultEntity<>(HttpStatus.BAD_REQUEST, "重新实训失败");
             }
         }
-//
-//        if (Constant.MODULE_CPJGGH.equals(module)) {
-//            Boolean b=scoreService.deleteScore(userId,Constant.PRO_CPGH,module);
-//            if(b){
-//                CpjgghStuAnswerExample cpjgghStuAnswerExample = new CpjgghStuAnswerExample();
-//                cpjgghStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
-//                cpjgghStuAnswerMapper.deleteByExample(cpjgghStuAnswerExample);
-//                return new ResultEntity<>(HttpStatus.OK, "重新实训成功");
-//            }else {
-//                return new ResultEntity<>(HttpStatus.BAD_REQUEST, "重新实训失败");
-//            }
-//        }
 
+        if (Constant.MODULE_CPTGCS.equals(module)) {
+            Boolean b=scoreService.deleteScore(userId,Constant.PRO_CPGH,module);
+            if(b){
+                CptgcsStuAnswerExample cptgcsStuAnswerExample = new CptgcsStuAnswerExample();
+                cptgcsStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
+                cptgcsStuAnswerMapper.deleteByExample(cptgcsStuAnswerExample);
+                return new ResultEntity<>(HttpStatus.OK, "重新实训成功");
+            }else {
+                return new ResultEntity<>(HttpStatus.BAD_REQUEST, "重新实训失败");
+            }
+        }
 
+        if (Constant.MODULE_CPYXDW.equals(module)) {
+            Boolean b=scoreService.deleteScore(userId,Constant.PRO_CPGH,module);
+            if(b){
+                CpyxdwStuAnswerExample cpyxdwStuAnswerExample = new CpyxdwStuAnswerExample();
+                cpyxdwStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
+                cpyxdwStuAnswerMapper.deleteByExample(cpyxdwStuAnswerExample);
+                return new ResultEntity<>(HttpStatus.OK, "重新实训成功");
+            }else {
+                return new ResultEntity<>(HttpStatus.BAD_REQUEST, "重新实训失败");
+            }
+        }
 
 
         return new ResultEntity<>(HttpStatus.BAD_REQUEST, "重新实训失败");
@@ -217,7 +228,7 @@ public class CptfController {
 
 
     @PostMapping("/commitCptgcsAnswer")
-    @ApiOperation("发布推广测试-提交")
+    @ApiOperation("产品推广测试-提交")
     @AnonymousAccess
     public ResultEntity commitCptgcsAnswer(@RequestBody CptgcsStuAnswer stuAnswer) {
         //先查询是否提交过
@@ -298,15 +309,17 @@ public class CptfController {
 
 
 
-    @ApiOperation("发布渠道-回显接口")
+
-    @GetMapping("/selectByCpdjUserId")
+
+    @ApiOperation("产品推广测试-回显接口")
+    @GetMapping("/selectCptgcsByUserId")
     @AnonymousAccess
-    public ResultEntity selectByCpdjUserId(@ApiParam("用户ID") String userId) {
+    public ResultEntity selectCptgcsByUserId(@ApiParam("用户ID") String userId) {
 
         CptgcsStuAnswerExample cptgcsStuAnswerExample = new CptgcsStuAnswerExample();
         cptgcsStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
 
-        List< CptgcsStuAnswer> stuAnswerList =  cptgcsStuAnswerMapper.selectByExampleWithBLOBs(cptgcsStuAnswerExample);
+        List<CptgcsStuAnswer> stuAnswerList = cptgcsStuAnswerMapper.selectByExampleWithBLOBs(cptgcsStuAnswerExample);
 
         if (CollectionUtils.isEmpty(stuAnswerList)) {
             return new ResultEntity<>(HttpStatus.OK);
@@ -316,6 +329,85 @@ public class CptfController {
 
     }
 
+    @Autowired
+    private CpyxdwStuAnswerMapper cpyxdwStuAnswerMapper;
+
+
+
+
+
+    @PostMapping("/commitCpyxdwAnswer")
+    @ApiOperation("产品营销定位-提交")
+    @AnonymousAccess
+    public ResultEntity commitCpyxdwAnswer(@RequestBody CpyxdwStuAnswer stuAnswer) {
+        //先查询是否提交过
+        StuGradeExample stuGradeExample = new StuGradeExample();
+        stuGradeExample.createCriteria().andUserIdEqualTo(stuAnswer.getUserId()).andProjectEqualTo(Constant.PRO_CPTFCS).andModuleEqualTo(Constant.MODULE_CPYXDW);
+        List<StuGrade> stuGradeList = stuGradeMapper.selectByExampleWithBLOBs(stuGradeExample);
+        if(CollectionUtils.isEmpty(stuGradeList)){
+
+            int count = 0;
+
+            //1.产品点击率
+            String axis = stuAnswer.getxAxis();
+            if (axis!=null)
+            {
+                if (!"点击率".equals(axis)) {
+                    count++;
+                }
+            }
+
+
+            String yAxis = stuAnswer.getyAxis();
+            if (yAxis!=null)
+            {
+                if (!"转换率".equals(yAxis)) {
+                    count++;
+                }
+            }
+
+
+
+
+            //主观题
+            int i = compareListsUtil.calculateLengthPenaltyEnhanced(stuAnswer.getAnalysis());
+            count = count + i;
+            stuAnswer.setId(IdUtil.simpleUUID());
+
+            cpyxdwStuAnswerMapper.insertSelective(stuAnswer);
+
+            Boolean aBoolean = scoreService.insertScore(stuAnswer.getUserId(), Constant.PRO_CPTFCS, Constant.MODULE_CPYXDW, BigDecimal.valueOf(100 - count), stuAnswer.getAnalysis());
+            if(aBoolean){
+                return new ResultEntity<>(HttpStatus.OK, "提交成功");
+            }else {
+                return new ResultEntity<>(HttpStatus.BAD_REQUEST, "提交失败");
+            }
+        }else {
+            return new ResultEntity<>(HttpStatus.BAD_REQUEST, "已提交过");
+        }
+    }
+
+
+
+
+
+    @ApiOperation("产品营销定位-回显接口")
+    @GetMapping("/selectCpyxdwByUserId")
+    @AnonymousAccess
+    public ResultEntity selectCpyxdwByUserId(@ApiParam("用户ID") String userId) {
+
+        CpyxdwStuAnswerExample cpyxdwStuAnswerExample = new CpyxdwStuAnswerExample();
+        cpyxdwStuAnswerExample.createCriteria().andUserIdEqualTo(userId);
+
+        List<CpyxdwStuAnswer> stuAnswerList = cpyxdwStuAnswerMapper.selectByExampleWithBLOBs(cpyxdwStuAnswerExample);
+
+        if (CollectionUtils.isEmpty(stuAnswerList)) {
+            return new ResultEntity<>(HttpStatus.OK);
+        } else {
+            return new ResultEntity<>(HttpStatus.OK, stuAnswerList.get(0));
+        }
+
+    }
 
 
 

src/main/java/com/sztzjy/linkCommerce/controller/stu/DownloadDataController.java

@@ -0,0 +1,152 @@
+package com.sztzjy.linkCommerce.controller.stu;
+
+import cn.hutool.core.util.IdUtil;
+import com.sztzjy.linkCommerce.annotation.AnonymousAccess;
+import com.sztzjy.linkCommerce.entity.dto.StuWordCreateDto;
+import com.sztzjy.linkCommerce.util.TagWordCloudService;
+import com.sztzjy.linkCommerce.util.file.IFileUtil;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
+import java.io.IOException;
+
+/**
+ * @author 17803
+ * @date 2024-06-17 10:29
+ */
+
+@Api(tags = "用户画像模型下载数据")
+@RestController
+@RequestMapping("api/downDataByUserPortrait")
+public class DownloadDataController {
+
+
+    @Autowired
+    private IFileUtil iFileUtil;
+
+    @Value("${file.path}")
+    private String filePath;
+
+
+    //@AnonymousAccess
+    @ApiOperation("RFM分析方法下载数据")
+    @GetMapping("/analysisByRfmDown")
+    @AnonymousAccess
+    public void analysisByRfmDown(HttpServletResponse response){
+
+        iFileUtil.download(response,"RFM分析方法data.xls");
+
+    }
+
+
+    //@AnonymousAccess
+    @ApiOperation("帕累托分析方法下载数据")
+    @GetMapping("/analysisByPareto")
+    @AnonymousAccess
+    public void analysisByPareto(HttpServletResponse response){
+
+        iFileUtil.download(response,"帕累托分析方法data.xls");
+
+    }
+
+
+
+    //@AnonymousAccess
+    @ApiOperation("波士顿矩阵分析图下载数据")
+    @GetMapping("/bostonMatrixAnalysisData")
+    @AnonymousAccess
+    public void bostonMatrixAnalysisData(HttpServletResponse response){
+
+        iFileUtil.download(response,"波士顿矩阵分析图data.xls");
+
+    }
+
+
+
+
+    //@AnonymousAccess
+    @ApiOperation("漏斗模型分析下载数据")
+    @GetMapping("/analysisByFunnelModel")
+    @AnonymousAccess
+    public void analysisByFunnelModel(HttpServletResponse response){
+
+        iFileUtil.download(response,"漏斗模型分析data.xls");
+
+    }
+
+
+    //@AnonymousAccess
+    @ApiOperation("购物篮分析下载数据")
+    @GetMapping("/analysisByShopBasket")
+    @AnonymousAccess
+    public void analysisByShopBasket(HttpServletResponse response){
+
+        iFileUtil.download(response,"购物篮分析data.xls");
+
+    }
+
+
+    //@AnonymousAccess
+    @ApiOperation("复购分析下载数据")
+    @GetMapping("/analysisByRepurchase")
+    @AnonymousAccess
+    public void analysisByRepurchase(HttpServletResponse response){
+
+        iFileUtil.download(response,"复购分析data.xls");
+
+    }
+
+    //@AnonymousAccess
+    @ApiOperation("留存分析下载数据")
+    @GetMapping("/analysisByRetention")
+    @AnonymousAccess
+    public void analysisByRetention(HttpServletResponse response){
+
+        iFileUtil.download(response,"留存分析data.xls");
+
+    }
+
+    //@AnonymousAccess
+    @ApiOperation("用户粘性分析下载数据")
+    @GetMapping("/analysisByStickiness")
+    @AnonymousAccess
+    public void analysisByStickiness(HttpServletResponse response){
+
+        iFileUtil.download(response,"用户粘性分析data.xls");
+
+    }
+
+
+    //@AnonymousAccess
+    @ApiOperation("情感分析/文本挖掘词云生成")
+    @PostMapping("/wordCreateByAna")
+    @AnonymousAccess
+    public void wordCreateByAna(@RequestBody StuWordCreateDto stuWordCreateDto) throws IOException {
+
+        TagWordCloudService tagWordCloudService = new TagWordCloudService();
+
+        File file1 = new File(filePath + "/word/");
+
+        if (! file1.exists())
+        {
+            file1.mkdir();
+        }
+
+        //1成图片路径，2.图云背景图片
+        String url =   filePath+"/word/" + IdUtil.simpleUUID()+"."+"png";
+
+
+        tagWordCloudService.generate(url,filePath+"/1.png",stuWordCreateDto.getContext());
+
+
+    }
+
+
+
+
+}

src/main/java/com/sztzjy/linkCommerce/controller/stu/UserBehaviorProfilingAnaController.java

@@ -2,10 +2,12 @@ package com.sztzjy.linkCommerce.controller.stu;
 
 
 import com.hankcs.hanlp.collection.sequence.SString;
+import com.sztzjy.linkCommerce.annotation.AnonymousAccess;
 import com.sztzjy.linkCommerce.entity.StuMachineLearning;
 import com.sztzjy.linkCommerce.service.UserBehaviorProfilingAnaService;
 import com.sztzjy.linkCommerce.util.ResultDataEntity;
 import com.sztzjy.linkCommerce.util.ResultEntity;
+import com.sztzjy.linkCommerce.util.file.IFileUtil;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import io.swagger.annotations.ApiParam;
@@ -13,6 +15,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletResponse;
 import java.util.List;
 
 
@@ -54,5 +57,17 @@ public class UserBehaviorProfilingAnaController {
 
     }
 
+    @Autowired
+    private IFileUtil iFileUtil;
+
+    @ApiOperation("RFM分析方法下载数据")
+    @GetMapping("/analysisByRfmDown")
+    @AnonymousAccess
+    public void analysisByRfmDown(HttpServletResponse response){
+
+        iFileUtil.download(response,"RFM分析方法data.xls");
+
+    }
+
 
 }

src/main/java/com/sztzjy/linkCommerce/entity/CpyxdwStuAnswer.java

@@ -13,10 +13,10 @@ public class CpyxdwStuAnswer {
     @ApiModelProperty(notes = "主键（格式：CPYXSA_时间戳_哈希值，例：CPYXSA_2023110614_9d7b6c5e）")
     private String id;
 
-    @ApiModelProperty(notes = "横轴定位维度（如：价格水平、技术含量等）")
+    @ApiModelProperty(notes = "横轴定位维度")
     private String xAxis;
 
-    @ApiModelProperty(notes = "纵轴定位维度（如：市场份额、品牌认知度等）")
+    @ApiModelProperty(notes = "纵轴定位维度")
     private String yAxis;
 
     @ApiModelProperty(notes = "用户ID（关联userinfo表）")
@@ -31,7 +31,7 @@ public class CpyxdwStuAnswer {
     @ApiModelProperty(notes = "定位坐标数据（JSON格式，")
     private String positionData;
 
-    @ApiModelProperty(notes = "定位分析说明（500字以内）")
+    @ApiModelProperty(notes = "主观题（500字以内）")
     private String analysis;
 
     public String getId() {

src/main/java/com/sztzjy/linkCommerce/entity/dto/StuWordCreateDto.java

@@ -0,0 +1,16 @@
+package com.sztzjy.linkCommerce.entity.dto;
+
+import lombok.Data;
+
+/**
+ * @author 17803
+ * @date 2024-06-21 14:21
+ */
+
+@Data
+public class StuWordCreateDto {
+
+    private String context;
+    private String userId;
+
+}

src/main/java/com/sztzjy/linkCommerce/util/TagWordCloudService.java

@@ -0,0 +1,111 @@
+package com.sztzjy.linkCommerce.util;
+
+import com.kennycason.kumo.CollisionMode;
+import com.kennycason.kumo.WordCloud;
+import com.kennycason.kumo.WordFrequency;
+import com.kennycason.kumo.bg.CircleBackground;
+import com.kennycason.kumo.font.KumoFont;
+import com.kennycason.kumo.font.scale.SqrtFontScalar;
+import com.kennycason.kumo.image.AngleGenerator;
+import com.kennycason.kumo.nlp.FrequencyAnalyzer;
+import com.kennycason.kumo.nlp.tokenizers.ChineseWordTokenizer;
+import com.kennycason.kumo.palette.ColorPalette;
+
+import java.awt.*;
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+
+public class TagWordCloudService {
+
+    // 照片纵横比
+    private double ratio = 1;
+    // 获取词云图片
+    // wordFile：单词及其频率文件路径
+    // pngOutputPath：图片输出路径，应该以.png结尾
+    // shapePicPath：词云形状图片路径，其背景应为透明背景，格式为png
+    public void generate(String pngOutputPath, String shapePicPath,String context) throws IOException {
+//        final FrequencyAnalyzer frequencyAnalyzer = new FrequencyAnalyzer();
+        // 共检索多少个词
+//        frequencyAnalyzer.setWordFrequenciesToReturn(1000);
+        // 单词最短长度，一个汉字和一个英文字符都是1
+//        frequencyAnalyzer.setMinWordLength(2);
+//        frequencyAnalyzer.setStopWords(loadStopWords());
+        // 设置中文支持，另一种加载方式不用设置
+//        frequencyAnalyzer.setWordTokenizer(new ChineseWordTokenizer());
+//        final List<WordFrequency> wordFrequencies;
+//        // 加载词云有两种方式，一种是在txt文件中统计词出现的个数，另一种是直接给出每个词出现的次数，这里使用第二种
+//        // 文件格式如下
+////        100: frog
+////        94: dog
+////        43: cog
+////        20: bog
+//        FrequencyFileLoader frequencyFileLoader = new FrequencyFileLoader();
+//        File file = new File(wordFile);
+//        wordFrequencies = frequencyFileLoader.load(file);
+        // 可以直接从文件中读取
+        FrequencyAnalyzer frequencyAnalyzer = new FrequencyAnalyzer();
+        frequencyAnalyzer.setWordFrequenciesToReturn(600);
+        frequencyAnalyzer.setMinWordLength(2);
+        frequencyAnalyzer.setWordTokenizer(new ChineseWordTokenizer());
+        InputStream inputStream = null;
+        List<WordFrequency> wordFrequencies = null;
+
+        try  {
+            inputStream = new ByteArrayInputStream(context.getBytes(StandardCharsets.UTF_8));
+             wordFrequencies = frequencyAnalyzer.load(inputStream);
+
+        }catch (Exception e) {
+            e.printStackTrace();
+        }finally {
+            if (inputStream != null) {
+                try {
+                    inputStream.close();
+                } catch (Exception e) {
+                    e.printStackTrace();
+                }
+            }
+        }
+
+//        final Dimension dimension = new Dimension(600, 600);
+//        final WordCloud wordCloud = new WordCloud(dimension, CollisionMode.PIXEL_PERFECT);
+//        wordCloud.setPadding(2);
+//        wordCloud.setBackground(new CircleBackground(300));
+       // wordCloud.setColorPalette(new ColorPalette(new Color(0x4055F1), new Color(0x408DF1), new Color(0x40AAF1), new Color(0x40C5F1), new Color(0x40D3F1), new Color(0xFFFFFF)));
+//        wordCloud.setFontScalar(new SqrtFontScalar(10, 40));
+
+
+        // 生成图片的像素大小
+        //设置图片分辨率
+        Dimension dimension = new Dimension(600, 600);
+//        final Dimension dimension = new Dimension(1024, (int)(1024*ratio));
+        final WordCloud wordCloud = new WordCloud(dimension, CollisionMode.PIXEL_PERFECT);
+        // 调节词云的稀疏程度，越高越稀疏
+        wordCloud.setPadding(2);
+
+//        //设置背景色
+        wordCloud.setBackgroundColor(new Color(255,255,255));
+        //设置背景图片
+        //wordCloud.setBackground(new PixelBoundaryBackground(shapePicPath));
+//        wordCloud.setBackground(new CircleBackground(300));
+        wordCloud.setBackground(new CircleBackground(300));
+
+        // 颜色模板，不同频率的颜色会不同
+       wordCloud.setColorPalette(new ColorPalette(new Color(0x4055F1), new Color(0x408DF1), new Color(0x40AAF1), new Color(0x40C5F1), new Color(0x40D3F1), new Color(0xFFFFFF)));
+        // 设置字体
+        Font font = new Font("楷体", 0, 25);
+        wordCloud.setKumoFont(new KumoFont(font));
+        // 设置偏转角，角度为0时，字体都是水平的
+//        wordCloud.setAngleGenerator(new AngleGenerator(0, 90, 9));
+        wordCloud.setAngleGenerator(new AngleGenerator(0));
+        // 字体的大小范围，最小是多少，最大是多少
+        wordCloud.setFontScalar(new SqrtFontScalar(10, 40));
+
+        wordCloud.build(wordFrequencies);
+        wordCloud.writeToFile(pngOutputPath);
+
+
+    }
+}