新增股东校验是否为主体公司

ruoyi-admin/src/main/resources/application-druid.yml

@@ -6,7 +6,7 @@ spring:
         druid:
             # 主库数据源
             master:
-                url: jdbc:mysql://1.95.212.35:3306/ruoyi
+                url: jdbc:mysql://1.95.212.35:3306/ruoyi2
                 username: root
                 password: 6221bf4cc656828d
             # 从库数据源

ruoyi-equity/src/main/java/com/ruoyi/equity/service/impl/ShareholderInfoServiceImpl.java

@@ -75,6 +75,13 @@ public class ShareholderInfoServiceImpl extends ServiceImpl<ShareholderInfoMappe
      */
     @Override
     public int insertShareholderInfo(ShareholderInfo shareholderInfo) {
+        //校验是主体公司。
+        if (shareholderInfo.getIsSubject() == 1){
+        ShareholderInfo info = shareholderInfoMapper.getSubjectCompany();
+        if (!Objects.isNull(info)){
+            throw new OnlyOneMainCompanyAllowedException();
+        }
+        }
         shareholderInfo.setCreateTime(DateUtils.getNowDate());
         return shareholderInfoMapper.insertShareholderInfo(shareholderInfo);
     }
@@ -87,6 +94,13 @@ public class ShareholderInfoServiceImpl extends ServiceImpl<ShareholderInfoMappe
      */
     @Override
     public int updateShareholderInfo(ShareholderInfo shareholderInfo) {
+        //校验是主体公司。
+        if (shareholderInfo.getIsSubject() == 1){
+            ShareholderInfo info = shareholderInfoMapper.getSubjectCompany();
+            if (!Objects.isNull(info)){
+                throw new OnlyOneMainCompanyAllowedException();
+            }
+        }
         shareholderInfo.setUpdateTime(DateUtils.getNowDate());
         return shareholderInfoMapper.updateShareholderInfo(shareholderInfo);
     }