From 02e99e878911715d2785e371ae6fb576597578b0 Mon Sep 17 00:00:00 2001
From: luodangjia <luodangjia>
Date: 星期五, 09 八月 2024 09:34:49 +0800
Subject: [PATCH] 8.9

---
 ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/OrderController.java |  242 ++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 213 insertions(+), 29 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 6a2ed6a..417122e 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,34 +1,52 @@
 package com.ruoyi.admin.controller;
 
 
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import com.ruoyi.admin.entity.MasterWorker;
-import com.ruoyi.admin.entity.RecoveryServe;
-import com.ruoyi.admin.entity.Site;
+import com.ruoyi.admin.entity.*;
+import com.ruoyi.admin.netty.NettyChannelMap;
+import com.ruoyi.admin.netty.NettyWebSocketController;
 import com.ruoyi.admin.service.*;
-import com.ruoyi.admin.sorcket.WebSocketServer;
+import com.ruoyi.admin.utils.AddressDto;
+import com.ruoyi.admin.utils.DescribeInstances;
+import com.ruoyi.admin.utils.OcrBase;
 import com.ruoyi.admin.vo.OrderDetailVO;
 import com.ruoyi.common.core.constant.Constants;
+import com.ruoyi.common.core.domain.BaseEntity;
 import com.ruoyi.common.core.domain.R;
+import com.ruoyi.common.core.domain.Result;
+import com.ruoyi.common.core.utils.GaoDeMapUtil;
 import com.ruoyi.common.core.utils.SnowflakeIdWorker;
+import com.ruoyi.common.core.vo.CityInfoVO;
+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.entity.ChangeDispatch;
+import com.ruoyi.order.api.entity.Order;
 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 com.tencentcloudapi.ocr.v20181119.models.BusinessCardOCRResponse;
+import com.tencentcloudapi.ocr.v20181119.models.ClassifyDetectOCRResponse;
+import com.tencentcloudapi.ocr.v20181119.models.SmartStructuralOCRV2Response;
+import com.tencentcloudapi.ocr.v20181119.models.WaybillOCRResponse;
+import io.netty.channel.ChannelHandlerContext;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
+import org.apache.commons.collections4.Get;
 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.*;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 /**
  * <p>
@@ -57,8 +75,6 @@
     private ExchangeDispatchClient dispatchClient;
     @Resource
     private TokenService tokenService;
-    @Resource
-    private FranchiseeService franchiseeService;
 
     /**
      * 雪花算法类
@@ -87,6 +103,7 @@
      *
      * @param id 订单id
      */
