ds-sdk/sql/sql_tpl/value.go

package sql_tpl

import (
	"errors"
	"git.sxidc.com/go-tools/utils/encoding"
	"reflect"
	"strconv"
	"time"
)

const (
	timeWriteFormat = time.DateTime + ".000000 +08:00"
)

type AfterParsedStrValueOption func(strValue string) (string, error)

func WithAESKey(aesKey string) AfterParsedStrValueOption {
	return func(strValue string) (string, error) {
		encrypted, err := encoding.AESEncrypt(strValue, aesKey)
		if err != nil {
			return "", err
		}

		return "'" + encrypted + "'", nil
	}
}

func parseValue(value any, opts ...AfterParsedStrValueOption) (string, error) {
	valueValue := reflect.ValueOf(value)

	if !valueValue.IsValid() {
		return "", errors.New("无效值")
	}

	if valueValue.Kind() == reflect.Ptr && valueValue.IsNil() {
		return "", errors.New("空值")
	}

	if valueValue.Kind() == reflect.Ptr {
		valueValue = valueValue.Elem()
	}

	var parsedValue string

	switch v := valueValue.Interface().(type) {
	case string:
		parsedValue = v

		if opts == nil || len(opts) == 0 {
			return "'" + parsedValue + "'", nil
		}
	case bool:
		parsedValue = strconv.FormatBool(v)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case time.Time:
		parsedValue = v.Format(timeWriteFormat)

		if opts == nil || len(opts) == 0 {
			return "'" + parsedValue + "'", nil
		}
	case int:
		parsedValue = strconv.Itoa(v)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case int8:
		parsedValue = strconv.FormatInt(int64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case int16:
		parsedValue = strconv.FormatInt(int64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case int32:
		parsedValue = strconv.FormatInt(int64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case int64:
		parsedValue = strconv.FormatInt(v, 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case uint:
		parsedValue = strconv.FormatUint(uint64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case uint8:
		parsedValue = strconv.FormatUint(uint64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case uint16:
		parsedValue = strconv.FormatUint(uint64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case uint32:
		parsedValue = strconv.FormatUint(uint64(v), 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	case uint64:
		parsedValue = strconv.FormatUint(v, 10)

		if opts == nil || len(opts) == 0 {
			return parsedValue, nil
		}
	default:
		return "", errors.New("不支持的类型")
	}

	for _, opt := range opts {
		innerParsedValue, err := opt(parsedValue)
		if err != nil {
			return "", err
		}

		parsedValue = innerParsedValue
	}

	return parsedValue, nil
}