123

ace-modules/ace-admin/src/main/java/com/github/wxiaoqi/security/admin/biz/BaseUserMemberBiz.java

@@ -72,9 +72,13 @@ public class BaseUserMemberBiz extends BaseBiz<BaseUserMemberMapper,BaseUserMemb
         if(freeDays==null||freeDays==0){
             freeDays=totalNumber;
         }
+        Integer buyCount=1;
+        if(userMemberDTO.getBuyCount()!=null){
+            buyCount=userMemberDTO.getBuyCount();
+        }
         if(userMemberVo==null){
             baseUserMember.setRentFreeDays(freeDays);
-            baseUserMember.setBuyCount(1);
+            baseUserMember.setBuyCount(buyCount);
             insertSelective(baseUserMember);
         }else {
             Integer discount=userMemberVo.getDiscount()==null?0:userMemberVo.getDiscount();
@@ -90,10 +94,6 @@ public class BaseUserMemberBiz extends BaseBiz<BaseUserMemberMapper,BaseUserMemb
             baseUserMember.setId(userMemberVo.getId());
             totalNumber+=userMemberVo.getTotalNumber()==null?0:userMemberVo.getTotalNumber();
             freeDays+=userMemberVo.getRentFreeDays()==null?0:userMemberVo.getRentFreeDays();
-            Integer buyCount=1;
-            if(userMemberDTO.getBuyCount()!=null){
-                buyCount=userMemberDTO.getBuyCount();
-            }
             baseUserMember.setBuyCount(userMemberVo.getBuyCount()+buyCount);
             baseUserMember.setTotalNumber(totalNumber);
             baseUserMember.setRentFreeDays(freeDays);