diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/AskQuestionsController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/AskQuestionsController.java index 9a5ecd01..6ef0837a 100644 --- a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/AskQuestionsController.java +++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/AskQuestionsController.java @@ -2,12 +2,11 @@ package com.adc.da.slrs.StandardQA.controller; import com.adc.da.base.web.BaseController; import com.adc.da.slrs.StandardQA.dao.AskQuestionsDao; -import com.adc.da.slrs.StandardQA.entity.AskQuestions; -import com.adc.da.slrs.StandardQA.entity.AskQuestionsPage; -import com.adc.da.slrs.StandardQA.entity.AskUser; -import com.adc.da.slrs.StandardQA.entity.Reply; +import com.adc.da.slrs.StandardQA.entity.*; import com.adc.da.slrs.StandardQA.service.Impl.AskQuestionsServiceImpl; import com.adc.da.slrs.StandardQA.service.Impl.AskUserServiceImpl; +import com.adc.da.slrs.StandardQA.service.Impl.SubscriptionServiceImpl; +import com.adc.da.util.LoginUserUtil; import com.adc.da.util.UUIDUtils; import com.adc.da.util.http.ResponseMessage; import com.adc.da.util.http.Result; @@ -45,6 +44,9 @@ public class AskQuestionsController extends BaseController { @Autowired private AskUserServiceImpl askUserService; + @Autowired + private SubscriptionServiceImpl subscriptionService; + @GetMapping("/getAskQuestionsPage") public ResponseMessage> getAskQuestionsPage(AskQuestionsPage askQuestions){ IPage iPage = new Page<>(); @@ -58,7 +60,19 @@ public class AskQuestionsController extends BaseController { @PostMapping("/list") public ResponseMessage> getAskQuestionslistPage(@RequestBody AskQuestionsPage askQuestions){ - return Result.success(askQuestionsService.listPage(askQuestions)); + IPage iPage = askQuestionsService.listPage(askQuestions); + if(!iPage.getRecords().isEmpty()){ + iPage.getRecords().forEach(questions -> { + QueryWrapper queryWrapper = new QueryWrapper(); + queryWrapper.eq("ASK_ID",questions.getId()); + queryWrapper.eq("USID", LoginUserUtil.getUserId()); + List list = subscriptionService.list(queryWrapper); + if(!list.isEmpty()){ + questions.setSubState(list.get(0).getSubState()); + } + }); + } + return Result.success(iPage); } /** diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/SubscriptionController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/SubscriptionController.java index f8a6ec16..f300887b 100644 --- a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/SubscriptionController.java +++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/controller/SubscriptionController.java @@ -7,6 +7,7 @@ import com.adc.da.slrs.StandardQA.service.Impl.SubscriptionServiceImpl; import com.adc.da.util.http.ResponseMessage; import com.adc.da.util.http.Result; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import io.swagger.annotations.Api; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PostMapping; @@ -14,6 +15,8 @@ import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; +import java.util.List; + /** * @Author: qk * @Date: 2022/6/7 0007 11:18 @@ -31,14 +34,29 @@ public class SubscriptionController extends BaseController { QueryWrapper queryWrapper = new QueryWrapper(); queryWrapper.eq("ASK_ID",subscription.getAskId()); queryWrapper.eq("USID",subscription.getUsId()); - int number = subscriptionService.count(queryWrapper); - if(number>0){ - return Result.success("200","此问题已订阅",null); - } - if(subscriptionService.save(subscription)){ - return Result.success("200","订阅成功",null); + List list = subscriptionService.list(queryWrapper); + if(!list.isEmpty()){ + UpdateWrapper updateWrapper = new UpdateWrapper(); + updateWrapper.eq("ASK_ID",subscription.getAskId()); + updateWrapper.eq("USID",subscription.getUsId()); + updateWrapper.set("SUB_STATE",subscription.getSubState()); + if(subscriptionService.update(updateWrapper)){ + if("y".equals(subscription.getSubState())){ + return Result.success("200","订阅成功",null); + }else { + return Result.success("200","取消订阅成功",null); + } + + }else { + return Result.success("-1","订阅失败",null); + } }else { - return Result.success("-1","订阅失败",null); + subscription.setSubState("y"); + if(subscriptionService.save(subscription)){ + return Result.success("200","订阅成功",null); + }else { + return Result.success("-1","订阅失败",null); + } } } } diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/AskQuestions.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/AskQuestions.java index 84cf8afc..0f8c9755 100644 --- a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/AskQuestions.java +++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/AskQuestions.java @@ -101,4 +101,7 @@ public class AskQuestions extends BaseEntity { @TableField(exist = false) private String subscription; + @TableField(exist = false) + private String subState; + } diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/Subscription.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/Subscription.java index 47c1169e..5e46f3bf 100644 --- a/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/Subscription.java +++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/StandardQA/entity/Subscription.java @@ -28,4 +28,8 @@ public class Subscription extends BaseEntity { @TableField("USID") @ApiModelProperty(value = "用户ID") private String usId; + + @TableField("SUB_STATE") + @ApiModelProperty(value = "订阅状态 默认 n 未订阅 y 已订阅") + private String subState; } diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarStandProjectLibrary/controller/SarStandProjectLibraryController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarStandProjectLibrary/controller/SarStandProjectLibraryController.java index 1abb263f..00375a93 100644 --- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarStandProjectLibrary/controller/SarStandProjectLibraryController.java +++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarStandProjectLibrary/controller/SarStandProjectLibraryController.java @@ -74,7 +74,7 @@ public class SarStandProjectLibraryController extends BaseController MaintenanceList = sarStandProjectLibraryService.queryMaintenanceProjectWorkFlow(Page,PageSize,sarDto); Map map = new HashMap(); map.put("Maintenance",MaintenanceList);