baize/convenient/relation/remote/simple.go

package remote

import (
	"fmt"
	"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/domain"
	"git.sxidc.com/go-framework/baize/framework/core/domain/entity"
	"github.com/iancoleman/strcase"
	"reflect"
)

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

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

	// 数据库Schema
	Schema string

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

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

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

	// 查询右实体关联使用的请求参数，注意是Query类型
	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()))

	leftRemote := false
	if reflect.TypeOf(new(LI)).Elem().Kind() == reflect.String {
		leftRemote = true
	}

	rightRemote := false
	if reflect.TypeOf(new(RI)).Elem().Kind() == reflect.String {
		rightRemote = true
	}

	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,
					leftRemote, leftTableName, simple.Left.DomainCNName(), leftRelationFieldName, leftRelationColumnName,
					rightRemote, rightTableName, rightRelationColumnName),
			})
		}

		if !options.disableLeftQuery {
			// 左到右查询
			if rightRemote {
				binding.GetBind(binder, &binding.SimpleBindItem[response.InfosData[string]]{
					Path:          leftDomainPath + rightDomainPath + "/query",
					ResponseFunc:  response.SendInfosResponse[string],
					RequestParams: simple.LeftQueryQueryParams,
					Objects:       []domain.Object{simple.Left},
					ServiceFunc:   QueryToRemote(middleTableName, leftRemote, leftTableName, leftRelationColumnName, rightRelationColumnName),
				})
			} else {
				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: QueryToExist[RI](middleTableName,
						leftRemote, 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,
					rightRemote, rightTableName, simple.Right.DomainCNName(), rightRelationFieldName, rightRelationColumnName,
					leftRemote, leftTableName, leftRelationColumnName),
			})
		}

		if !options.disableRightQuery {
			// 右到左查询
			if leftRemote {
				binding.GetBind(binder, &binding.SimpleBindItem[response.InfosData[string]]{
					Path:          rightDomainPath + leftDomainPath + "/query",
					ResponseFunc:  response.SendInfosResponse[string],
					RequestParams: simple.RightQueryQueryParams,
					Objects:       []domain.Object{simple.Right},
					ServiceFunc:   QueryToRemote(middleTableName, rightRemote, rightTableName, rightRelationColumnName, leftRelationColumnName),
				})
			} else {
				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: QueryToExist[LI](middleTableName,
						rightRemote, 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
	}
}