baize/convenient/gwtools/relation/one2one/one2one.go

package one2one

import (
	"git.sxidc.com/go-framework/baize/framework/gateway"
	"git.sxidc.com/go-tools/utils/template"
	"github.com/iancoleman/strcase"
	"net/http"
)

func BindSimple(builder *gateway.Builder, params *Simple, opts ...Option) {
	options := new(Options)

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

	params.options = options

	params.bind(builder)
}

type Simple struct {
	// 除去后缀的服务URL，如http://localhost:8080/example/api/v1
	ServiceVersionedUrl string

	// 左领域名称
	LeftDomainCamelName string

	// 右领域名称
	RightDomainCamelName string

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

func (simple *Simple) bind(builder *gateway.Builder) {
	options := simple.options

	leftDomainPath := "/" + strcase.ToLowerCamel(template.Id(simple.LeftDomainCamelName))
	rightDomainPath := "/" + strcase.ToLowerCamel(template.Id(simple.RightDomainCamelName))

	if !options.disableLeft {
		if !options.disableLeftUpdate {
			// 左到右更新
			builder.
				Url(http.MethodPost, leftDomainPath+rightDomainPath+"/update").
				Post(&gateway.PostRequest{
					Url: simple.ServiceVersionedUrl + leftDomainPath + rightDomainPath + "/update",
				}).
				Build()
		}

		if !options.disableLeftQuery {
			// 左到右查询
			builder.
				Url(http.MethodGet, leftDomainPath+rightDomainPath+"/query").
				Get(&gateway.GetRequest{
					Url: simple.ServiceVersionedUrl + leftDomainPath + rightDomainPath + "/query",
				}).
				Build()
		}

		if !options.disableLeftWithRightQuery {
			// 左到右查询，携带右方信息
			builder.
				Url(http.MethodGet, leftDomainPath+rightDomainPath+"/queryWith").
				Get(&gateway.GetRequest{
					Url:                 simple.ServiceVersionedUrl + leftDomainPath + rightDomainPath + "/queryWith",
					QueryParamsFormFunc: gateway.FormQueryParamsWithTenantIDAndUserIDFunc("tenantId", ""),
				}).
				Build()
		}
	}

	if !options.disableRight {
		if !options.disableRightUpdate {
			// 右到左更新
			builder.
				Url(http.MethodPost, rightDomainPath+leftDomainPath+"/update").
				Post(&gateway.PostRequest{
					Url: simple.ServiceVersionedUrl + rightDomainPath + leftDomainPath + "/update",
				}).
				Build()
		}

		if !options.disableRightQuery {
			// 右到左查询
			builder.
				Url(http.MethodGet, rightDomainPath+leftDomainPath+"/query").
				Get(&gateway.GetRequest{
					Url: simple.ServiceVersionedUrl + rightDomainPath + leftDomainPath + "/query",
				}).
				Build()
		}

		if !options.disableRightWithLeftQuery {
			// 右到左查询，携带左方信息
			builder.
				Url(http.MethodGet, rightDomainPath+leftDomainPath+"/queryWith").
				Get(&gateway.GetRequest{
					Url:                 simple.ServiceVersionedUrl + rightDomainPath + leftDomainPath + "/queryWith",
					QueryParamsFormFunc: gateway.FormQueryParamsWithTenantIDAndUserIDFunc("tenantId", ""),
				}).
				Build()
		}
	}
}

type Option func(options *Options)

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

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

	// 关闭左侧更新
	disableLeftUpdate bool

	// 关闭左侧查询
	disableLeftQuery bool

	// 关闭右侧更新
	disableRightUpdate bool

	// 关闭右侧查询
	disableRightQuery bool

	// 关闭左实体带右实体信息查询
	disableLeftWithRightQuery bool

	// 关闭右实体带左实体信息查询
	disableRightWithLeftQuery 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
	}
}

func WithDisableLeftWithRightQuery() Option {
	return func(options *Options) {
		options.disableLeftWithRightQuery = true
	}
}

func WithDisableRightWithLeftQuery() Option {
	return func(options *Options) {
		options.disableRightWithLeftQuery = true
	}
}