From 4536a68f86f50e09032066e00e9e3cacf366cbd4 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 23 Jun 2021 11:02:33 +0800
Subject: [PATCH] modify

---
 zq-erp/src/main/java/com/matrix/system/app/action/ApiOrderAction.java |   44 ++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 40 insertions(+), 4 deletions(-)

diff --git a/zq-erp/src/main/java/com/matrix/system/app/action/ApiOrderAction.java b/zq-erp/src/main/java/com/matrix/system/app/action/ApiOrderAction.java
index 4f7c62a..67a8262 100644
--- a/zq-erp/src/main/java/com/matrix/system/app/action/ApiOrderAction.java
+++ b/zq-erp/src/main/java/com/matrix/system/app/action/ApiOrderAction.java
@@ -1,6 +1,7 @@
 package com.matrix.system.app.action;
 
 import com.matrix.core.constance.MatrixConstance;
+import com.matrix.core.exception.GlobleException;
 import com.matrix.core.pojo.AjaxResult;
 import com.matrix.core.pojo.PaginationVO;
 import com.matrix.core.tools.WebUtil;
@@ -13,7 +14,9 @@
 import com.matrix.system.app.vo.ShoppingGoodsListVo;
 import com.matrix.system.common.bean.SysUsers;
 import com.matrix.system.common.tools.DataAuthUtil;
-import com.matrix.system.hive.bean.ShoppingGoodsCategory;
+import com.matrix.system.hive.action.util.QueryUtil;
+import com.matrix.system.hive.bean.*;
+import com.matrix.system.hive.dao.SysShopInfoDao;
 import com.matrix.system.hive.plugin.util.CollectionUtils;
 import com.matrix.system.hive.pojo.ShoppingCarItem;
 import com.matrix.system.hive.pojo.ShoppingCarItemsVo;
@@ -29,6 +32,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 
 /**
@@ -49,6 +53,9 @@
     @Autowired
     private SysOrderService sysOrderService;
 
+    @Autowired
+    private SysShopInfoDao shopInfoDao;
+
     @ApiOperation(value = "获取商品类型列表", notes = "获取商品类型列表")
     @ApiResponses({
             @ApiResponse(code = 200, message = "ok", response = ShoppingGoodsCategory.class)
@@ -57,9 +64,17 @@
     public AjaxResult findShoppingGoodsType() {
         SysUsers user = WebUtil.getSessionAttribute(MatrixConstance.LOGIN_KEY);
 
+        SysShopInfo zbShop = shopInfoDao.selectZbShop(user.getCompanyId());
         ShoppingGoodsCategory category = new ShoppingGoodsCategory();
-        category.setShopId(user.getShopId());
-        return AjaxResult.buildSuccessInstance(shoppingGoodsCategoryService.findByModel(category));
+        category.setShopId(zbShop.getId());
+        List<ShoppingGoodsCategory> list = new ArrayList<>();
+        List<ShoppingGoodsCategory> byModel = shoppingGoodsCategoryService.findByModel(category);
+        ShoppingGoodsCategory type = new ShoppingGoodsCategory();
+        type.setName("全部");
+        type.setParentId(0L);
+        list.add(type);
+        list.addAll(byModel);
+        return AjaxResult.buildSuccessInstance(list);
     }
 
     @ApiOperation(value = "获取商品列表", notes = "获取商品列表")
@@ -68,7 +83,11 @@
     })
     @PostMapping(value = "/findShoppingGoods")
     public AjaxResult findShoppingGoods(@RequestBody @Validated ShoppingGoodsListDto shoppingGoodsListDto) {
-
+        SysUsers user = WebUtil.getSessionAttribute(MatrixConstance.LOGIN_KEY);
+        if (!DataAuthUtil.hasAllShopAuth()) {
+            shoppingGoodsListDto.setShopId(user.getShopId());
+        }
+        QueryUtil.setQueryLimitCom(shoppingGoodsListDto);
         return AjaxResult.buildSuccessInstance(shoppingGoodsService.findShoppingGoodsListForApi(shoppingGoodsListDto), shoppingGoodsService.findShoppingGoodsListTotalForApi(shoppingGoodsListDto));
     }
 
@@ -132,6 +151,7 @@
         if (!DataAuthUtil.hasAllShopAuth()) {
             orderListDto.setShopId(user.getShopId());
         }
+        QueryUtil.setQueryLimitCom(orderListDto);
         return AjaxResult.buildSuccessInstance(sysOrderService.findApiOrderListInPage(orderListDto, pageVo), sysOrderService.findApiOrderListTotal(orderListDto));
     }
 
@@ -147,4 +167,20 @@
         return ajaxResult;
     }
 
+    @ApiOperation(value = "取消订单", notes = "取消订单")
+    @GetMapping(value = "/cancelOrder/{orderId}")
+    public AjaxResult cancelOrder(@PathVariable("orderId") Long orderId) {
+        SysOrder order = sysOrderService.findById(orderId);
+        if (order == null) {
+            throw new GlobleException("订单不存在");
+        }
+
+
+        int i = sysOrderService.cancelOrder(orderId);
+        if (i > 0) {
+            return AjaxResult.buildSuccessInstance("取消成功");
+        }
+        return AjaxResult.buildFailInstance("取消失败");
+    }
+
 }
\ No newline at end of file

--
Gitblit v1.9.1