From 580d70b15ae47bc180a0b579af8c47c506eefac6 Mon Sep 17 00:00:00 2001
From: huliguo <2023611923@qq.com>
Date: 星期五, 11 七月 2025 20:13:38 +0800
Subject: [PATCH] 修改

---
 src/main/java/com/linghu/controller/OrderController.java |  172 ++++++++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 134 insertions(+), 38 deletions(-)

diff --git a/src/main/java/com/linghu/controller/OrderController.java b/src/main/java/com/linghu/controller/OrderController.java
index ea27ecc..28509fb 100644
--- a/src/main/java/com/linghu/controller/OrderController.java
+++ b/src/main/java/com/linghu/controller/OrderController.java
@@ -4,15 +4,23 @@
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.fasterxml.jackson.databind.util.BeanUtil;
 import com.linghu.model.common.ResponseResult;
-import com.linghu.model.entity.Order;
+import com.linghu.model.entity.Keyword;
+import com.linghu.model.entity.Orders;
+import com.linghu.model.dto.KeywordDto;
 import com.linghu.model.dto.OrderDto;
+import com.linghu.model.page.CustomPage;
+import com.linghu.service.KeywordService;
 import com.linghu.service.OrderService;
 
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.text.SimpleDateFormat;
+import java.time.LocalDateTime;
 import java.util.Date;
 import java.util.List;
 
@@ -20,19 +28,24 @@
  * 订单管理接口
  */
 @RestController
