feat:共创模式删除时增加业务数据校验

陈志颖
pkg/application/cooperationMode/service/cooperation_mode.go
pkg/infrastructure/dao/pg_cooperation_mode_dao.go
--- a/pkg/application/cooperationMode/service/cooperation_mode.go
查看文件 @8c46ce9
+++ b/pkg/application/cooperationMode/service/cooperation_mode.go
查看文件 @8c46ce9
@@ -92,6 +92,7 @@ func (cooperationModeService *CooperationModeService) CreateCooperationMode(crea
 	} else {
 		cooperationModeDao = value
 	}
+
 	// 校验共创模式编码唯一性
 	numberAvailable, _ := cooperationModeDao.CheckModeNumberAvailable(map[string]interface{}{
 		"companyId":             createCooperationModeCommand.CompanyId,
@@ -231,6 +232,8 @@ func (cooperationModeService *CooperationModeService) RemoveCooperationMode(remo
 	defer func() {
 		_ = transactionContext.RollbackTransaction()
 	}()
+
+	// 共创模式仓储初始化
 	var cooperationModeRepository domain.CooperationModeRepository
 	if value, err := factory.CreateCooperationModeRepository(map[string]interface{}{
 		"transactionContext": transactionContext,
@@ -239,6 +242,18 @@ func (cooperationModeService *CooperationModeService) RemoveCooperationMode(remo
 	} else {
 		cooperationModeRepository = value
 	}
+
+	// 共创模式DAO初始化
+	var cooperationModeDao *dao.CooperationModeDao
+	if value, err := factory.CreateCooperationModeDao(map[string]interface{}{
+		"transactionContext": transactionContext,
+	}); err != nil {
+		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
+	} else {
+		cooperationModeDao = value
+	}
+
+	//  获取共创模式
 	cooperationMode, err := cooperationModeRepository.FindOne(map[string]interface{}{"cooperationModeId": removeCooperationModeCommand.CooperationModeId})
 	if err != nil {
 		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, "共创模式不存在")
@@ -246,13 +261,25 @@ func (cooperationModeService *CooperationModeService) RemoveCooperationMode(remo
 	if cooperationMode == nil {
 		return nil, application.ThrowError(application.RES_NO_FIND_ERROR, fmt.Sprintf("%s", strconv.FormatInt(removeCooperationModeCommand.CooperationModeId, 10)))
 	}
-	if cooperationMode, err := cooperationModeRepository.Remove(cooperationMode); err != nil {
+
+	// 校验共创模式是否可以删除
+	deleteAvailable, _ := cooperationModeDao.CheckModeIsDeleteAvailable(map[string]interface{}{
+		"companyId":             removeCooperationModeCommand.CompanyId,
+		"orgId":                 removeCooperationModeCommand.OrgId,
+		"cooperationModeNumber": cooperationMode.CooperationModeNumber,
+	})
+	if !deleteAvailable {
+		return nil, application.ThrowError(application.TRANSACTION_ERROR, "当前共创模式存在业务数据，不可删除")
+	}
+
+	// 移除共创模式
+	if cooperationModeRemoved, err := cooperationModeRepository.Remove(cooperationMode); err != nil {
 		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 	} else {
 		if err := transactionContext.CommitTransaction(); err != nil {
 			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
 		}
-		return cooperationMode, nil
+		return cooperationModeRemoved, nil
 	}
 }
@@ -271,6 +298,8 @@ func (cooperationModeService *CooperationModeService) BatchRemoveCooperationMode
 	defer func() {
 		_ = transactionContext.RollbackTransaction()
 	}()
+
+	// 共创模式仓储初始化
 	var cooperationModeRepository domain.CooperationModeRepository
 	if value, err := factory.CreateCooperationModeRepository(map[string]interface{}{
 		"transactionContext": transactionContext,
@@ -279,13 +308,41 @@ func (cooperationModeService *CooperationModeService) BatchRemoveCooperationMode
 	} else {
 		cooperationModeRepository = value
 	}
-	cooperationModeIds, _ := utils.SliceAtoi(batchRemoveCooperationModeCommand.CooperationModeIds)
+
+	// 共创模式ID列表类型转换
+	cooperationModeIds, err := utils.SliceAtoi(batchRemoveCooperationModeCommand.CooperationModeIds)
+	if err != nil {
+		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, "共创模式类型错误")
+	}
+
+	// 共创模式DAO初始化
+	var cooperationModeDao *dao.CooperationModeDao
+	if value, err := factory.CreateCooperationModeDao(map[string]interface{}{
+		"transactionContext": transactionContext,
+	}); err != nil {
+		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
+	} else {
+		cooperationModeDao = value
+	}
+
+	// 获取共创模式
 	if count, cooperationModes, err := cooperationModeRepository.Find(map[string]interface{}{
 		"cooperationModeIds": cooperationModeIds,
 	}); err != nil {
 		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 	} else {
 		if count > 0 {
+			for _, cooperationMode := range cooperationModes {
+				// 校验共创模式是否可以删除
+				deleteAvailable, _ := cooperationModeDao.CheckModeIsDeleteAvailable(map[string]interface{}{
+					"companyId":             batchRemoveCooperationModeCommand.CompanyId,
+					"orgId":                 batchRemoveCooperationModeCommand.OrgId,
+					"cooperationModeNumber": cooperationMode.CooperationModeNumber,
+				})
+				if !deleteAvailable {
+					return nil, application.ThrowError(application.TRANSACTION_ERROR, "所选共创模式存在业务数据，不可删除")
+				}
+			}
 			if cooperationMode, err := cooperationModeRepository.BatchRemove(cooperationModes); err != nil {
 				return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 			} else {
--- a/pkg/infrastructure/dao/pg_cooperation_mode_dao.go
查看文件 @8c46ce9
+++ b/pkg/infrastructure/dao/pg_cooperation_mode_dao.go
查看文件 @8c46ce9
@@ -49,6 +49,24 @@ func (dao *CooperationModeDao) CheckModeNumberAvailable(queryOptions map[string]
 	return !ok, err
 }
+// CheckModeIsDeleteAvailable 校验模式是否有项目数据关联
+func (dao *CooperationModeDao) CheckModeIsDeleteAvailable(queryOptions map[string]interface{}) (bool, error) {
+	tx := dao.transactionContext.PgTx
+	var cooperationProjectModels []*models.CooperationProject
+	query := tx.Model(&cooperationProjectModels)
+	if cooperationModeNumber, ok := queryOptions["cooperationModeNumber"]; ok && cooperationModeNumber != "" {
+		query = query.Where("cooperation_mode_number = ?", cooperationModeNumber)
+	}
+	if companyId, ok := queryOptions["companyId"]; ok && companyId.(int64) != 0 {
+		query = query.Where(`cooperation_project.company @> '{"companyId":"?"}'`, companyId)
+	}
+	if orgId, ok := queryOptions["orgId"]; ok && orgId.(int64) != 0 {
+		query = query.Where(`cooperation_project.org @> '{"orgId":"?"}'`, orgId)
+	}
+	ok, err := query.Exists()
+	return !ok, err
+}
+
 // UpdateCooperationModeSlice 更新多个共创模式
 func (dao *CooperationModeDao) UpdateCooperationModeSlice(modeIds []int64, status int32) ([]*domain.CooperationMode, error) {
 	tx := dao.transactionContext.PgTx