baize/convenient/entity_crud/simple.go

package entity_crud

import (
	"git.sxidc.com/go-framework/baize/framework/binding"
	"git.sxidc.com/go-framework/baize/framework/binding/request"
	"git.sxidc.com/go-framework/baize/framework/binding/response"
	"git.sxidc.com/go-framework/baize/framework/core/api"
	"git.sxidc.com/go-framework/baize/framework/core/domain"
	"git.sxidc.com/go-framework/baize/framework/core/domain/entity"
)

// Simple 实体CRUD的Bind参数
// I 为查询相关接口返回的Info类型
type Simple[I any] struct {
	// 使用的领域实体，注意是Entity类型
	Entity entity.Entity

	// 数据库Schema
	Schema string

	// 创建使用的请求参数
	CreateJsonBody request.Params

	// 删除使用的请求参数，注意是WithID类型
	DeleteQueryParams request.IDRequestParam

	// 更新使用的请求参数，注意是WithID类型
	UpdateJsonBody request.IDRequestParam

	// 查询使用的请求参数，注意是Query类型
	QueryQueryParams request.QueryRequestParams

	// 根据ID查询使用的请求参数，注意是WithID类型
	GetByIDQueryParams request.IDRequestParam

	// 可选配置项，通过WithXXX配置
	createOptions  *CreateOptions
	deleteOptions  *DeleteOptions
	updateOptions  *UpdateOptions
	queryOptions   *QueryOptions[I]
	getByIDOptions *GetByIDOptions[I]
}

func (simple *Simple[I]) bind(binder *binding.Binder) {
	createOptions := simple.createOptions
	deleteOptions := simple.deleteOptions
	updateOptions := simple.updateOptions
	queryOptions := simple.queryOptions
	getByIDOptions := simple.getByIDOptions

	tableName := domain.TableName(simple.Schema, simple.Entity)
	domainPath := domain.RelativeDomainPath(simple.Entity)

	// 创建
	if !createOptions.disable {
		if !createOptions.needTx {
			binding.PostBind[string](binder, &binding.SimpleBindItem[string]{
				Path:          domainPath + "/create",
				ResponseFunc:  response.SendIDResponse,
				RequestParams: simple.CreateJsonBody,
				Objects:       []domain.Object{simple.Entity},
				ServiceFunc:   Create(tableName, createOptions.callbacks),
			}, createOptions.middlewares...)
		} else {
			binding.PostBind(binder, &binding.SimpleBindItem[string]{
				Path:          domainPath + "/create",
				ResponseFunc:  response.SendIDResponse,
				RequestParams: simple.CreateJsonBody,
				Objects:       []domain.Object{simple.Entity},
				ServiceFunc:   CreateTx(tableName, createOptions.callbacks),
			}, createOptions.middlewares...)
		}
	}

	// 删除
	if !deleteOptions.disable {
		if !deleteOptions.needTx {
			binding.DeleteBind(binder, &binding.SimpleBindItem[any]{
				Path:          domainPath + "/delete",
				ResponseFunc:  response.SendMsgResponse,
				RequestParams: simple.DeleteQueryParams,
				Objects:       []domain.Object{simple.Entity},
				ServiceFunc:   Delete(tableName, deleteOptions.callbacks),
			}, deleteOptions.middlewares...)
		} else {
			binding.DeleteBind(binder, &binding.SimpleBindItem[any]{
				Path:          domainPath + "/delete",
				ResponseFunc:  response.SendMsgResponse,
				RequestParams: simple.DeleteQueryParams,
				Objects:       []domain.Object{simple.Entity},
				ServiceFunc:   DeleteTx(tableName, deleteOptions.callbacks),
			}, deleteOptions.middlewares...)
		}
	}

	// 修改
	if !updateOptions.disable {
		if !updateOptions.needTx {
			binding.PutBind(binder, &binding.SimpleBindItem[any]{
				Path:          domainPath + "/update",
				ResponseFunc:  response.SendMsgResponse,
				RequestParams: simple.UpdateJsonBody,
				Objects:       []domain.Object{simple.Entity},
				ServiceFunc:   Update(tableName, updateOptions.callbacks),
			}, updateOptions.middlewares...)
		} else {
			binding.PutBind(binder, &binding.SimpleBindItem[any]{
				Path:          domainPath + "/update",
				ResponseFunc:  response.SendMsgResponse,
				RequestParams: simple.UpdateJsonBody,
				Objects:       []domain.Object{simple.Entity},
				ServiceFunc:   UpdateTx(tableName, updateOptions.callbacks),
			}, updateOptions.middlewares...)
		}
	}

	// 查询
	if !queryOptions.disable {
		binding.GetBind(binder, &binding.SimpleBindItem[response.InfosData[I]]{
			Path:          domainPath + "/query",
			ResponseFunc:  response.SendInfosResponse[I],
			RequestParams: simple.QueryQueryParams,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   Query[I](tableName, queryOptions.callbacks, queryOptions.conditionFieldCallback),
		}, queryOptions.middlewares...)
	}

	// 通过ID获取
	if !getByIDOptions.disable {
		binding.GetBind(binder, &binding.SimpleBindItem[I]{
			Path:          domainPath + "/get",
			ResponseFunc:  response.SendInfoResponse[I],
			RequestParams: simple.GetByIDQueryParams,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   GetByID[I](tableName, getByIDOptions.callbacks),
		}, getByIDOptions.middlewares...)
	}
}

