From 562699fa6d0c279fe0f4f81ce87c336a34a3fb91 Mon Sep 17 00:00:00 2001
From: hjl <1657978663@qq.com>
Date: 星期二, 23 七月 2024 14:32:23 +0800
Subject: [PATCH] feat: 数据迁移;netty通信;bug修改

---
 ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java |   78 ++++++++++++++++++++++++++++++++-------
 1 files changed, 64 insertions(+), 14 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 6045314..2181b11 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
@@ -5,17 +5,25 @@
 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.netty.NettyChannelMap;
+import com.ruoyi.admin.netty.NettyWebSocketController;
+import com.ruoyi.admin.service.MasterWorkerService;
+import com.ruoyi.admin.service.OrderService;
+import com.ruoyi.admin.service.RecoveryServeService;
+import com.ruoyi.admin.service.SiteService;
 import com.ruoyi.admin.vo.OrderDetailVO;
 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.annotation.Logical;
+import com.ruoyi.common.security.annotation.RequiresPermissions;
 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.order.api.request.OrderCountDataRequest;
 import com.ruoyi.system.api.model.LoginUser;
+import io.netty.channel.ChannelHandlerContext;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
@@ -25,7 +33,6 @@
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
-import java.io.IOException;
 import java.math.BigDecimal;
 import java.util.*;
 import java.util.stream.Collectors;
@@ -57,8 +64,6 @@
     private ExchangeDispatchClient dispatchClient;
     @Resource
     private TokenService tokenService;
-    @Resource
-    private FranchiseeService franchiseeService;
 
     /**
      * 雪花算法类
@@ -87,6 +92,7 @@
      *
      * @param id 订单id
      */
