baize/convenient/value_object/service.go

package value_object

import (
	"git.sxidc.com/go-framework/baize/api"
	"git.sxidc.com/go-framework/baize/binding"
	"git.sxidc.com/go-framework/baize/binding/request"
	"git.sxidc.com/go-framework/baize/binding/response"
	"git.sxidc.com/go-framework/baize/domain"
	"git.sxidc.com/go-framework/baize/infrastructure"
	"git.sxidc.com/go-framework/baize/infrastructure/database"
	"git.sxidc.com/go-framework/baize/infrastructure/database/sql"
	"git.sxidc.com/go-framework/baize/tag/sql/sql_mapping"
	"git.sxidc.com/service-supports/fserr"
	"reflect"
)

func Create(tableName string, callbacks *Callbacks[any]) binding.ServiceFunc[any] {
	return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (any, error) {
		dbExecutor := i.DBExecutor()

		valueObject, ok := objects[0].(domain.ValueObject)
		if !ok {
			return nil, fserr.New("需要传递领域对象应该为值对象")
		}

		err := valueObject.ForCreate()
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = callbackBeforeDBOperate(callbacks, valueObject, i, dbExecutor)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = database.InsertEntity(dbExecutor, tableName, valueObject)
		if err != nil {
			if database.IsErrorDBRecordHasExist(err) {
				err = fserr.New(valueObject.DomainCNName() + "已存在")
			}

			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = callbackAfterDBOperate(callbacks, valueObject, i, dbExecutor)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		return callbackOnSuccessReturn(callbacks, valueObject, i, nil)
	}
}

func Delete(tableName string, callbacks *Callbacks[any]) binding.ServiceFunc[any] {
	return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (any, error) {
		dbExecutor := i.DBExecutor()

		valueObject, ok := objects[0].(domain.ValueObject)
		if !ok {
			return nil, fserr.New("需要传递领域对象应该为值对象")
		}

		err := valueObject.CheckKeyFields()
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = callbackBeforeDBOperate(callbacks, valueObject, i, dbExecutor)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = database.DeleteEntity(dbExecutor, tableName, valueObject)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = callbackAfterDBOperate(callbacks, valueObject, i, dbExecutor)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		return callbackOnSuccessReturn(callbacks, valueObject, i, nil)
	}
}

type ConditionFieldCallback func(conditions *sql.Conditions, fieldName string, columnName string, value any) (hasDeal bool)

func Query[O any](tableName string, callbacks *Callbacks[response.InfosData[O]], conditionFieldCallback ConditionFieldCallback) binding.ServiceFunc[response.InfosData[O]] {
	return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (response.InfosData[O], error) {
		dbExecutor := i.DBExecutor()

		queryParams, ok := params.(request.Query)
		if !ok {
			return response.InfosData[O]{}, fserr.New("请求参数不是Query接口")
		}

		valueObject, ok := objects[0].(domain.ValueObject)
		if !ok {
			return response.InfosData[O]{}, fserr.New("需要传递领域对象应该为值对象")
		}

		conditions := sql.NewConditions()

		fields, err := sql_mapping.DefaultUsage(valueObject)
		if err != nil {
			return response.InfosData[O]{}, err
		}

		for _, field := range fields {
			hasDeal := false
			if conditionFieldCallback != nil {
				hasDeal = conditionFieldCallback(conditions, field.FieldName, field.ColumnName, field.Value)
			}

			if !hasDeal {
				fieldValue := reflect.ValueOf(field.Value)
				if !fieldValue.IsZero() {
					conditions.Equal(field.ColumnName, field.Value)
				}
			}
		}

		err = callbackBeforeDBOperate(callbacks, valueObject, i, dbExecutor)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, response.InfosData[O]{})
		}

		results, totalCount, err := database.Query(dbExecutor, &sql.QueryExecuteParams{
			TableName:  tableName,
			Conditions: conditions,
			PageNo:     queryParams.GetPageNo(),
			PageSize:   queryParams.GetPageSize(),
		})
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, response.InfosData[O]{})
		}

		err = callbackAfterDBOperate(callbacks, valueObject, i, dbExecutor)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, response.InfosData[O]{})
		}

		infos := make([]O, 0)
		err = sql.ParseSqlResult(results, &infos)
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, response.InfosData[O]{})
		}

		output := response.InfosData[O]{
			Infos:      infos,
			TotalCount: totalCount,
			PageNo:     queryParams.GetPageNo(),
		}

		return callbackOnSuccessReturn(callbacks, valueObject, i, output)
	}
}

func CreateTx(tableName string, callbacks *Callbacks[any]) binding.ServiceFunc[any] {
	return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (any, error) {
		dbExecutor := i.DBExecutor()

		valueObject, ok := objects[0].(domain.ValueObject)
		if !ok {
			return nil, fserr.New("需要传递领域对象应该为值对象")
		}

		err := database.Transaction(dbExecutor, func(tx database.Executor) error {
			err := callbackBeforeDBOperate(callbacks, valueObject, i, tx)
			if err != nil {
				return err
			}

			err = database.InsertEntity(dbExecutor, tableName, valueObject)
			if err != nil {
				if database.IsErrorDBRecordHasExist(err) {
					err = fserr.New(valueObject.DomainCNName() + "已存在")
				}

				return err
			}

			err = callbackAfterDBOperate(callbacks, valueObject, i, tx)
			if err != nil {
				return err
			}

			return nil
		})
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		return callbackOnSuccessReturn(callbacks, valueObject, i, nil)
	}
}

func DeleteTx(tableName string, callbacks *Callbacks[any]) binding.ServiceFunc[any] {
	return func(c *api.Context, params request.Params, objects []domain.Object, i *infrastructure.Infrastructure) (any, error) {
		dbExecutor := i.DBExecutor()

		valueObject, ok := objects[0].(domain.ValueObject)
		if !ok {
			return nil, fserr.New("需要传递领域对象应该为值对象")
		}

		err := valueObject.CheckKeyFields()
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		err = database.Transaction(dbExecutor, func(tx database.Executor) error {
			err = callbackBeforeDBOperate(callbacks, valueObject, i, tx)
			if err != nil {
				return err
			}

			err = database.DeleteEntity(dbExecutor, tableName, valueObject)
			if err != nil {
				return err
			}

			err = callbackAfterDBOperate(callbacks, valueObject, i, tx)
			if err != nil {
				return err
			}

			return nil
		})
		if err != nil {
			return callbackOnErrorReturn(callbacks, valueObject, err, i, nil)
		}

		return callbackOnSuccessReturn(callbacks, valueObject, i, nil)
	}
}