+    @RequiresPermissions(value = {"user_detail", "order_detail"}, logical = Logical.OR)
     @ApiOperation(value = "订单列表-查询订单详情(包含服务信息、师傅信息、服务记录、订单评价)", tags = {"后台-订单管理"})
     @GetMapping(value = "/orderDetail")
     @ApiImplicitParams({
@@ -96,12 +113,16 @@
         // 独立orderService
         return R.ok(orderService.orderListDetail(id));
     }
+    @Resource
+    private RecoveryClassifyService recoveryClassifyService;
+
 
     /**
      * 订单列表
      *
      * @param orderQueryRequest 订单列表查询参数
      */
+    @RequiresPermissions("order_list")
     @ApiOperation(value = "订单列表-分页", tags = {"后台-订单管理"})
     @PostMapping(value = "/queryPage")
     public R<Page<Order>> queryPage(@RequestBody OrderQueryRequest orderQueryRequest) {
@@ -109,19 +130,64 @@
         if (null == loginUser) {
             return R.loginExpire("登录失效!");
         }
-        List<String> userCity = new ArrayList<>();
         if (loginUser.getIsFranchisee()) {
-            userCity = loginUser.getCityList();
+            if (null == orderQueryRequest.getCityList() || orderQueryRequest.getCityList().isEmpty()) {
+                orderQueryRequest.setCityList(loginUser.getCityList());
+            } else {
+                orderQueryRequest.getCityList().addAll(loginUser.getCityList());
+            }
         }
-        return R.ok(orderClient.queryPage(orderQueryRequest).getData());
+        List<Integer> servIds = new ArrayList<>();
+        if (orderQueryRequest.getClassNameId()!=null){
+            List<Integer> classIds = recoveryClassifyService.lambdaQuery().eq(RecoveryClassify::getId, orderQueryRequest.getClassNameId()).list().stream().map(RecoveryClassify::getId).collect(Collectors.toList());
+            servIds = recoveryServeService.lambdaQuery().eq(BaseEntity::getIsDelete,0).in(RecoveryServe::getClassifyId, classIds).list().stream().map(RecoveryServe::getId).collect(Collectors.toList());
+        }
+        orderQueryRequest.setServIds(servIds);
+        Page<Order> data = orderClient.queryPage(orderQueryRequest).getData();
+        if (null != data) {
+            for (Order record : data.getRecords()) {
+                // 回收服务信息
+                Integer serverId = record.getServerId();
+                MasterWorker masterWorker = masterWorkerService.lambdaQuery()
+                        .eq(MasterWorker::getId, serverId)
+                        .eq(MasterWorker::getIsDelete, 0).one();
+                if (null != masterWorker) {
+                    record.setServerName(masterWorker.getRealName());
+                    record.setServerPhone(masterWorker.getPhone());
+                }
+                if (record.getAddress()!=null) {
+                    record.setReservationAddress(record.getReservationAddress() + record.getAddress());
+                }
+            }
+        }
+        return R.ok(data);
     }
 
     /**
      * 订单列表
      */
+    @RequiresPermissions("order_list")
     @ApiOperation(value = "订单列表-各订单数量统计", tags = {"后台-订单管理"})
     @PostMapping(value = "/orderPageCount")
     public R<OrderPageCountVO> orderPageCount(@RequestBody OrderQueryRequest orderQueryRequest) {
+        LoginUser loginUser = tokenService.getLoginUser();
+        if (null == loginUser) {
+            return R.loginExpire("登录失效!");
+        }
+        if (loginUser.getIsFranchisee()) {
+            if (null == orderQueryRequest.getCityList() || orderQueryRequest.getCityList().isEmpty()) {
+                orderQueryRequest.setCityList(loginUser.getCityList());
+            } else {
+                orderQueryRequest.getCityList().addAll(loginUser.getCityList());
+            }
+        }
+        List<Integer> servIds = new ArrayList<>();
+        if (orderQueryRequest.getClassNameId()!=null){
+            List<Integer> classIds = recoveryClassifyService.lambdaQuery().eq(RecoveryClassify::getId, orderQueryRequest.getClassNameId()).list().stream().map(RecoveryClassify::getId).collect(Collectors.toList());
+            servIds = recoveryServeService.lambdaQuery().eq(BaseEntity::getIsDelete,0).in(RecoveryServe::getClassifyId, classIds).list().stream().map(RecoveryServe::getId).collect(Collectors.toList());
+        }
+        orderQueryRequest.setServIds(servIds);
+
         return R.ok(orderClient.orderPageCount(orderQueryRequest).getData());
     }
 
@@ -130,6 +196,7 @@
      *
      * @param id 订单id
      */
+    @RequiresPermissions("order_detail")
     @ApiOperation(value = "订单列表-订单详情", tags = {"后台-订单管理"})
     @GetMapping(value = "/detail")
     @ApiImplicitParams({
@@ -139,12 +206,16 @@
         return R.ok(orderClient.detail(id).getData());
     }
 
+
+    @Resource
+    private RecoveryServePriceService recoveryServePriceService;
     /**
      * 新增订单
      * 后台订单为指定订单,默认为待上门状态
      *
      * @param order 站点信息
      */
+    @RequiresPermissions("order_save")
     @ApiOperation(value = "订单列表-新增订单", tags = {"后台-订单管理"})
     @PostMapping(value = "/save")
     public R<String> save(@RequestBody OrderRequest order) {
@@ -174,26 +245,86 @@
         RecoveryServe recoveryServe = recoveryServeService.lambdaQuery()
                 .eq(RecoveryServe::getId, order.getServeId())
                 .eq(RecoveryServe::getIsDelete, 0).one();
+
         order.setServeName(recoveryServe.getServeName());
         order.setServePrice(recoveryServe.getDefaultPrice());
-        order.setOrderMoney(recoveryServe.getDefaultPrice());
+        RecoveryServePrice one = recoveryServePriceService.lambdaQuery().eq(RecoveryServePrice::getCity, order.getCityCode()).eq(RecoveryServePrice::getRecoveryServeId, order.getServeId()).eq(BaseEntity::getIsDelete, 0).one();
+        if (one==null) {
+            order.setOrderMoney(recoveryServe.getDefaultPrice());
+        }else {
+            order.setOrderMoney(one.getRecoveryPrice());
+        }
         Boolean data = orderClient.save(order).getData();
         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();
     }
 
+
+
+
+//    @RequiresPermissions({"reassignment_edit","order_edit"})
+
+    @ApiOperation(value = "订单列表-编辑", tags = {"后台-订单管理"})
+    @PostMapping(value = "/edit")
+    public R<String> edit(@RequestBody OrderRequest order) {
+        // 站点信息
+        Site site = siteService.lambdaQuery()
+                .eq(Site::getId, order.getSiteId())
+                .eq(Site::getIsDelete, 0).one();
+        if (null != site) {
+            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());
+        } else {
+            // 待派单状态
+            order.setState(Constants.ZERO);
+        }
+        RecoveryServe recoveryServe = recoveryServeService.lambdaQuery()
+                .eq(RecoveryServe::getId, order.getServeId())
+                .eq(RecoveryServe::getIsDelete, 0).one();
+        RecoveryServePrice one = recoveryServePriceService.lambdaQuery().eq(RecoveryServePrice::getCity, order.getCityCode()).eq(RecoveryServePrice::getRecoveryServeId, order.getServeId()).eq(BaseEntity::getIsDelete, 0).last("limit 1").one();
+        if (one==null) {
+            order.setOrderMoney(recoveryServe.getDefaultPrice());
+        }else {
+            order.setOrderMoney(one.getRecoveryPrice());
+        }
+        order.setServeName(recoveryServe.getServeName());
+        // 后台订单
+        order.setType(Constants.ONE);
+        Boolean data = orderClient.edit(order).getData();
+
+        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();
+
+    }
     /**
      * 订单派单/改派
      *
      * @param type 1:订单派单;2:订单改派
      */
+    @RequiresPermissions("order_reassignment")
     @ApiOperation(value = "订单列表-订单派单/改派", tags = {"后台-订单管理"})
     @GetMapping(value = "/reassignment")
     @ApiImplicitParams({
@@ -208,6 +339,7 @@
         MasterWorker masterWorker = masterWorkerService.lambdaQuery()
                 .eq(MasterWorker::getId, workerId)
                 .eq(MasterWorker::getIsDelete, 0).one();
+        Order item = orderClient.detail(orderId).getData();
         Order order = orderClient.exchangeOrder(type, orderId, workerId,
                 masterWorker.getRealName(), masterWorker.getPhone()).getData();
         // 订单派单
@@ -216,27 +348,35 @@
             if (order.getState().equals(Constants.SIX) || order.getState().equals(Constants.THREE)) {
                 orderClient.updateState(order.getId(), Constants.ONE);
             }
+            // 订单状态为 待完工时,需要更改状态 待上门且清空师傅到达预约点时间
+            if (order.getState().equals(Constants.TWO)) {
+                orderClient.updateStateAndArrivalTime(orderId, Constants.ONE);
+            }
             // 生成改派信息
             ChangeDispatch changeDispatch = new ChangeDispatch();
-            changeDispatch.setWorkerId(order.getServerId());
-            changeDispatch.setWorkerName(order.getServerName());
+            changeDispatch.setWorkerId(item.getServerId());
+            changeDispatch.setWorkerName(item.getServerName());
             changeDispatch.setApplyReason(applyReason);
             changeDispatch.setApplyTime(new Date());
             changeDispatch.setState(Constants.ONE);
-            changeDispatch.setOrderId(order.getId());
-            changeDispatch.setOrderNumber(order.getOrderNumber());
-            if (null != order.getUserId()) {
-                changeDispatch.setUserId(order.getUserId());
+            changeDispatch.setOrderId(orderId);
+            changeDispatch.setOrderNumber(item.getOrderNumber());
+            if (null != item.getUserId()) {
+                changeDispatch.setUserId(item.getUserId());
             }
-            changeDispatch.setUserName(order.getReservationName());
+            changeDispatch.setUserName(item.getReservationName());
             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();
     }
 
@@ -245,6 +385,7 @@
      *
      * @param ids 站点多条id拼接
      */
+    @RequiresPermissions("order_delete")
     @ApiOperation(value = "订单列表-批量删除订单", tags = {"后台-订单管理"})
     @GetMapping(value = "/batchDelete")
     @ApiImplicitParams({
@@ -254,11 +395,14 @@
         return R.ok(orderClient.batchDelete(ids).getData());
     }
 
+
+
     /**
      * 订单列表-excel导出
      *
      * @param orderQueryRequest 筛选参数
      */
+    @RequiresPermissions("order_export")
     @ApiOperation(value = "订单列表-excel导出", tags = {"后台-订单管理"})
     @PostMapping(value = "/excelExport")
     public R<String> excelExport(@RequestBody OrderQueryRequest orderQueryRequest, HttpServletResponse response) {
@@ -268,11 +412,30 @@
     }
 
     /**
+     * 旧数据迁移
+     *
+     * @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);
+    }
+
+    @ApiOperation(value = "ocr识别", tags = {"后台-首页"})
+    @PostMapping (value = "/ocr")
+    public R<JSONObject> ocr(String ocrAddress) {
+        return R.ok(DescribeInstances.ocr(ocrAddress));
+    }
+
+    /**
      * 订单列表-excel导出
      *
      * @param name  师傅姓名
      * @param phone 师傅电话
      */
+    @RequiresPermissions("order_count")
     @ApiOperation(value = "订单统计", tags = {"后台-订单管理"})
     @GetMapping(value = "/orderCount")
     @ApiImplicitParams({
@@ -284,7 +447,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;
@@ -296,11 +475,16 @@
             masterWorkerList = masterWorkerService.lambdaQuery()
                     .eq(MasterWorker::getIsDelete, 0).list();
         }
-        Map<Integer, String> map = masterWorkerList.stream().collect(Collectors.toMap(MasterWorker::getId,
+        // 师傅头像列表
+        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()) {
             Integer workerId = record.getWorkerId();
-            record.setProfilePicture(map.get(workerId));
+            record.setRealName(realNameMap.get(workerId));
+            record.setProfilePicture(profilePictureMap.get(workerId));
         }
         return R.ok(data);
     }

--
Gitblit v1.7.1