+    @RequiresPermissions(value = {"user_detail", "order_detail"}, logical = Logical.OR)
     @ApiOperation(value = "订单列表-查询订单详情(包含服务信息、师傅信息、服务记录、订单评价)", tags = {"后台-订单管理"})
     @GetMapping(value = "/orderDetail")
     @ApiImplicitParams({
@@ -102,6 +108,7 @@
      *
      * @param orderQueryRequest 订单列表查询参数
      */
+    @RequiresPermissions("order_list")
     @ApiOperation(value = "订单列表-分页", tags = {"后台-订单管理"})
     @PostMapping(value = "/queryPage")
     public R<Page<Order>> queryPage(@RequestBody OrderQueryRequest orderQueryRequest) {
@@ -132,6 +139,7 @@
     /**
      * 订单列表
      */
+    @RequiresPermissions("order_list")
     @ApiOperation(value = "订单列表-各订单数量统计", tags = {"后台-订单管理"})
     @PostMapping(value = "/orderPageCount")
     public R<OrderPageCountVO> orderPageCount(@RequestBody OrderQueryRequest orderQueryRequest) {
@@ -143,6 +151,7 @@
      *
      * @param id 订单id
      */
+    @RequiresPermissions("order_detail")
     @ApiOperation(value = "订单列表-订单详情", tags = {"后台-订单管理"})
     @GetMapping(value = "/detail")
     @ApiImplicitParams({
@@ -158,6 +167,7 @@
      *
      * @param order 站点信息
      */
+    @RequiresPermissions("order_save")
     @ApiOperation(value = "订单列表-新增订单", tags = {"后台-订单管理"})
     @PostMapping(value = "/save")
     public R<String> save(@RequestBody OrderRequest order) {
@@ -194,10 +204,12 @@
         if (null == data) {
             return R.fail(orderClient.save(order).getMsg());
         }
-        try {
-            WebSocketServer.sendInfo("您有一条新的订单,请注意查收!", String.valueOf(order.getServerId()));
-        } catch (IOException e) {
-            return R.fail("订单推送失败!");
+        System.out.println("服务人员id:" + order.getServerId());
+        ChannelHandlerContext context = NettyChannelMap.getData(String.valueOf(order.getServerId()));
+        System.out.println("socket连接信息:" + context);
+        if (null != context) {
+            System.out.println("服务端发送消息到: " + order.getServerId());
+            NettyWebSocketController.sendMsgToClient(context, "您有一条新的订单,请注意查收!");
         }
         return data ? R.ok() : R.fail();
     }
@@ -207,6 +219,7 @@
      *
      * @param type 1:订单派单;2:订单改派
      */
+    @RequiresPermissions("order_reassignment")
     @ApiOperation(value = "订单列表-订单派单/改派", tags = {"后台-订单管理"})
     @GetMapping(value = "/reassignment")
     @ApiImplicitParams({
@@ -250,11 +263,15 @@
             changeDispatch.setIsDelete(Constants.ZERO);
             result = dispatchClient.saveRecord(changeDispatch).getData();
         }
-        try {
-            WebSocketServer.sendInfo("您有一条新的订单,请注意查收!", String.valueOf(workerId));
-        } catch (IOException e) {
-            return R.fail("订单推送失败!");
+        ChannelHandlerContext context = NettyChannelMap.getData(String.valueOf(workerId));
+        if (null != context) {
+            NettyWebSocketController.sendMsgToClient(context, "您有一条新的订单,请注意查收!");
         }
+//        try {
+//            WebSocketServer.sendInfo("您有一条新的订单,请注意查收!", String.valueOf(workerId));
+//        } catch (IOException e) {
+//            return R.fail("订单推送失败!");
+//        }
         return result ? R.ok() : R.fail();
     }
 
@@ -263,6 +280,7 @@
      *
      * @param ids 站点多条id拼接
      */
+    @RequiresPermissions("order_delete")
     @ApiOperation(value = "订单列表-批量删除订单", tags = {"后台-订单管理"})
     @GetMapping(value = "/batchDelete")
     @ApiImplicitParams({
@@ -277,6 +295,7 @@
      *
      * @param orderQueryRequest 筛选参数
      */
+    @RequiresPermissions("order_export")
     @ApiOperation(value = "订单列表-excel导出", tags = {"后台-订单管理"})
     @PostMapping(value = "/excelExport")
     public R<String> excelExport(@RequestBody OrderQueryRequest orderQueryRequest, HttpServletResponse response) {
@@ -286,11 +305,24 @@
     }
 
     /**
+     * 旧数据迁移
+     *
+     * @param index 权限
+     */
+    @ApiOperation(value = "旧数据迁移", tags = {"后台-首页"})
+    @GetMapping(value = "/oldData")
+    public R<String> oldData(@RequestParam Integer index, @RequestParam("pageNum") Integer pageNum,
+                             @RequestParam("pageSize") Integer pageSize) {
+        return orderClient.oldData(index, pageNum, pageSize);
+    }
+
+    /**
      * 订单列表-excel导出
      *
      * @param name  师傅姓名
      * @param phone 师傅电话
      */
+    @RequiresPermissions("order_count")
     @ApiOperation(value = "订单统计", tags = {"后台-订单管理"})
     @GetMapping(value = "/orderCount")
     @ApiImplicitParams({
@@ -302,7 +334,23 @@
     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();
+        LoginUser loginUser = tokenService.getLoginUser();
+        if (null == loginUser) {
+            return R.loginExpire("登录失效!");
+        }
+        List<String> cityList = new ArrayList<>();
+        if (loginUser.getIsFranchisee()) {
+            cityList = loginUser.getCityList();
+        }
+        // 查询参数
+        OrderCountDataRequest orderCountDataRequest = new OrderCountDataRequest();
+        orderCountDataRequest.setWorkerName(name);
+        orderCountDataRequest.setWorkerPhone(phone);
+        orderCountDataRequest.setCityList(cityList);
+        orderCountDataRequest.setPageNum(pageNum);
+        orderCountDataRequest.setPageSize(pageSize);
+        // 远程调用
+        Page<OrderCountVO> data = orderClient.orderCount(orderCountDataRequest).getData();
         List<Integer> idList = data.getRecords().stream().map(OrderCountVO::getWorkerId)
                 .collect(Collectors.toList());
         List<MasterWorker> masterWorkerList;
@@ -314,8 +362,10 @@
             masterWorkerList = masterWorkerService.lambdaQuery()
                     .eq(MasterWorker::getIsDelete, 0).list();
         }
+        // 师傅头像列表
         Map<Integer, String> profilePictureMap = masterWorkerList.stream().collect(Collectors.toMap(MasterWorker::getId,
                 mw -> Optional.ofNullable(mw.getProfilePicture()).orElse("")));
+        // 师傅姓名
         Map<Integer, String> realNameMap = masterWorkerList.stream().collect(Collectors.toMap(MasterWorker::getId,
                 mw -> Optional.ofNullable(mw.getRealName()).orElse("")));
         for (OrderCountVO record : data.getRecords()) {

--
Gitblit v1.7.1