采集详情

huliguo

4 天以前 b36ba9d3da79245376f5a8a1c31cddb1f9a6d1b5

采集详情

 pom.xml

@@ -99,6 +99,7 @@
            <artifactId>jackson-databind</artifactId>
            <version>2.15.2</version>
        </dependency>

        <dependency>
            <groupId>com.opencrypt</groupId>
            <artifactId>open-crypt</artifactId>

 src/main/java/com/linghu/config/MybatisPlusConfig.java

New file
@@ -0,0 +1,24 @@
package com.linghu.config;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
@MapperScan("com.linghu.mapper")
public class MybatisPlusConfig {

    /**
     * 添加分页插件
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL)); // 如果配置多个插件, 切记分页最后添加
        // 如果有多数据源可以不配具体类型, 否则都建议配上具体的 DbType
        return interceptor;
    }
}

 src/main/java/com/linghu/controller/KeywordController.java

@@ -6,9 +6,16 @@
import com.linghu.listener.KeywordExcelListener;
import com.linghu.mapper.KeywordMapper;
import com.linghu.model.common.ResponseResult;
import com.linghu.model.dto.ExportFeedDTO;

import com.linghu.model.entity.Keyword;
import com.linghu.model.entity.Platform;
import com.linghu.model.entity.Reference;
import com.linghu.model.entity.Type;
import com.linghu.model.excel.FeedExportExcel;
import com.linghu.model.excel.PlatformExcel;
import com.linghu.model.excel.ReferenceExcel;
import com.linghu.model.vo.*;
import com.linghu.model.excel.KeywordExcel;
import com.linghu.model.excel.PlatformExcel;
import com.linghu.model.vo.KeywordStaticsListVO;
@@ -29,6 +36,8 @@

import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.time.LocalDateTime;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
@@ -81,10 +90,21 @@
     */
    @GetMapping("/getResultByTypeId")
    @ApiOperation(value = "根据类别查看")
    public ResponseResult<List<PlatformProportionVO>> getResultByTypeId(@RequestParam("id") Integer keywordId,
    public ResponseResult<ResultByTypeIdVO> getResultByTypeId(@RequestParam("keywordId") Integer keywordId,
            @RequestParam(value = "questionId", required = false) Integer questionId,
            @RequestParam(value = "typeId", required = false) Integer typeId) {
        return keywordService.getResultByTypeId(keywordId, questionId, typeId);
        ResultByTypeIdVO vo = new ResultByTypeIdVO();
        Keyword keyword = keywordService.getById(keywordId);
        List<PlatformProportionVO> nowList = keywordMapper.getResultByTypeId(keywordId, questionId,
                keyword.getNum(), typeId);
        vo.setNowList(nowList);
        if (keyword.getNum()>1){
            List<PlatformProportionVO> beforeList = keywordMapper.getResultByTypeId(keywordId, questionId,
                    keyword.getNum()-1, typeId);
            vo.setBeforeList(beforeList);
        }

        return ResponseResult.success(vo);
    }

    /**
@@ -92,7 +112,7 @@
     */
    @PostMapping(value = "/exportGetResultByTypeId")
    @ApiOperation(value = "导出：根据类别查看")
    public ResponseEntity<byte[]> exportGetResultByTypeId(@RequestParam("id") Integer keywordId,
    public ResponseEntity<byte[]> exportGetResultByTypeId(@RequestParam("keywordId") Integer keywordId,
            @RequestParam(value = "questionId", required = false) Integer questionId,
            @RequestParam(value = "typeId", required = false) Integer typeId,
            @RequestParam(value = "isNow") Integer isNow) {
@@ -117,10 +137,23 @@
     */
    @GetMapping("/getResultByPlatformId")
    @ApiOperation(value = "根据平台查看")
    public ResponseResult<List<ResultListVO>> getResultByPlatformId(@RequestParam("id") Integer keywordId,
    public ResponseResult<ResultByPlatformVO> getResultByPlatformId(@RequestParam("keywordId") Integer keywordId,
            @RequestParam(value = "questionId", required = false) Integer questionId,
            @RequestParam(value = "platformId", required = false) Integer platformId) {
        return keywordService.getResultByPlatformId(keywordId, questionId, platformId);
        ResultByPlatformVO vo = new ResultByPlatformVO();
        Keyword keyword = keywordService.getById(keywordId);

        List<ResultListVO> nowList = keywordMapper.getResultByPlatformId(keywordId, questionId, keyword.getNum(),
                platformId);
        vo.setNowList(nowList);

        if (keyword.getNum()>1){
            List<ResultListVO> beforeList = keywordMapper.getResultByPlatformId(keywordId, questionId, keyword.getNum()-1,
                    platformId);
            vo.setBeforeList(beforeList);
        }

        return ResponseResult.success(vo);
    }

    /**
@@ -152,7 +185,7 @@
     * 查看详情
     */
    @GetMapping("/getResultById")
    @ApiOperation(value = "根据平台查看")
    @ApiOperation(value = "查看")
    public ResponseResult getResultById(@RequestParam("referenceId") Integer referenceId) {
        Reference reference = referenceService.getById(referenceId);
        if (reference == null) {
@@ -160,10 +193,96 @@
        }
        return ResponseResult.success(reference);
    }
    /**
     * 下载模版
     */
    @GetMapping("/download")
    @ApiOperation("下载投喂模板")
    public ResponseEntity<byte[]> downTemplate() throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        EasyExcel.write(out, ReferenceExcel.class).sheet("投喂模板").doWrite(new ArrayList<>());

        return ResponseEntity.ok()
                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=platform_template.xlsx")
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .body(out.toByteArray());
    }
    /**
     * 投喂
     */
    @PostMapping("/importFeed")
    @ApiOperation("投喂")
    public ResponseResult<List<FeedExportExcel>> importTemplate(@RequestParam("file") MultipartFile file) {
        try {
            // 检查文件是否为空
            if (file.isEmpty()) {
                return ResponseResult.error("上传文件不能为空");
            }

            // 读取Excel数据
            List<ReferenceExcel> excelList = EasyExcel.read(file.getInputStream())
                    .head(ReferenceExcel.class)
                    .sheet()
                    .doReadSync();

            // 数据转换与验证
            List<String> errorMessages = new ArrayList<>();
            List<FeedExportExcel> result = new ArrayList<>();
            for (ReferenceExcel excel : excelList) {
                // 检查必要字段
                if (!StringUtils.hasText(excel.getPlatform_name())) {
                    errorMessages.add("平台名称不能为空");
                    continue;
                }
                if (!StringUtils.hasText(excel.getTitle())) {
                    errorMessages.add("标题不能为空");
                    continue;
                }
                if (!StringUtils.hasText(String.valueOf(excel.getCreate_time()))) {
                    errorMessages.add("发布时间不能为空");
                    continue;
                }
                if (!StringUtils.hasText(excel.getUrl())) {
                    errorMessages.add("发布网址不能为空");
                    continue;
                }
                //todo 查询出当前这次的结果 做出对比

            }

            // 处理错误
            if (!errorMessages.isEmpty()) {
                return ResponseResult.error("数据验证失败: " + String.join("; ", errorMessages));
            }

            // 返回信息
            return ResponseResult.success(result);
        } catch (Exception e) {
            // 记录详细异常信息

            return ResponseResult.error("文件解析失败：" + e.getMessage());
        }
    }

    /**
     * 导出投喂结果
     */
    @PostMapping("/exportFeed")
    @ApiOperation(value = "导出投喂结果")
    public ResponseEntity<byte[]> exportGetResultByPlatformId(@RequestBody ExportFeedDTO dto) {
        // 3. 导出Excel
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        EasyExcel.write(out, FeedExportExcel.class)
                .sheet("导出投喂结果")
                .doWrite(dto.getExcels());

        // 4. 构建响应
        return ResponseEntity.ok()
                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=references_export.xlsx")
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .body(out.toByteArray());
    }


    /**
     * 批量新增关键词

 src/main/java/com/linghu/model/dto/ExportFeedDTO.java

New file
@@ -0,0 +1,13 @@
package com.linghu.model.dto;

import com.linghu.model.excel.FeedExportExcel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

@Data
public class ExportFeedDTO {
    @ApiModelProperty("投喂结果")
    private List<FeedExportExcel> excels;
}

 src/main/java/com/linghu/model/excel/FeedExportExcel.java

New file
@@ -0,0 +1,43 @@
package com.linghu.model.excel;

import com.alibaba.excel.annotation.ExcelProperty;
import com.fasterxml.jackson.annotation.JsonFormat;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.time.LocalDateTime;

@Data
public class FeedExportExcel {
    @ExcelProperty("平台名称")
    @ApiModelProperty("平台名称")
    private String platform_name;
    @ApiModelProperty("平台是否匹配 0-否 1-是")
    private Integer platformIs;

    @ExcelProperty("标题")
    @ApiModelProperty("标题")
    private String title;
    @ApiModelProperty("标题是否匹配 0-否 1-是")
    private Integer titleIs;


    @ExcelProperty("发布时间")
    @ApiModelProperty("发布时间")
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    private LocalDateTime create_time;

    @ApiModelProperty("发布时间是否匹配 0-否 1-是")
    private Integer createIs;

    @ExcelProperty("发布网址")
    @ApiModelProperty("发布网址")
    private String url;
    @ApiModelProperty("发布网址是否匹配 0-否 1-是")
    private Integer urlIs;

    @ExcelProperty("收录情况")
    @ApiModelProperty("收录情况")
    private String inclusionStatus;

}

 src/main/java/com/linghu/model/excel/ReferenceExcel.java

New file
@@ -0,0 +1,23 @@
package com.linghu.model.excel;

import com.alibaba.excel.annotation.ExcelProperty;
import com.fasterxml.jackson.annotation.JsonFormat;
import lombok.Data;

import java.time.LocalDateTime;

@Data
public class ReferenceExcel {
    @ExcelProperty("平台名称")
    private String platform_name;

    @ExcelProperty("标题")
    private String title;
    @ExcelProperty("发布时间")
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    private LocalDateTime create_time;

    @ExcelProperty("发布网址")
    private String url;

}

 src/main/java/com/linghu/model/vo/ResultByPlatformVO.java

New file
@@ -0,0 +1,14 @@
package com.linghu.model.vo;

import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

@Data
public class ResultByPlatformVO {
    @ApiModelProperty("当前次结果查询")
    private List<ResultListVO> nowList;
    @ApiModelProperty("后一次结果查询")
    private List<ResultListVO> beforeList;
}

 src/main/java/com/linghu/model/vo/ResultByTypeIdVO.java

New file
@@ -0,0 +1,14 @@
package com.linghu.model.vo;

import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

@Data
public class ResultByTypeIdVO {
    @ApiModelProperty("当前次结果查询")
    private List<PlatformProportionVO> nowList;
    @ApiModelProperty("后一次结果查询")
    private List<PlatformProportionVO> beforeList;
}