package com.linghu.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; 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.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; /** * 订单管理接口 */ @RestController @RequestMapping("/orders") @Api(value = "订单管理接口", tags = "订单管理") public class OrderController { @Autowired private OrderService orderService; @Autowired private KeywordService keywordService; /** * 新增订单 */ @PostMapping @ApiOperation(value = "新增订单") public ResponseResult add(@RequestBody OrderDto orderDto) { // 将dto转entity Orders order = new Orders(); BeanUtils.copyProperties(orderDto, order); if (order.getClient_name() == null || order.getClient_name().trim().isEmpty()) { return ResponseResult.error("客户名称不能为空"); } // 生成订单ID:日期+当天的订单数量(如:202507060001) SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMdd"); String dateStr = dateFormat.format(new Date()); // 查询当天订单数量 LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); queryWrapper.likeRight(Orders::getOrder_id, dateStr); long count = orderService.count(queryWrapper); // 生成订单ID String orderId = String.format("%s%04d", dateStr, count + 1); order.setOrder_id(orderId); // 设置初始状态 order.setStatus(1); // 待处理 order.setDel_flag(0); // 未删除 order.setCreate_time(LocalDateTime.now()); boolean save = orderService.save(order); // 保存关键词 boolean saveOrderWithKeywords = orderService.saveOrderWithKeywords(orderDto, order.getOrder_id()); if (!saveOrderWithKeywords) { return ResponseResult.error("添加关键词失败"); } //更新关键词数量 LambdaQueryWrapper 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 (save) { return ResponseResult.success(order); } return ResponseResult.error("添加订单失败"); } /** * 删除订单(逻辑删除) */ @DeleteMapping("/{orderId}") @ApiOperation(value = "删除订单") public ResponseResult delete(@PathVariable String orderId) { Orders order = orderService.getById(orderId); if (order == null) { return ResponseResult.error("订单不存在"); } order.setDel_flag(1); order.setUpdate_time(LocalDateTime.now()); boolean success = orderService.updateById(order); return success ? ResponseResult.success() : ResponseResult.error("删除订单失败"); } /** * 更新订单 */ @PutMapping @ApiOperation(value = "更新订单") public ResponseResult update(@RequestBody OrderDto orderDto) { if (orderDto.getOrder_id() == null) { return ResponseResult.error("订单ID不能为空"); } if (orderDto.getClient_name() == null || orderDto.getClient_name().trim().isEmpty()) { return ResponseResult.error("客户名称不能为空"); } Orders existingOrder = orderService.getById(orderDto.getOrder_id()); if (existingOrder == null) { return ResponseResult.error("订单不存在"); } 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 queryKeywordsQueryWrapper = new LambdaQueryWrapper<>(); queryKeywordsQueryWrapper.eq(Keyword::getOrder_id, orderDto.getOrder_id()); int count1 = keywordService.count(queryKeywordsQueryWrapper); orderDto.setKeyword_num(count1); if (orderService.updateById(orderDto)) { return ResponseResult.success(); } return ResponseResult.error("更新订单失败"); } /** * 根据ID查询订单 */ @GetMapping("/{orderId}") @ApiOperation("根据ID查询订单") public ResponseResult getById(@PathVariable String orderId) { Orders order = orderService.getById(orderId); if (order == null || order.getDel_flag() == 1) { return ResponseResult.error("订单不存在"); } return ResponseResult.success(order); } /** * 查询订单列表 */ @GetMapping @ApiOperation("查询订单列表") public ResponseResult> list( @RequestParam(required = false) Integer pageNum, @RequestParam(required = false) Integer pageSize, @RequestParam(required = false) String clientName, @RequestParam(required = false) Integer status, @RequestParam(required = false) String timeRange) { LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); queryWrapper.eq(Orders::getDel_flag, 0); // 添加查询条件 if (clientName != null && !clientName.trim().isEmpty()) { queryWrapper.like(Orders::getClient_name, clientName); } if (status != null) { 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); } } // 分页查询 if (pageNum != null && pageSize != null) { Page pageInfo = new Page<>(pageNum, pageSize); Page result = orderService.page(pageInfo, queryWrapper); return ResponseResult.success(new CustomPage<>(result)); } // 不分页查询:手动创建 CustomPage List list = orderService.list(queryWrapper); CustomPage page = new CustomPage<>(new Page<>()); page.setRecords(list); page.setTotal(list.size()); return ResponseResult.success(page); } /** * 获取客户列表 * @param * @return */ @GetMapping("/clientList") @ApiOperation("获取客户列表") public ResponseResult> getClientList(@RequestParam(required = false) String clientName, @RequestParam(required = false,defaultValue = "1") Integer pageNum, @RequestParam(required = false, defaultValue = "10") Integer pageSize) { Page result = orderService.getClientList(clientName,pageNum, pageSize); return ResponseResult.success(new CustomPage<>( result)); } @GetMapping("/{orderId}/keywordList") @ApiOperation("获取订单关联的关键词及提问词") public ResponseResult> getKeywordList(@PathVariable String orderId){ List result = orderService.getKeywordListByOrderId(orderId); return ResponseResult.success(result); } }