package many2many import ( "git.sxidc.com/go-framework/baize/framework/binding" "git.sxidc.com/go-framework/baize/framework/core/api" "git.sxidc.com/go-framework/baize/framework/core/api/request" "git.sxidc.com/go-framework/baize/framework/core/api/response" "git.sxidc.com/go-framework/baize/framework/core/domain" "git.sxidc.com/go-framework/baize/framework/core/domain/entity" "git.sxidc.com/go-framework/baize/framework/core/infrastructure" "git.sxidc.com/go-framework/baize/framework/core/infrastructure/database" "git.sxidc.com/go-framework/baize/framework/core/infrastructure/database/sql" "git.sxidc.com/go-tools/utils/slice" "github.com/pkg/errors" ) func Update(middleTableName string, fromTableName string, fromDomainCNName string, fromRelationFieldName string, fromRelationColumnName string, toTableName string, toRelationColumnName string) binding.ServiceFunc[any] { return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (any, error) { object := objects[0] if object == nil { return nil, errors.New("领域实体为空") } dbExecutor := i.DBExecutor() fromEntity, ok := object.(entity.Entity) if !ok { return nil, errors.New("领域对象不是实体") } // 字段校验 err := entity.CheckFieldID(fromEntity) if err != nil { return nil, err } // from存在性校验 fromExist, err := database.CheckExist(dbExecutor, &sql.CheckExistExecuteParams{ TableName: fromTableName, Conditions: sql.NewConditions().Equal(entity.ColumnID, fromEntity.GetID()), }) if err != nil { return nil, err } if !fromExist { return nil, errors.New(fromEntity.DomainCNName() + "不存在") } if !domain.HasField(object, fromRelationFieldName) { return nil, errors.New("关联字段" + fromRelationFieldName + "不存在") } toIDs, err := domain.Field[[]string](object, fromRelationFieldName) if err != nil { return nil, err } if toIDs != nil && len(toIDs) != 0 { for _, toID := range toIDs { err := entity.CheckIDTypeValue(fromDomainCNName, fromRelationFieldName, toID) if err != nil { return nil, err } } toIDs = slice.RemoveRepeatElement(toIDs) } err = database.Transaction(dbExecutor, func(tx database.Executor) error { err := database.Delete(tx, &sql.DeleteExecuteParams{ TableName: middleTableName, Conditions: sql.NewConditions().Equal(fromRelationColumnName, fromEntity.GetID()), }) if err != nil { return err } if toIDs == nil || len(toIDs) == 0 { return nil } toCount, err := database.Count(dbExecutor, &sql.CountExecuteParams{ TableName: toTableName, Conditions: sql.NewConditions().In(entity.ColumnID, toIDs), }) if err != nil { return err } if int(toCount) != len(toIDs) { return errors.New("部分{{ $toCNName }}不存在") } tableRows := make([]sql.TableRow, len(toIDs)) for index, toID := range toIDs { tableRows[index] = *(sql.NewTableRow(). Add(fromRelationColumnName, fromEntity.GetID()). Add(toRelationColumnName, toID)) } err = database.InsertBatch(tx, &sql.InsertBatchExecuteParams{ TableName: middleTableName, TableRowBatch: tableRows, }) if err != nil { if database.IsErrorDBRecordHasExist(err) { return errors.New("关联项已存在") } return err } return nil }) if err != nil { return nil, err } return nil, nil } } func Query[TI any](middleTableName string, fromTableName string, fromRelationColumnName string, toTableName string, toRelationColumnName string) binding.ServiceFunc[response.InfosData[TI]] { return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (response.InfosData[TI], error) { errResponse := response.InfosData[TI]{ Infos: make([]TI, 0), } if params == nil { return errResponse, errors.New("请求参数为空") } object := objects[0] if object == nil { return errResponse, errors.New("领域实体为空") } dbExecutor := i.DBExecutor() queryParams, ok := params.(request.QueryWithIDRequestParams) if !ok { return errResponse, errors.New("请求参数不是Query接口") } fromEntity, ok := object.(entity.Entity) if !ok { return errResponse, errors.New("领域对象不是实体") } // from存在性校验 fromExist, err := database.CheckExist(dbExecutor, &sql.CheckExistExecuteParams{ TableName: fromTableName, Conditions: sql.NewConditions().Equal(entity.ColumnID, fromEntity.GetID()), }) if err != nil { return errResponse, err } if !fromExist { return errResponse, errors.New(fromEntity.DomainCNName() + "不存在") } toIDResults, totalCount, err := database.Query(dbExecutor, &sql.QueryExecuteParams{ TableName: middleTableName, SelectClauses: []string{toRelationColumnName}, Conditions: sql.NewConditions().Equal(fromRelationColumnName, fromEntity.GetID()), PageNo: queryParams.GetPageNo(), PageSize: queryParams.GetPageSize(), }) if err != nil { return errResponse, err } if toIDResults == nil || len(toIDResults) == 0 { return response.InfosData[TI]{ Infos: make([]TI, 0), TotalCount: 0, PageNo: queryParams.GetPageNo(), }, nil } toIDs := make([]string, 0) err = sql.ParseSqlResult(toIDResults, &toIDs) if err != nil { return errResponse, err } toResults, _, err := database.Query(dbExecutor, &sql.QueryExecuteParams{ TableName: toTableName, Conditions: sql.NewConditions().In(entity.ColumnID, toIDs), }) if err != nil { return errResponse, err } infos := make([]TI, 0) err = sql.ParseSqlResult(toResults, &infos) if err != nil { return errResponse, err } return response.InfosData[TI]{ Infos: infos, TotalCount: totalCount, PageNo: queryParams.GetPageNo(), }, nil } }