From ec6d43aa07ee0e8faf34498057ebcfbb446aa015 Mon Sep 17 00:00:00 2001
From: hjl <1657978663@qq.com>
Date: 星期二, 16 七月 2024 09:21:51 +0800
Subject: [PATCH] feat: 代码重构

---
 ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java |  192 +++++++++++++++++++++++++----------------------
 1 files changed, 103 insertions(+), 89 deletions(-)

diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java
index ebbe530..6a2ed6a 100644
--- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java
+++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java
@@ -1,31 +1,33 @@
 package com.ruoyi.admin.controller;
 
 
-import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import com.ruoyi.admin.entity.*;
-import com.ruoyi.admin.request.OrderQueryRequest;
+import com.ruoyi.admin.entity.MasterWorker;
+import com.ruoyi.admin.entity.RecoveryServe;
+import com.ruoyi.admin.entity.Site;
 import com.ruoyi.admin.service.*;
 import com.ruoyi.admin.sorcket.WebSocketServer;
-import com.ruoyi.admin.vo.OrderCountVO;
 import com.ruoyi.admin.vo.OrderDetailVO;
-import com.ruoyi.admin.vo.OrderPageCountVO;
 import com.ruoyi.common.core.constant.Constants;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.utils.SnowflakeIdWorker;
+import com.ruoyi.common.security.service.TokenService;
+import com.ruoyi.order.api.entity.*;
+import com.ruoyi.order.api.feignClient.ExchangeDispatchClient;
+import com.ruoyi.order.api.feignClient.OrderClient;
+import com.ruoyi.system.api.model.LoginUser;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.math.BigDecimal;
-import java.util.Arrays;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -46,32 +48,23 @@
     @Resource
     private MasterWorkerService masterWorkerService;
     @Resource
-    private SiteService siteService;
-    @Resource
     private RecoveryServeService recoveryServeService;
     @Resource
-    private RecoveryServePriceService recoveryServePriceService;
+    private SiteService siteService;
     @Resource
-    private ChangeDispatchService changeDispatchService;
+    private OrderClient orderClient;
+    @Resource
+    private ExchangeDispatchClient dispatchClient;
+    @Resource
+    private TokenService tokenService;
+    @Resource
+    private FranchiseeService franchiseeService;
 
     /**
      * 雪花算法类
      */
     private static final SnowflakeIdWorker SNOW_FLAKE_ID_WORKER = new SnowflakeIdWorker(5, 5);
 
-    /**
-     * socket测试
-     */
-    @ApiOperation(value = "socket测试", tags = {"后台-订单管理"})
-    @GetMapping(value = "/socketTest")
-    public R<String> socketTest(@RequestParam String msg, @RequestParam String id) {
-        try {
-            WebSocketServer.sendInfo(msg, id);
-            return R.ok("消息推送成功!");
-        } catch (IOException e) {
-            return R.fail();
-        }
-    }
 
     /**
      * 根据前台用户id查询所有订单信息
@@ -84,7 +77,7 @@
             @ApiImplicitParam(value = "手机号", name = "phone", dataType = "String", required = true)
     })
     public R<List<Order>> queryList(@RequestParam("phone") String phone) {
-        return R.ok(orderService.lambdaQuery().eq(Order::getReservationPhone, phone).eq(Order::getIsDelete, 0).list());
+        return R.ok(orderClient.queryList(phone).getData());
     }
 
     /**
@@ -99,7 +92,8 @@
     @ApiImplicitParams({
             @ApiImplicitParam(value = "订单id", name = "id", dataType = "Integer", required = true)
     })
-    public R<OrderDetailVO> detail(@RequestParam String id) {
+    public R<OrderDetailVO> orderDetail(@RequestParam String id) {
+        // 独立orderService
         return R.ok(orderService.orderListDetail(id));
     }
 
@@ -110,8 +104,16 @@
      */
     @ApiOperation(value = "订单列表-分页", tags = {"后台-订单管理"})
     @PostMapping(value = "/queryPage")