-@RequestMapping("/order")
+@RequestMapping("/orders")
+@Api(value = "订单管理接口", tags = "订单管理")
 public class OrderController {
 
     @Autowired
     private OrderService orderService;
 
+    @Autowired
+    private KeywordService keywordService;
+
     /**
      * 新增订单
      */
     @PostMapping
-    public ResponseResult<Order> add(@RequestBody OrderDto orderDto) {
-        //将dto转entity
-        Order order = new Order();
+    @ApiOperation(value = "新增订单")
+    public ResponseResult<Orders> add(@RequestBody OrderDto orderDto) {
+        // 将dto转entity
+        Orders order = new Orders();
         BeanUtils.copyProperties(orderDto, order);
 
         if (order.getClient_name() == null || order.getClient_name().trim().isEmpty()) {
@@ -44,8 +57,8 @@
         String dateStr = dateFormat.format(new Date());
 
         // 查询当天订单数量
-        LambdaQueryWrapper<Order> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.likeRight(Order::getOrder_id, dateStr);
+        LambdaQueryWrapper<Orders> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.likeRight(Orders::getOrder_id, dateStr);
         long count = orderService.count(queryWrapper);
 
         // 生成订单ID
@@ -55,12 +68,22 @@
         // 设置初始状态
         order.setStatus(1); // 待处理
         order.setDel_flag(0); // 未删除
-        order.setCreate_time(new Date());
-        //保存关键词
-        boolean saveOrderWithKeywords = orderService.saveOrderWithKeywords(orderDto);
-      
+        order.setCreate_time(LocalDateTime.now());
+        boolean save = orderService.save(order);
+        // 保存关键词
+        boolean saveOrderWithKeywords = orderService.saveOrderWithKeywords(orderDto, order.getOrder_id());
+        if (!saveOrderWithKeywords) {
+            return ResponseResult.error("添加关键词失败");
+        }
+        //更新关键词数量
+        LambdaQueryWrapper<Keyword> queryKeywordsQueryWrapper =  new LambdaQueryWrapper<>();
+        queryKeywordsQueryWrapper.eq(Keyword::getOrder_id, order.getOrder_id());
+        int count1 = keywordService.count(queryKeywordsQueryWrapper);
+        order.setKeyword_num(count1);
+        order.setUpdate_time(LocalDateTime.now());
+        orderService.updateById(order);
 
-        if (orderService.save(order)) {
+        if (save) {
             return ResponseResult.success(order);
         }
         return ResponseResult.error("添加订单失败");
@@ -70,41 +93,55 @@
      * 删除订单(逻辑删除)
      */
     @DeleteMapping("/{orderId}")
+    @ApiOperation(value = "删除订单")
     public ResponseResult<Void> delete(@PathVariable String orderId) {
-        Order order = orderService.getById(orderId);
+        Orders order = orderService.getById(orderId);
         if (order == null) {
             return ResponseResult.error("订单不存在");
         }
 
         order.setDel_flag(1);
-        order.setUpdate_time(new Date());
+        order.setUpdate_time(LocalDateTime.now());
+        boolean success = orderService.updateById(order);
 
-        if (orderService.updateById(order)) {
-            return ResponseResult.success();
-        }
-        return ResponseResult.error("删除订单失败");
+        return success ? ResponseResult.success() : ResponseResult.error("删除订单失败");
     }
 
     /**
      * 更新订单
      */
     @PutMapping
-    public ResponseResult<Void> update(@RequestBody Order order) {
-        if (order.getOrder_id() == null) {
+    @ApiOperation(value = "更新订单")
+    public ResponseResult<Void> update(@RequestBody OrderDto orderDto) {
+        if (orderDto.getOrder_id() == null) {
             return ResponseResult.error("订单ID不能为空");
         }
-        if (order.getClient_name() == null || order.getClient_name().trim().isEmpty()) {
+        if (orderDto.getClient_name() == null || orderDto.getClient_name().trim().isEmpty()) {
             return ResponseResult.error("客户名称不能为空");
         }
 
-        Order existingOrder = orderService.getById(order.getOrder_id());
+        Orders existingOrder = orderService.getById(orderDto.getOrder_id());
         if (existingOrder == null) {
             return ResponseResult.error("订单不存在");
         }
 
-        order.setUpdate_time(new Date());
+        orderDto.setUpdate_time(LocalDateTime.now());
+        if (orderDto.getKeywords()!= null&&orderDto.getKeywords().trim().length() > 0){
+            // 保存关键词
+            boolean saveOrderWithKeywords = orderService.saveOrderWithKeywords(orderDto, orderDto.getOrder_id());
+            if (!saveOrderWithKeywords) {
+                return ResponseResult.error("添加关键词失败");
+            }
+        }
+        //更新关键词数量
+        LambdaQueryWrapper<Keyword> queryKeywordsQueryWrapper =  new LambdaQueryWrapper<>();
+        queryKeywordsQueryWrapper.eq(Keyword::getOrder_id, orderDto.getOrder_id());
+        int count1 = keywordService.count(queryKeywordsQueryWrapper);
+        orderDto.setKeyword_num(count1);
 
-        if (orderService.updateById(order)) {
+
+
+        if (orderService.updateById(orderDto)) {
             return ResponseResult.success();
         }
         return ResponseResult.error("更新订单失败");
@@ -114,8 +151,9 @@
      * 根据ID查询订单
      */
     @GetMapping("/{orderId}")
-    public ResponseResult<Order> getById(@PathVariable String orderId) {
-        Order order = orderService.getById(orderId);
+    @ApiOperation("根据ID查询订单")
+    public ResponseResult<Orders> getById(@PathVariable String orderId) {
+        Orders order = orderService.getById(orderId);
         if (order == null || order.getDel_flag() == 1) {
             return ResponseResult.error("订单不存在");
         }
@@ -126,32 +164,90 @@
      * 查询订单列表
      */
     @GetMapping
-    public ResponseResult<List<Order>> list(
+    @ApiOperation("查询订单列表")
+    public ResponseResult<CustomPage<Orders>> list(
             @RequestParam(required = false) Integer pageNum,
             @RequestParam(required = false) Integer pageSize,
             @RequestParam(required = false) String clientName,
-            @RequestParam(required = false) Integer status) {
+            @RequestParam(required = false) Integer status,
+            @RequestParam(required = false) String timeRange) {
 
-        LambdaQueryWrapper<Order> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(Order::getDel_flag, 0); // 只查询未删除的订单
+        LambdaQueryWrapper<Orders> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(Orders::getDel_flag, 0);
 
         // 添加查询条件
         if (clientName != null && !clientName.trim().isEmpty()) {
-            queryWrapper.like(Order::getClient_name, clientName);
+            queryWrapper.like(Orders::getClient_name, clientName);
         }
         if (status != null) {
-            queryWrapper.eq(Order::getStatus, status);
+            queryWrapper.eq(Orders::getStatus, status);
         }
+        // 改造时间筛选逻辑
+        if (timeRange != null && !timeRange.trim().isEmpty()) {
+            LocalDateTime now = LocalDateTime.now();
+            LocalDateTime startTime = null;
 
+            switch (timeRange.trim()) {
+                case "week":
+                    startTime = now.minusWeeks(1);
+                    break;
+                case "month":
+                    startTime = now.minusMonths(1);
+                    break;
+                case "threeMonths":
+                    startTime = now.minusMonths(3);
+                    break;
+                case "year":
+                    startTime = now.minusYears(1);
+                    break;
+                default:
+                    // 可添加日志记录无效参数
+                    break;
+            }
+            if (startTime != null) {
+                queryWrapper.ge(Orders::getCreate_time, startTime);
+            }
+        }
+        // 排序
+        queryWrapper.orderByDesc(Orders::getCreate_time);
         // 分页查询
         if (pageNum != null && pageSize != null) {
-            Page<Order> pageInfo = new Page<>(pageNum, pageSize);
-            Page<Order> result = orderService.page(pageInfo, queryWrapper);
-            return ResponseResult.success(result.getRecords());
+            Page<Orders> pageInfo = new Page<>(pageNum, pageSize);
+            Page<Orders> result = orderService.page(pageInfo, queryWrapper);
+            return ResponseResult.success(new CustomPage<>(result));
         }
 
-        // 不分页
-        List<Order> list = orderService.list(queryWrapper);
-        return ResponseResult.success(list);
+        // 不分页查询:手动创建 CustomPage
+        List<Orders> list = orderService.list(queryWrapper);
+        CustomPage<Orders> page = new CustomPage<>(new Page<>());
+        page.setRecords(list);
+        page.setTotal(list.size());
+
+        return ResponseResult.success(page);
+    }
+
+    /**
+     * 获取客户列表
+     * @param
+     * @return
+     */
+    @GetMapping("/clientList")
+    @ApiOperation("获取客户列表")
+    public ResponseResult<CustomPage<String>> getClientList(@RequestParam(required = false) String clientName,
+                                                      @RequestParam(required = false,defaultValue = "1") Integer pageNum,
+                                                      @RequestParam(required = false, defaultValue = "10") Integer pageSize) {
+
+        Page<String> result = orderService.getClientList(clientName,pageNum, pageSize);
+
+
+        return ResponseResult.success(new CustomPage<>( result));
+    }
+
+
+    @GetMapping("/{orderId}/keywordList")
+    @ApiOperation("获取订单关联的关键词及提问词")
+    public ResponseResult<List<KeywordDto>> getKeywordList(@PathVariable String orderId){
+        List<KeywordDto> result = orderService.getKeywordListByOrderId(orderId);
+        return ResponseResult.success(result);
     }
 }
\ No newline at end of file

--
Gitblit v1.7.1