ds-sdk/sql/sql_mapping.go

package sql

import (
	"errors"
	"github.com/iancoleman/strcase"
	"reflect"
	"strings"
	"time"
)

const (
	defaultKeyColumnName               = "id"
	sqlMappingTagPartSeparator         = ";"
	sqlMappingTagPartKeyValueSeparator = ":"
)

const (
	sqlMappingTagKey      = "sqlmapping"
	sqlMappingIgnore      = "-"
	sqlMappingColumn      = "column"
	sqlMappingKey         = "key"
	sqlMappingNotUpdate   = "notUpdate"
	sqlMappingUpdateClear = "updateClear"
	sqlMappingAes         = "aes"
)

type Mapping struct {
	MappingElement map[string]any
}

func ParseSqlMapping(e any) (*Mapping, error) {
	if e == nil {
		return nil, errors.New("没有传递实体")
	}

	entityType := reflect.TypeOf(e)
	if entityType.Kind() == reflect.Ptr {
		entityType = entityType.Elem()
	}

	if entityType.Kind() != reflect.Struct {
		return nil, errors.New("传递的实体不是结构类型")
	}

	entityValue := reflect.ValueOf(e)
	if entityValue.Kind() == reflect.Ptr {
		entityValue = entityValue.Elem()
	}

	sqlMapping := new(Mapping)
	sqlMapping.MappingElement = make(map[string]any)

	fieldNum := entityType.NumField()
	for i := 0; i < fieldNum; i++ {
		field := entityType.Field(i)
		fieldValue := entityValue.Field(i)

		element, err := parseSqlMappingElement(field, fieldValue)
		if err != nil {
			return nil, err
		}

		if element == nil {
			continue
		}

		sqlMapping.MappingElement[field.Name] = element
	}

	return sqlMapping, nil
}

type MappingStruct struct {
	MappingTypesAndValues
}

type MappingColumn struct {
	Name           string
	IsKey          bool
	CanUpdate      bool
	CanUpdateClear bool
	AESKey         string

	MappingTypesAndValues
}

type MappingTypesAndValues struct {
	// 原字段的反射结构
	OriginFieldType  reflect.Type
	OriginFieldValue reflect.Value

	// 值类型的反射结构
	FieldTypeElem  reflect.Type
	FieldValueElem reflect.Value
}

func parseSqlMappingElement(field reflect.StructField, fieldValue reflect.Value) (any, error) {
	fieldValueTypeElem := field.Type
	if field.Type.Kind() == reflect.Ptr {
		fieldValueTypeElem = field.Type.Elem()
	}

	fieldValueElem := fieldValue
	if fieldValue.Kind() == reflect.Ptr {
		if !fieldValue.IsValid() || fieldValue.IsNil() {
			fieldValue.Set(reflect.New(fieldValueTypeElem).Elem().Addr())
		}

		fieldValueElem = fieldValue.Elem()
	}

	if fieldValueTypeElem.Kind() == reflect.Struct && fieldValueTypeElem != reflect.TypeOf(time.Time{}) {
		return &MappingStruct{
			MappingTypesAndValues: MappingTypesAndValues{
				OriginFieldType:  field.Type,
				OriginFieldValue: fieldValue,
				FieldTypeElem:    fieldValueTypeElem,
				FieldValueElem:   fieldValueElem,
			},
		}, nil
	}

	sqlColumn := &MappingColumn{
		Name:           strcase.ToSnake(field.Name),
		IsKey:          false,
		CanUpdate:      true,
		CanUpdateClear: false,
		AESKey:         "",

		MappingTypesAndValues: MappingTypesAndValues{
			OriginFieldType:  field.Type,
			OriginFieldValue: fieldValue,
			FieldTypeElem:    fieldValueTypeElem,
			FieldValueElem:   fieldValueElem,
		},
	}

	if sqlColumn.Name == defaultKeyColumnName {
		sqlColumn.IsKey = true
		sqlColumn.CanUpdate = false
	}

	sqlMappingTag, ok := field.Tag.Lookup(sqlMappingTagKey)
	if !ok {
		return sqlColumn, nil
	}

	if sqlMappingTag == sqlMappingIgnore {
		return nil, nil
	}

	sqlMappingParts := strings.Split(sqlMappingTag, sqlMappingTagPartSeparator)
	if sqlMappingParts != nil || len(sqlMappingParts) != 0 {
		for _, sqlMappingPart := range sqlMappingParts {
			sqlPartKeyValue := strings.SplitN(strings.TrimSpace(sqlMappingPart), sqlMappingTagPartKeyValueSeparator, 2)
			switch sqlPartKeyValue[0] {
			case sqlMappingColumn:
				sqlColumn.Name = strings.TrimSpace(sqlPartKeyValue[1])
			case sqlMappingKey:
				sqlColumn.IsKey = true
				sqlColumn.CanUpdate = false
			case sqlMappingNotUpdate:
				sqlColumn.CanUpdate = false
			case sqlMappingUpdateClear:
				sqlColumn.CanUpdateClear = true
			case sqlMappingAes:
				if len(strings.TrimSpace(sqlPartKeyValue[1])) != 32 {
					return nil, errors.New("AES密钥长度应该为32个字节")
				}

				sqlColumn.AESKey = strings.TrimSpace(sqlPartKeyValue[1])
			default:
				continue
			}
		}
	}

	return sqlColumn, nil
}