-    public R<IPage<Order>> queryPage(@RequestBody OrderQueryRequest orderQueryRequest) {
-        return R.ok(orderService.queryPage(orderQueryRequest));
+    public R<Page<Order>> queryPage(@RequestBody OrderQueryRequest orderQueryRequest) {
+        LoginUser loginUser = tokenService.getLoginUser();
+        if (null == loginUser) {
+            return R.loginExpire("登录失效!");
+        }
+        List<String> userCity = new ArrayList<>();
+        if (loginUser.getIsFranchisee()) {
+            userCity = loginUser.getCityList();
+        }
+        return R.ok(orderClient.queryPage(orderQueryRequest).getData());
     }
 
     /**
@@ -120,21 +122,21 @@
     @ApiOperation(value = "订单列表-各订单数量统计", tags = {"后台-订单管理"})
     @PostMapping(value = "/orderPageCount")
     public R<OrderPageCountVO> orderPageCount(@RequestBody OrderQueryRequest orderQueryRequest) {
-        return R.ok(orderService.orderPageCount(orderQueryRequest));
+        return R.ok(orderClient.orderPageCount(orderQueryRequest).getData());
     }
 
     /**
      * 站点详情
      *
-     * @param id 站点id
+     * @param id 订单id
      */
     @ApiOperation(value = "订单列表-订单详情", tags = {"后台-订单管理"})
     @GetMapping(value = "/detail")
     @ApiImplicitParams({
-            @ApiImplicitParam(value = "站点id", name = "id", dataType = "Integer", required = true)
+            @ApiImplicitParam(value = "订单id", name = "id", dataType = "Integer", required = true)
     })
-    public R<Order> detail(@RequestParam Integer id) {
-        return R.ok(orderService.getById(id));
+    public R<Order> detail(@RequestParam String id) {
+        return R.ok(orderClient.detail(id).getData());
     }
 
     /**
@@ -145,42 +147,46 @@
      */
     @ApiOperation(value = "订单列表-新增订单", tags = {"后台-订单管理"})
     @PostMapping(value = "/save")
-    public R<String> save(@RequestBody Order order) {
+    public R<String> save(@RequestBody OrderRequest order) {
         // 站点信息
         Site site = siteService.lambdaQuery()
                 .eq(Site::getId, order.getSiteId())
                 .eq(Site::getIsDelete, 0).one();
         order.setSiteName(site.getSiteName());
+        // 师傅信息
+        if (null != order.getServerId()) {
+            MasterWorker masterWorker = masterWorkerService.lambdaQuery()
+                    .eq(MasterWorker::getId, order.getServerId())
+                    .eq(MasterWorker::getIsDelete, 0).one();
+            order.setServerName(masterWorker.getRealName());
+            order.setServerPhone(masterWorker.getPhone());
+            // 待上门
+            order.setState(Constants.ONE);
+        } else {
+            // 待派单状态
+            order.setState(Constants.ZERO);
+        }
+        // 后台订单
+        order.setType(Constants.ONE);
+        order.setSubsidy(BigDecimal.ZERO);
+        order.setOrderNumber(String.valueOf(SNOW_FLAKE_ID_WORKER.nextId()));
         // 回收服务信息
         RecoveryServe recoveryServe = recoveryServeService.lambdaQuery()
                 .eq(RecoveryServe::getId, order.getServeId())
                 .eq(RecoveryServe::getIsDelete, 0).one();
         order.setServeName(recoveryServe.getServeName());
-        // 不同城市会有不同的回收价格
-        RecoveryServePrice price = recoveryServePriceService.lambdaQuery()
-                .eq(RecoveryServePrice::getRecoveryServeId, recoveryServe.getId())
-                .eq(RecoveryServePrice::getIsDelete, 0)
-                .eq(RecoveryServePrice::getCity, order.getCity()).one();
-        if (null == price) {
-            order.setServePrice(recoveryServe.getDefaultPrice());
-            order.setOrderMoney(recoveryServe.getDefaultPrice());
-        } else {
-            order.setServePrice(price.getRecoveryPrice());
-            order.setOrderMoney(price.getRecoveryPrice());
+        order.setServePrice(recoveryServe.getDefaultPrice());
+        order.setOrderMoney(recoveryServe.getDefaultPrice());
+        Boolean data = orderClient.save(order).getData();
+        if (null == data) {
+            return R.fail(orderClient.save(order).getMsg());
         }
-        // 师傅信息
-        MasterWorker masterWorker = masterWorkerService.lambdaQuery()
-                .eq(MasterWorker::getId, order.getServerId())
-                .eq(MasterWorker::getIsDelete, 0).one();
-        order.setServerName(masterWorker.getRealName());
-        order.setServerPhone(masterWorker.getPhone());
-        // 后台订单
-        order.setType(Constants.ONE);
-        // 待派单状态
-        order.setState(Constants.ONE);
-        order.setSubsidy(BigDecimal.ZERO);
-        order.setOrderNumber(String.valueOf(SNOW_FLAKE_ID_WORKER.nextId()));
-        return orderService.save(order) ? R.ok() : R.fail();
+        try {
+            WebSocketServer.sendInfo("您有一条新的订单,请注意查收!", String.valueOf(order.getServerId()));
+        } catch (IOException e) {
+            return R.fail("订单推送失败!");
+        }
+        return data ? R.ok() : R.fail();
     }
 
     /**
@@ -196,29 +202,20 @@
             @ApiImplicitParam(value = "服务人员id", name = "workerId", dataType = "Integer", required = true),
             @ApiImplicitParam(value = "改派原因", name = "applyReason", dataType = "String")
     })
-    public R<String> reassignment(@RequestParam Integer type, @RequestParam Integer orderId,
+    @Transactional(rollbackFor = Exception.class)
+    public R<String> reassignment(@RequestParam Integer type, @RequestParam String orderId,
                                   @RequestParam Integer workerId, String applyReason) {
-        Order order = orderService.lambdaQuery().eq(Order::getId, orderId)
-                .eq(Order::getIsDelete, 0).one();
         MasterWorker masterWorker = masterWorkerService.lambdaQuery()
                 .eq(MasterWorker::getId, workerId)
                 .eq(MasterWorker::getIsDelete, 0).one();
+        Order order = orderClient.exchangeOrder(type, orderId, workerId,
+                masterWorker.getRealName(), masterWorker.getPhone()).getData();
         // 订单派单
         boolean result = true;
-        if (Constants.ONE.equals(type)) {
-            order.setServerId(workerId);
-            order.setServerName(masterWorker.getRealName());
-            order.setServerPhone(masterWorker.getPhone());
-            order.setAcceptTime(new Date());
-            order.setState(Constants.ONE);
-            result = orderService.updateById(order);
-        } else if (Constants.TWO.equals(type)) {
-            order.setServerId(workerId);
-            // 更新接单时间
-            order.setAcceptTime(new Date());
-            order.setServerName(masterWorker.getRealName());
-            order.setServerPhone(masterWorker.getPhone());
-            result = orderService.updateById(order);
+        if (Constants.TWO.equals(type)) {
+            if (order.getState().equals(Constants.SIX) || order.getState().equals(Constants.THREE)) {
+                orderClient.updateState(order.getId(), Constants.ONE);
+            }
             // 生成改派信息
             ChangeDispatch changeDispatch = new ChangeDispatch();
             changeDispatch.setWorkerId(order.getServerId());
@@ -226,14 +223,14 @@
             changeDispatch.setApplyReason(applyReason);
             changeDispatch.setApplyTime(new Date());
             changeDispatch.setState(Constants.ONE);
-            changeDispatch.setOrderId(String.valueOf(order.getId()));
+            changeDispatch.setOrderId(order.getId());
             changeDispatch.setOrderNumber(order.getOrderNumber());
             if (null != order.getUserId()) {
-                changeDispatch.setUserId(String.valueOf(order.getUserId()));
+                changeDispatch.setUserId(order.getUserId());
             }
             changeDispatch.setUserName(order.getReservationName());
             changeDispatch.setIsDelete(Constants.ZERO);
-            changeDispatchService.save(changeDispatch);
+            result = dispatchClient.saveRecord(changeDispatch).getData();
         }
         try {
             WebSocketServer.sendInfo("您有一条新的订单,请注意查收!", String.valueOf(workerId));
@@ -253,11 +250,8 @@
     @ApiImplicitParams({
             @ApiImplicitParam(value = "多个id ',' 拼接", name = "ids", dataType = "String", required = true)
     })
-    public R<String> batchDelete(@RequestParam String ids) {
-        List<String> idList = Arrays.stream(ids.split(",")).collect(Collectors.toList());
-        List<Order> list = orderService.lambdaQuery().in(Order::getId, idList).list();
-        list.forEach(data -> data.setIsDelete(1));
-        return orderService.updateBatchById(list) ? R.ok() : R.fail();
+    public R<Boolean> batchDelete(@RequestParam String ids) {
+        return R.ok(orderClient.batchDelete(ids).getData());
     }
 
     /**
@@ -268,7 +262,9 @@
     @ApiOperation(value = "订单列表-excel导出", tags = {"后台-订单管理"})
     @PostMapping(value = "/excelExport")
     public R<String> excelExport(@RequestBody OrderQueryRequest orderQueryRequest, HttpServletResponse response) {
-        return orderService.excelExport(orderQueryRequest, response);
+        R<List<Order>> result = orderClient.excelExport(orderQueryRequest);
+        // 独立orderService
+        return orderService.excelExport(result.getData(), response);
     }
 
     /**
@@ -285,10 +281,28 @@
             @ApiImplicitParam(value = "页码", name = "pageNum", dataType = "Integer", required = true),
             @ApiImplicitParam(value = "每页条数", name = "pageSize", dataType = "Integer", required = true)
     })
-    public R<IPage<OrderCountVO>> orderCount(String name, String phone,
-                                             @RequestParam(name = "pageNum", defaultValue = "1") Integer pageNum,
-                                             @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize) {
-        return R.ok(orderService.orderCount(name, phone, Page.of(pageNum, pageSize)));
+    public R<Page<OrderCountVO>> orderCount(String name, String phone,
+                                            @RequestParam(name = "pageNum", defaultValue = "1") Integer pageNum,
+                                            @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize) {
+        Page<OrderCountVO> data = orderClient.orderCount(name, phone, pageNum, pageSize).getData();
+        List<Integer> idList = data.getRecords().stream().map(OrderCountVO::getWorkerId)
+                .collect(Collectors.toList());
+        List<MasterWorker> masterWorkerList;
+        if (!idList.isEmpty()) {
+            masterWorkerList = masterWorkerService.lambdaQuery()
+                    .in(MasterWorker::getId, idList)
+                    .eq(MasterWorker::getIsDelete, 0).list();
+        } else {
+            masterWorkerList = masterWorkerService.lambdaQuery()
+                    .eq(MasterWorker::getIsDelete, 0).list();
+        }
+        Map<Integer, String> map = masterWorkerList.stream().collect(Collectors.toMap(MasterWorker::getId,
+                mw -> Optional.ofNullable(mw.getProfilePicture()).orElse("")));
+        for (OrderCountVO record : data.getRecords()) {
+            Integer workerId = record.getWorkerId();
+            record.setProfilePicture(map.get(workerId));
+        }
+        return R.ok(data);
     }
 
 }

--
Gitblit v1.7.1