baize/domain/object.go

package domain

import (
	"git.sxidc.com/go-framework/baize/infrastructure/logger"
	"git.sxidc.com/service-supports/fserr"
	"git.sxidc.com/service-supports/fslog"
	"reflect"
)

type Object interface{}

func SetField[T any](object Object, fieldName string, value T) {
	fieldValue, err := getObjectFieldValue(object, fieldName)
	if err != nil {
		logger.GetInstance().Error(err)
		return
	}

	if !fieldValue.IsValid() || !fieldValue.CanSet() {
		err := fserr.New("object字段" + fieldName + "无法赋值")
		logger.GetInstance().Error(err)
		return
	}

	fieldValue.Set(reflect.ValueOf(value))
}

func Field[T any](object Object, fieldName string) T {
	var zero T

	fieldValue, err := getObjectFieldValue(object, fieldName)
	if err != nil {
		logger.GetInstance().Error(err)
		return zero
	}

	if !fieldValue.IsValid() {
		err := fserr.New("object字段" + fieldName + "无法赋值")
		logger.GetInstance().Error(err)
		return zero
	}

	retValue, ok := fieldValue.Interface().(T)
	if !ok {
		err := fserr.New("object字段" + fieldName + "无法转换类型")
		logger.GetInstance().Error(err)
		return zero
	}

	return retValue
}

func ToConcreteObject[T Object](object Object) T {
	concrete, ok := object.(T)
	if !ok {
		fslog.Error(fserr.New("领域对象转化失败"))

		var zero T
		return zero
	}

	return concrete
}

func getObjectFieldValue(object Object, fieldName string) (*reflect.Value, error) {
	objectType := reflect.TypeOf(object)

	if objectType.Kind() != reflect.Ptr {
		return nil, fserr.New("object必须是结构指针类型")
	}

	if objectType.Kind() == reflect.Ptr && objectType.Elem().Kind() != reflect.Struct {
		return nil, fserr.New("object必须是结构指针类型")
	}

	objectValue := reflect.ValueOf(object)
	if objectType.Kind() == reflect.Ptr && !objectValue.IsValid() {
		return nil, fserr.New("object为nil")
	}

	fieldValue := objectValue.Elem().FieldByName(fieldName)

	return &fieldValue, nil
}