设置主门店

haoban-manage3-wx/src/main/java/com/gic/haoban/manage/web/controller/WxStaffController.java

@@ -543,10 +543,23 @@ public class WxStaffController extends WebBaseController {
     @RequestMapping("/set-main-store")
     public HaobanResponse setMainStore(String staffId, String storeId, String wxEnterpriseId, String dateStr) {
         StaffDTO staff = staffApiService.selectById(staffId);
+        Date curDate = new Date();
         if (staff == null) {
             return resultResponse(HaoBanErrCode.ERR_10006);
         }
-        this.clerkMainStoreRelatedApiService.setMainStore(staffId, storeId, wxEnterpriseId);
+        //给测试使用
+        if (StringUtils.isNotBlank(dateStr)) {
+            curDate = cn.hutool.core.date.DateUtil.parse(dateStr);
+        }
+        int i = cn.hutool.core.date.DateUtil.weekOfYear(curDate);
+        String key = "haoban_set_main_store_" + i + ":" + staffId;
+        Object hasSet = RedisUtil.getCache(key);
+        if (hasSet == null) {
+            RedisUtil.setCache(key, true, 7 * 24 * 60 * 60L);
+            clerkMainStoreRelatedApiService.setMainStore(staffId, storeId, wxEnterpriseId);
+        } else {
+            return resultResponse(HaoBanErrCode.ERR_10031);
+        }
         return resultResponse(HaoBanErrCode.ERR_1);
     }
 
@@ -555,7 +568,17 @@ public class WxStaffController extends WebBaseController {
     @HttpLimit
     @RequestMapping("/check-main-store")
     public HaobanResponse checkMainStore(String staffId, String storeId, String wxEnterpriseId, String dateStr) {
+        Date curDate = new Date();
+        int i = cn.hutool.core.date.DateUtil.weekOfYear(curDate);
+        String key = "haoban_set_main_store_" + i + ":" + staffId;
+        Object hasSet = RedisUtil.getCache(key);
+        if (hasSet == null) {
+            //可修改
             return resultResponse(HaoBanErrCode.ERR_1, 1);
+        } else {
+            //不可以设置
+            return resultResponse(HaoBanErrCode.ERR_1, 0);
+        }
     }