func BindSimple[I any](binder *binding.Binder, simple *Simple[I], opts ...any) {
	createOptions := new(CreateOptions)
	deleteOptions := new(DeleteOptions)
	updateOptions := new(UpdateOptions)
	queryOptions := new(QueryOptions[I])
	getByIDOptions := new(GetByIDOptions[I])

	for _, opt := range opts {
		switch o := opt.(type) {
		case CreateOption:
			o(createOptions)
		case DeleteOption:
			o(deleteOptions)
		case UpdateOption:
			o(updateOptions)
		case QueryOption[I]:
			o(queryOptions)
		case GetByIDOption[I]:
			o(getByIDOptions)
		default:
			continue
		}
	}

	simple.createOptions = createOptions
	simple.deleteOptions = deleteOptions
	simple.updateOptions = updateOptions
	simple.queryOptions = queryOptions
	simple.getByIDOptions = getByIDOptions

	simple.bind(binder)
}

type CreateOption func(options *CreateOptions)
type DeleteOption func(options *DeleteOptions)
type UpdateOption func(options *UpdateOptions)
type QueryOption[I any] func(options *QueryOptions[I])
type GetByIDOption[I any] func(options *GetByIDOptions[I])

type CreateOptions struct {
	// 关闭创建
	disable bool

	// 创建是否使用事务
	needTx bool

	// 创建回调
	callbacks *CreateCallbacks

	// 创建中间件
	middlewares []api.Handler
}

type DeleteOptions struct {
	// 关闭删除
	disable bool

	// 删除是否使用事务
	needTx bool

	// 删除回调
	callbacks *DeleteCallbacks

	// 删除中间件
	middlewares []api.Handler
}

type UpdateOptions struct {
	// 关闭更新
	disable bool

	// 更新是否使用事务
	needTx bool

	// 更新回调
	callbacks *UpdateCallbacks

	// 更新中间件
	middlewares []api.Handler
}

type QueryOptions[I any] struct {
	// 关闭查询
	disable bool

	// 查询条件构造回调
	conditionFieldCallback ConditionFieldCallback

	// 查询回调
	callbacks *QueryCallbacks[I]

	// 查询中间件
	middlewares []api.Handler
}

type GetByIDOptions[I any] struct {
	// 关闭根据ID查询
	disable bool

	// 根据ID查询回调
	callbacks *GetByIDCallbacks[I]

	// 根据ID查询中间件
	middlewares []api.Handler
}

func WithDisableCreate() CreateOption {
	return func(options *CreateOptions) {
		options.disable = true
	}
}

func WithCreateTx() CreateOption {
	return func(options *CreateOptions) {
		options.needTx = true
	}
}

func WithCreateCallbacks(callbacks *CreateCallbacks) CreateOption {
	return func(options *CreateOptions) {
		options.callbacks = callbacks
	}
}

func WithCreateMiddlewares(middlewares []api.Handler) CreateOption {
	return func(options *CreateOptions) {
		options.middlewares = middlewares
	}
}

func WithDisableDelete() DeleteOption {
	return func(options *DeleteOptions) {
		options.disable = true
	}
}

func WithDeleteTx() DeleteOption {
	return func(options *DeleteOptions) {
		options.needTx = true
	}
}

func WithDeleteCallbacks(callbacks *DeleteCallbacks) DeleteOption {
	return func(options *DeleteOptions) {
		options.callbacks = callbacks
	}
}

func WithDeleteMiddlewares(middlewares []api.Handler) DeleteOption {
	return func(options *DeleteOptions) {
		options.middlewares = middlewares
	}
}

func WithDisableUpdate() UpdateOption {
	return func(options *UpdateOptions) {
		options.disable = true
	}
}

func WithUpdateTx() UpdateOption {
	return func(options *UpdateOptions) {
		options.needTx = true
	}
}

func WithUpdateCallbacks(callbacks *UpdateCallbacks) UpdateOption {
	return func(options *UpdateOptions) {
		options.callbacks = callbacks
	}
}

func WithUpdateMiddlewares(middlewares []api.Handler) UpdateOption {
	return func(options *UpdateOptions) {
		options.middlewares = middlewares
	}
}

func WithDisableQuery[I any]() QueryOption[I] {
	return func(options *QueryOptions[I]) {
		options.disable = true
	}
}

func WithQueryConditionFieldCallback[I any](callback ConditionFieldCallback) QueryOption[I] {
	return func(options *QueryOptions[I]) {
		options.conditionFieldCallback = callback
	}
}

func WithQueryCallbacks[I any](callbacks *QueryCallbacks[I]) QueryOption[I] {
	return func(options *QueryOptions[I]) {
		options.callbacks = callbacks
	}
}

func WithQueryMiddlewares[I any](middlewares []api.Handler) QueryOption[I] {
	return func(options *QueryOptions[I]) {
		options.middlewares = middlewares
	}
}

func WithDisableGetByID[I any]() GetByIDOption[I] {
	return func(options *GetByIDOptions[I]) {
		options.disable = true
	}
}

func WithGetByIDCallbacks[I any](callbacks *GetByIDCallbacks[I]) GetByIDOption[I] {
	return func(options *GetByIDOptions[I]) {
		options.callbacks = callbacks
	}
}

func WithGetByIDMiddlewares[I any](middlewares []api.Handler) GetByIDOption[I] {
	return func(options *GetByIDOptions[I]) {
		options.middlewares = middlewares
	}
}