baize/convenient/entity/simple.go

package entity

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"
)

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

	// 表名
	TableName string

	// 选择要使用的数据库Executor
	// DBExecutorOperations operations 数据库操作
	// DBExecutorDataService data_service 数据服务
	DBExecutorType string

	// URL领域相对路径，如/class，后面会自动补充
	DomainPath string

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

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

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

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

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

	options *Options[I]
}

func (simple *Simple[I]) bind(binder *binding.Binder) {
	dbExecutor := binder.ChooseDBExecutor(simple.DBExecutorType)
	options := simple.options

	// 创建
	if !options.createNeedTx {
		binding.PostBind(binder, &binding.SimpleBindItem[string]{
			Path:          simple.DomainPath + "/create",
			ResponseFunc:  response.SendIDResponse[string],
			RequestParams: simple.CreateJsonBody,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   Create(simple.TableName, dbExecutor, options.createCallbacks),
		}, options.createMiddlewares...)
	} else {
		binding.PostBind(binder, &binding.SimpleBindItem[string]{
			Path:          simple.DomainPath + "/create",
			ResponseFunc:  response.SendIDResponse[string],
			RequestParams: simple.CreateJsonBody,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   CreateTx(simple.TableName, dbExecutor, options.createCallbacks),
		}, options.createMiddlewares...)
	}

	// 删除
	if !options.deleteNeedTx {
		binding.DeleteBind(binder, &binding.SimpleBindItem[any]{
			Path:          simple.DomainPath + "/:id/delete",
			ResponseFunc:  response.SendMsgResponse,
			RequestParams: simple.DeleteQueryParams,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   Delete(simple.TableName, dbExecutor, options.deleteCallbacks),
		}, options.deleteMiddlewares...)
	} else {
		binding.DeleteBind(binder, &binding.SimpleBindItem[any]{
			Path:          simple.DomainPath + "/:id/delete",
			ResponseFunc:  response.SendMsgResponse,
			RequestParams: simple.DeleteQueryParams,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   DeleteTx(simple.TableName, dbExecutor, options.deleteCallbacks),
		}, options.deleteMiddlewares...)
	}

	// 修改
	if !options.updateNeedTx {
		binding.PutBind(binder, &binding.SimpleBindItem[any]{
			Path:          simple.DomainPath + "/update",
			ResponseFunc:  response.SendMsgResponse,
			RequestParams: simple.UpdateJsonBody,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   Update(simple.TableName, dbExecutor, options.updateCallbacks),
		}, options.updateMiddlewares...)
	} else {
		binding.PutBind(binder, &binding.SimpleBindItem[any]{
			Path:          simple.DomainPath + "/update",
			ResponseFunc:  response.SendMsgResponse,
			RequestParams: simple.UpdateJsonBody,
			Objects:       []domain.Object{simple.Entity},
			ServiceFunc:   UpdateTx(simple.TableName, dbExecutor, options.updateCallbacks),
		}, options.updateMiddlewares...)
	}

	// 查询
	binding.GetBind(binder, &binding.SimpleBindItem[response.InfosData[I]]{
		Path:          simple.DomainPath + "/query",
		ResponseFunc:  response.SendInfosResponse[I],
		RequestParams: simple.QueryParams,
		Objects:       []domain.Object{simple.Entity},
		ServiceFunc:   Query[I](simple.TableName, dbExecutor, options.queryCallbacks, options.queryConditionFieldCallback),
	}, options.queryMiddlewares...)

	// 通过ID获取
	binding.GetBind(binder, &binding.SimpleBindItem[I]{
		Path:          simple.DomainPath + "/get",
		ResponseFunc:  response.SendInfoResponse[I],
		RequestParams: simple.GetByIDQueryParams,
		Objects:       []domain.Object{simple.Entity},
		ServiceFunc:   GetByID[I](simple.TableName, dbExecutor, options.getByIDCallbacks),
	}, options.getByIDMiddlewares...)
}

func BindSimple[I any](binder *binding.Binder, simple *Simple[I], opts ...Option[I]) {
	options := new(Options[I])

	for _, opt := range opts {
		opt(options)
	}

	simple.options = options

	simple.bind(binder)
}

type Option[I any] func(options *Options[I])

type Options[I any] struct {
	// 创建是否使用事务
	createNeedTx bool

	// 创建回调
	createCallbacks *Callbacks[string]

	// 创建中间件
	createMiddlewares []api.Handler

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

	// 删除回调
	deleteCallbacks *Callbacks[any]

	// 删除中间件
	deleteMiddlewares []api.Handler

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

	// 更新回调
	updateCallbacks *Callbacks[any]

	// 更新中间件
	updateMiddlewares []api.Handler

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

	// 查询回调
	queryCallbacks *Callbacks[response.InfosData[I]]

	// 查询中间件
	queryMiddlewares []api.Handler

	// 根据ID查询回调
	getByIDCallbacks *Callbacks[I]

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

func WithCreateTx[I any]() Option[I] {
	return func(options *Options[I]) {
		options.createNeedTx = true
	}
}

func WithCreateCallbacks[I any](callbacks *Callbacks[string]) Option[I] {
	return func(options *Options[I]) {
		options.createCallbacks = callbacks
	}
}

func WithCreateMiddlewares[I any](middlewares []api.Handler) Option[I] {
	return func(options *Options[I]) {
		options.createMiddlewares = middlewares
	}
}

func WithDeleteTx[I any]() Option[I] {
	return func(options *Options[I]) {
		options.deleteNeedTx = true
	}
}

func WithDeleteCallbacks[I any](callbacks *Callbacks[any]) Option[I] {
	return func(options *Options[I]) {
		options.deleteCallbacks = callbacks
	}
}

func WithDeleteMiddlewares[I any](middlewares []api.Handler) Option[I] {
	return func(options *Options[I]) {
		options.deleteMiddlewares = middlewares
	}
}

func WithUpdateTx[I any]() Option[I] {
	return func(options *Options[I]) {
		options.updateNeedTx = true
	}
}

func WithUpdateCallbacks[I any](callbacks *Callbacks[any]) Option[I] {
	return func(options *Options[I]) {
		options.updateCallbacks = callbacks
	}
}

func WithUpdateMiddlewares[I any](middlewares []api.Handler) Option[I] {
	return func(options *Options[I]) {
		options.updateMiddlewares = middlewares
	}
}

func WithQueryConditionFieldCallback[I any](callback ConditionFieldCallback) Option[I] {
	return func(options *Options[I]) {
		options.queryConditionFieldCallback = callback
	}
}

func WithQueryCallbacks[I any](callbacks *Callbacks[response.InfosData[I]]) Option[I] {
	return func(options *Options[I]) {
		options.queryCallbacks = callbacks
	}
}

func WithQueryMiddlewares[I any](middlewares []api.Handler) Option[I] {
	return func(options *Options[I]) {
		options.queryMiddlewares = middlewares
	}
}

func WithGetByIDCallbacks[I any](callbacks *Callbacks[I]) Option[I] {
	return func(options *Options[I]) {
		options.getByIDCallbacks = callbacks
	}
}

func WithGetByIDMiddlewares[I any](middlewares []api.Handler) Option[I] {
	return func(options *Options[I]) {
		options.getByIDMiddlewares = middlewares
	}
}