Просмотр исходного кода

fix: 优化了创建优惠券组的接口

DingXiaoYu 2 лет назад
Родитель
Сommit
3d27594808

+ 3 - 0
src/main/java/com/seecoder/BlueWhale/serviceImpl/CouponServiceImpl.java

@@ -1,6 +1,7 @@
 package com.seecoder.BlueWhale.serviceImpl;
 
 import com.seecoder.BlueWhale.enums.CouponTypeEnum;
+import com.seecoder.BlueWhale.enums.RoleEnum;
 import com.seecoder.BlueWhale.exception.BlueWhaleException;
 import com.seecoder.BlueWhale.po.Coupon;
 import com.seecoder.BlueWhale.po.CouponGroup;
@@ -39,6 +40,8 @@ public class CouponServiceImpl implements CouponService {
             }
         }
         CouponGroup couponGroup=couponGroupVO.toPO();
+        User user=securityUtil.getCurrentUser();
+        couponGroup.setStoreId(user.getRole()==RoleEnum.STAFF ? user.getStoreId() : 0);
         couponGroupRepository.save(couponGroup);
         return true;
     }