baize/convenient/relation/many2many/simple.go

package many2many

import (
	"fmt"
	"git.sxidc.com/go-framework/baize/convenient/binding"
	"git.sxidc.com/go-framework/baize/convenient/binding/request"
	"git.sxidc.com/go-framework/baize/convenient/binding/response"
	"git.sxidc.com/go-framework/baize/framwork/domain"
	"git.sxidc.com/go-framework/baize/framwork/domain/entity"
	"github.com/iancoleman/strcase"
)

// Simple 关联的Bind参数
// LI 为左边实体的Info类型
// RI 为右边实体的Info类型
type Simple[LI any, RI any] struct {
	// 左领域实体，注意是Entity类型
	Left entity.Entity

	// 右领域实体，注意是Entity类型
	Right entity.Entity

	// 数据库schema
	Schema string

	// 更新左实体关联使用的请求参数
	LeftUpdateJsonBody request.Params

	// 查询左实体关联使用的请求参数，注意是QueryWithID类型
	LeftQueryQueryParams request.QueryWithIDRequestParams

	// 更新右实体关联使用的请求参数
	RightUpdateJsonBody request.Params

	// 查询右实体关联使用的请求参数，注意是QueryWithID类型
	RightQueryQueryParams request.QueryWithIDRequestParams

	// 可选配置项，通过WithXXX配置
	options *Options
}

func (simple *Simple[LI, RI]) bind(binder *binding.Binder) {
	options := simple.options

	leftDomainPath := domain.RelativeDomainPath(simple.Left)
	rightDomainPath := domain.RelativeDomainPath(simple.Right)

	leftRelationFieldName := fmt.Sprintf("%sIDs", simple.Right.DomainCamelName())
	rightRelationFieldName := fmt.Sprintf("%sIDs", simple.Left.DomainCamelName())

	middleTableName := domain.SnakeDomainName(simple.Left) + "_and_" + domain.SnakeDomainName(simple.Right)
	leftTableName := domain.TableName(simple.Schema, simple.Left)
	leftRelationColumnName := fmt.Sprintf("%s_id", strcase.ToSnake(simple.Right.DomainCamelName()))
	rightTableName := domain.TableName(simple.Schema, simple.Right)
	rightRelationColumnName := fmt.Sprintf("%s_id", strcase.ToSnake(simple.Left.DomainCamelName()))

	if !options.disableLeft {
		if !options.disableLeftUpdate {
			// 左到右更新
			binding.PostBind(binder, &binding.SimpleBindItem[any]{
				Path:          leftDomainPath + rightDomainPath + "/update",
				ResponseFunc:  response.SendMsgResponse,
				RequestParams: simple.LeftUpdateJsonBody,
				Objects:       []domain.Object{simple.Left},
				ServiceFunc: Update(middleTableName,
					leftTableName, simple.Left.DomainCNName(), leftRelationFieldName, leftRelationColumnName,
					rightTableName, rightRelationColumnName),
			})
		}

		if !options.disableLeftQuery {
			// 左到右查询
			binding.GetBind(binder, &binding.SimpleBindItem[response.InfosData[RI]]{
				Path:          leftDomainPath + rightDomainPath + "/query",
				ResponseFunc:  response.SendInfosResponse[RI],
				RequestParams: simple.LeftQueryQueryParams,
				Objects:       []domain.Object{simple.Left},
				ServiceFunc: Query[RI](middleTableName,
					leftTableName, leftRelationColumnName,
					rightTableName, rightRelationColumnName),
			})
		}
	}

	if !options.disableRight {
		if !options.disableRightUpdate {
			// 右到左更新
			binding.PostBind(binder, &binding.SimpleBindItem[any]{
				Path:          rightDomainPath + leftDomainPath + "/update",
				ResponseFunc:  response.SendMsgResponse,
				RequestParams: simple.RightUpdateJsonBody,
				Objects:       []domain.Object{simple.Right},
				ServiceFunc: Update(middleTableName,
					rightTableName, simple.Right.DomainCNName(), rightRelationFieldName, rightRelationColumnName,
					leftTableName, leftRelationColumnName),
			})
		}

		if !options.disableRightQuery {
			// 右到左查询
			binding.GetBind(binder, &binding.SimpleBindItem[response.InfosData[LI]]{
				Path:          rightDomainPath + leftDomainPath + "/query",
				ResponseFunc:  response.SendInfosResponse[LI],
				RequestParams: simple.RightQueryQueryParams,
				Objects:       []domain.Object{simple.Right},
				ServiceFunc: Query[LI](middleTableName,
					rightTableName, rightRelationColumnName,
					leftTableName, leftRelationColumnName),
			})
		}
	}
}

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

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

	simple.options = options

	simple.bind(binder)
}

type Option func(options *Options)

type Options struct {
	// 关闭左侧到右侧关联
	disableLeft bool

	// 关闭右侧到左侧关联
	disableRight bool

	// 关闭左侧更新
	disableLeftUpdate bool

	// 关闭左侧查询
	disableLeftQuery bool

	// 关闭右侧更新
	disableRightUpdate bool

	// 关闭右侧查询
	disableRightQuery bool
}

func WithDisableLeft() Option {
	return func(options *Options) {
		options.disableLeft = true
	}
}

func WithDisableRight() Option {
	return func(options *Options) {
		options.disableRight = true
	}
}

func WithDisableLeftUpdate() Option {
	return func(options *Options) {
		options.disableLeftUpdate = true
	}
}

func WithDisableLeftQuery() Option {
	return func(options *Options) {
		options.disableLeftQuery = true
	}
}

func WithDisableRightUpdate() Option {
	return func(options *Options) {
		options.disableRightUpdate = true
	}
}

func WithDisableRightQuery() Option {
	return func(options *Options) {
		options.disableRightQuery = true
	}
}