initContext.go

package data

import (
	"reflect"
	"strings"

	"github.com/farseer-go/fs/container"
	"github.com/farseer-go/fs/core"
	"github.com/farseer-go/fs/flog"
	"github.com/farseer-go/fs/types"
)

// 继承多租户时，默认仓库可以找到正确的上下文
var getInternalContextType = reflect.TypeOf((*IGetInternalContext)(nil)).Elem()
var emptyGetInternalContextVal = reflect.New(getInternalContextType).Elem()

// IDbContext 数据库上下文
type IDbContext interface{}

// NewContext 数据库上下文初始化
// keyName：数据库配置名称，对应./farseer.yaml 中的 Database节点
// autoCreateTable：true表示自动创建表
// 同一个上下文生命周期内，共享一个orm client
func NewContext[TDbContext IDbContext](dbName string) *TDbContext {
	var context TDbContext
	InitContext(&context, dbName)

	return &context
}

// InitContext 数据库上下文初始化
// keyName：数据库配置名称
// autoCreateTable：true表示自动创建表
// 同一个上下文生命周期内，共享一个orm client
func InitContext[TDbContext IDbContext](repositoryContext *TDbContext, dbName string) {
	if dbName == "" {
		panic("dbName入参必须设置有效的值")
	}

	transaction := container.Resolve[core.ITransaction](dbName)
	if transaction == nil {
		flog.Panicf("初始化TDbContext失败，请确认./farseer.yaml配置文件中的Database.%s是否正确", dbName)
	}
	internalContextType := reflect.ValueOf(transaction.(*internalContext))
	contextValueOf := reflect.ValueOf(repositoryContext).Elem()

	// 是否有单独的获取上下文的实现（用于多租户时，默认仓库可以找到正确的上下文）
	getInternalContextVal := emptyGetInternalContextVal
	if reflect.TypeOf(repositoryContext).Implements(getInternalContextType) {
		getInternalContextVal = reflect.ValueOf(any(repositoryContext).(IGetInternalContext))
	}

	// 遍历上下文中的TableSet字段类型
	for i := 0; i < contextValueOf.NumField(); i++ {
		field := contextValueOf.Field(i)
		if field.CanSet() {
			// 找到TableSet字段类型
			_, isDataTableSet := types.IsDataTableSet(field)
			_, isDataDomainSet := types.IsDataDomainSet(field)
			// 初始化表名
			if isDataTableSet || isDataDomainSet {
				data := contextValueOf.Type().Field(i).Tag.Get("data")
				param := make(map[string]string)
				for _, kv := range strings.Split(data, ";") {
					if kv == "" {
						continue
					}
					arrKV := strings.Split(kv, "=")
					if len(arrKV) == 2 {
						param[arrKV[0]] = arrKV[1]
					} else {
						param[arrKV[0]] = ""
					}
				}
				if isDataDomainSet {
					// 再取DomainSet的子属性，并设置值
					field.Addr().MethodByName("Init").Call([]reflect.Value{internalContextType, reflect.ValueOf(param), getInternalContextVal})
				} else if isDataTableSet {
					// 再取tableSet的子属性，并设置值
					field.Addr().MethodByName("Init").Call([]reflect.Value{internalContextType, reflect.ValueOf(param)})
				}
			} else if field.Type().String() == "core.ITransaction" || field.Type().String() == "data.IInternalContext" {
				field.Set(internalContextType)
			}
		}
	}
}

// RegisterContext 注册上下文（临时生命周期）
func RegisterContext[TDbContext IDbContext](dbName string, autoCreateTable bool) {
	container.RegisterTransient(func() IDbContext {
		var context TDbContext
		InitContext(&context, dbName)
		return &context
	}, dbName)
}

// GetContext 获取上下文实例（每次获取都会创建一个实例）
func GetContext[TDbContext IDbContext](dbName string) *TDbContext {
	return container.Resolve[IDbContext](dbName).(*TDbContext)
}