公司的注册 授权节分

src/main/java/com/dk/oauth/service/impl/CompanyServiceImpl.java

@@ -273,30 +273,44 @@ public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> impl
                 IntegralItem integralItem = new IntegralItem();
                 integralItem.setIntegralType(Constant.IntegralType.INTEGRAL_TYPE_INVITE.getName());
                 integralItem.setIntegral(integralTactic.getShareRegIntegral());
-//                integralItem.setCpId(company.getCpId());
-//                integralItem.setCpIdFrom(companyVO.getCpIdFrom());
+                integralItem.setCpId(company.getCpId());
+                integralItem.setCpIdFrom(companyVO.getCpIdFrom());
                 integralItem.setAccDate(LocalDateTime.now());
                 integralItem.setUserId(companyVO.getOwner());
                 integralItemMapper.insert(integralItem);
                 // 积分 新公司的
                 Integral integral = new Integral();
-//                integral.setCpId(company.getCpId());
-                integral.setIntegral(integralTactic.getShareRegIntegral());
-                integralMapper.insert(integral);
+                integral.setCpId(company.getCpId());
+                integral.setFlgValid(true);
+                List<Integral> integrals = integralMapper.selectByCond(integral);
+                if (integrals != null && integrals.size() > 0) {
+                    integral.setIntegral(integralTactic.getShareRegIntegral() + integrals.get(0).getIntegral());
+                    integralMapper.updateById(integral);
+                }else{
+                    integral.setIntegral(integralTactic.getShareRegIntegral());
+                    integralMapper.insert(integral);
+                }
 
                 // 积分明细 来源公司公司的
                 integralItem = new IntegralItem();
                 integralItem.setIntegralType(Constant.IntegralType.INTEGRAL_TYPE_INVITE.getName());
                 integralItem.setIntegral(integralTactic.getShareRegIntegral());
-//                integralItem.setCpId(companyVO.getCpIdFrom());
+                integralItem.setCpId(companyVO.getCpIdFrom());
                 integralItem.setAccDate(LocalDateTime.now());
                 integralItem.setUserId(companyVO.getOwner());
                 integralItemMapper.insert(integralItem);
                 // 积分明细 来源公司公司的
                 integral = new Integral();
-//                integral.setCpId(company.getCpIdFrom());
-                integral.setIntegral(integralTactic.getShareRegIntegral());
-                integralMapper.insert(integral);
+                integral.setCpId(companyVO.getCpIdFrom());
+                integral.setFlgValid(true);
+                integrals = integralMapper.selectByCond(integral);
+                if (integrals != null && integrals.size() > 0) {
+                    integral.setIntegral(integralTactic.getShareRegIntegral() + integrals.get(0).getIntegral());
+                    integralMapper.updateById(integral);
+                }else{
+                    integral.setIntegral(integralTactic.getShareRegIntegral());
+                    integralMapper.insert(integral);
+                }
             }
         }