修改

guyue

2 天以前 9a34ef1f3c2aea28705939eb1dc32c28f1222ad9

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

@@ -226,6 +226,7 @@
               }, 1) // 限制并发数为1，确保顺序执行
               .collectList() // 收集所有批次的响应
               .flatMap(responses ->

                       saveKeywordTasks(keywordId, responses) // 保存关联关系
                               .thenReturn(responses) // 返回原始响应
               );
@@ -272,11 +273,12 @@
                .bodyToMono(new ParameterizedTypeReference<SearchTaskResponse>() {})
                .flatMap(taskResponse -> {
                    if (taskResponse != null && taskResponse.getTask_id() != null) {

                        // 使用 Reactor 的方式更新数据库
                        return Mono.fromRunnable(() -> {
                                    LambdaUpdateWrapper<Keyword> updateWrapper = new LambdaUpdateWrapper<>();
                                    updateWrapper.eq(Keyword::getKeyword_id, batchRequest.getKeyword_id());
                                    updateWrapper.set(Keyword::getStatus, "Submitted");
                                    updateWrapper.set(Keyword::getStatus, "submitted");
                                    updateWrapper.set(Keyword::getTask_id, taskResponse.getTask_id());
                                    keywordService.update(updateWrapper);
                                }).subscribeOn(Schedulers.boundedElastic()) // 在弹性线程池执行
@@ -342,9 +344,9 @@
                .map(data -> ResponseResult.success(data))
                .onErrorResume(e -> {
                    if (e.getMessage().contains("任务不存在")) {
                        return Mono.just(ResponseResult.error(404, "任务不存在"));
                        return Mono.just(ResponseResult.error(200, "任务不存在"));
                    } else if (e.getMessage().contains("无法取消")) {
                        return Mono.just(ResponseResult.error(400, "任务已完成，无法取消"));
                        return Mono.just(ResponseResult.error(200, "任务已完成，无法取消"));
                    }
                    return Mono.just(ResponseResult.error(500, "取消任务失败: " + e.getMessage()));
                });
@@ -453,20 +455,12 @@
    private Mono<Void> updateQuestionAndReference(TaskResultResponse result) {
        return Mono.fromRunnable(() -> {
            try {
                // 1. 更新关键词状态
                LambdaUpdateWrapper<Keyword> keywordUpdate = new LambdaUpdateWrapper<>();
                keywordUpdate.eq(Keyword::getTask_id, result.getTask_id())
                        .set(Keyword::getStatus, "completed");
                keywordService.update(keywordUpdate);

                // 1. 根据KeywordTask更新关键词状态
                // 查询关键词ID

                LambdaQueryWrapper<KeywordTask> keywordTaskWrapper = new LambdaQueryWrapper<>();
                keywordTaskWrapper.eq(KeywordTask::getTask_id, result.getTask_id());
                KeywordTask keywordTask = keywordTaskService.getOne(keywordTaskWrapper);

//                LambdaQueryWrapper<Keyword> keywordQuery = new LambdaQueryWrapper<>();
//                keywordQuery.eq(Keyword::getTask_id, keywordTask.getTask_id());
                Keyword keyword = keywordService.getById(keywordTask.getKeyword_id());

                if (keyword == null) {
@@ -475,6 +469,8 @@
                    throw new Exception("未找到关联的关键词，task_id: " + result.getTask_id());
//                    return;
                }
                keyword.setStatus("completed");
                keywordService.updateById(keyword);

                // 2. 批量查询所有问题
                LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
@@ -536,6 +532,8 @@
                                                    reference.setUrl(ref.getUrl());
                                                    reference.setDomain(ref.getDomain());
                                                    reference.setNum(finalMaxNumByKeywordId);
                                                    reference.setTask_id(result.getTask_id());
                                                    reference.setKeyword_id(keyword.getKeyword_id());
                                                    //域名和平台id映射
                                                    reference.setCreate_time(LocalDateTime.now());
                                                    Platform platform = platformService.getPlatformByDomain(reference.getDomain());
@@ -621,6 +619,7 @@
    }

    @GetMapping("/health")
    @ApiOperation("健康检查")
    public Mono<HealthResponse> checkThirdPartyHealth() {
        return webClient.get()
                .uri(baseUrl + "/health") // 假设第三方健康检查接口路径为/health
@@ -634,6 +633,7 @@
     * 查询服务器资源
     */
    @GetMapping("/server/resource")
    @ApiOperation(value = "查询服务器资源")
    public Mono<ServerResourceResponse> getServerResource() {
        return webClient.get()
                .uri(baseUrl + "/api/v1/system/resources")