ds-sdk/sql/parse_table_row.go

package sql

import (
	"errors"
	"fmt"
	"git.sxidc.com/go-tools/utils/encoding"
	"git.sxidc.com/go-tools/utils/reflectutils"
	"git.sxidc.com/go-tools/utils/strutils"
	"reflect"
	"strings"
	"time"
)

const (
	sqlResultTimeMicroFormat = "2006-01-02T15:04:05.000000+08:00"
	sqlResultTimeMilliFormat = "2006-01-02T15:04:05.000+08:00"
	sqlResultTimeSecFormat   = "2006-01-02T15:04:05+08:00"
)

func parseSqlTableRowTimeStr(timeStr string) (time.Time, error) {
	var layout string

	if strings.HasSuffix(timeStr, ".000000+08:00") {
		layout = sqlResultTimeMicroFormat
	} else if strings.HasSuffix(timeStr, ".000+08:00") {
		layout = sqlResultTimeMilliFormat
	} else {
		layout = sqlResultTimeSecFormat
	}

	return time.ParseInLocation(layout, timeStr, time.Local)
}

func ParseSqlTableRow(input any, output any) error {
	if input == nil || output == nil {
		return nil
	}

	// 输出的Type，可以是slice的指针或者是结构的指针
	outputType := reflect.TypeOf(output)
	if outputType.Kind() != reflect.Ptr {
		return errors.New("输出实体应该为结构的slice或者是结构的指针")
	}

	// 取元素类型
	if outputType.Kind() == reflect.Ptr {
		outputType = outputType.Elem()
	}

	// 检查元素类型是否为slice或者结构
	if outputType.Kind() != reflect.Slice && outputType.Kind() != reflect.Struct {
		return errors.New("输出实体应该为结构的slice或者是结构的指针")
	}

	// 如果输出类型为slice，则取slice元素类型
	outputElemType := outputType
	if outputElemType.Kind() == reflect.Slice {
		outputElemType = outputElemType.Elem()
	}

	// 校验元素类型是否为结构类型
	if outputElemType.Kind() != reflect.Struct {
		return errors.New("输出实体slice应该为结构的slice指针")
	}

	// 构造需要遍历的tableRows
	tableRows, ok := input.([]map[string]any)
	if !ok {
		tableRow, ok := input.(map[string]any)
		if !ok {
			return errors.New("输入数据应该为[]map[string]any或[]map[string]any")
		}

		tableRows = []map[string]any{tableRow}
	}

	// 构造输出实体slice
	outputEntities := reflect.MakeSlice(reflect.SliceOf(outputElemType), 0, 0)

	for _, tableRow := range tableRows {
		// 构造输出实体
		outputEntityValue := reflect.New(outputElemType).Elem().Addr()
		outputEntity := outputEntityValue.Interface()

		err := formOutputEntity(tableRow, outputEntity)
		if err != nil {
			return err
		}

		// 保存输出实体
		outputEntities = reflect.Append(outputEntities, outputEntityValue.Elem())
	}

	// 将输出实体赋值给输出指针变量
	outputValue := reflect.Indirect(reflect.ValueOf(output))

	if !outputValue.CanSet() {
		return nil
	}

	if outputType.Kind() == reflect.Slice {
		outputValue.Set(outputEntities)
	} else {
		outputValue.Set(outputEntities.Index(0))
	}

	return nil
}

func formOutputEntity(tableRow map[string]any, outputEntity any) error {
	sqlResult, err := ParseSqlResult(outputEntity)
	if err != nil {
		return err
	}

	for fieldName, resultElement := range sqlResult.ResultElement {
		switch element := resultElement.(type) {
		case *ResultStruct:
			err := formOutputEntity(tableRow, element.FieldValueElem.Addr().Interface())
			if err != nil {
				return err
			}
		case *ResultColumn:
			tableRowValue, ok := tableRow[element.Name]
			if !ok {
				continue
			}

			// 构造结构字段，如果结构字段是指针且为nil，需要构造元素
			fieldTypeElem := element.FieldTypeElem
			fieldValueElem := element.FieldValueElem
			outputKind := reflectutils.GroupValueKind(fieldValueElem)

			if !fieldValueElem.CanSet() {
				continue
			}

			switch outputKind {
			case reflect.Bool:
				err := reflectutils.AssignBoolValue(tableRowValue, fieldValueElem)
				if err != nil {
					return err
				}
			case reflect.String:
				strValue := tableRowValue.(string)

				if strutils.IsStringNotEmpty(element.ParseTime) {
					parsedTime, err := parseSqlTableRowTimeStr(strValue)
					if err != nil {
						return err
					}

					strValue = parsedTime.Format(element.ParseTime)
				} else if strutils.IsStringNotEmpty(element.AESKey) {
					if strutils.IsStringNotEmpty(strValue) {
						decryptedValue, err := encoding.AESDecrypt(strValue, element.AESKey)
						if err != nil {
							return err
						}

						strValue = decryptedValue
					}
				}

				err = reflectutils.AssignStringValue(strValue, fieldValueElem)
				if err != nil {
					return err
				}
			case reflect.Int64:
				err := reflectutils.AssignIntValue(tableRowValue, fieldValueElem)
				if err != nil {
					return err
				}
			case reflect.Uint64:
				err := reflectutils.AssignUintValue(tableRowValue, fieldValueElem)
				if err != nil {
					return err
				}
			case reflect.Float64:
				err := reflectutils.AssignFloatValue(tableRowValue, fieldValueElem)
				if err != nil {
					return err
				}
			case reflect.Struct:
				if fieldValueElem.Type() == reflect.TypeOf(time.Time{}) {
					parsedTime, err := parseSqlTableRowTimeStr(tableRowValue.(string))
					if err != nil {
						return err
					}

					fieldValueElem.Set(reflect.ValueOf(parsedTime))
					continue
				}

				return fmt.Errorf("字段: %s 列: %s 不支持的类型: %s",
					fieldName, element.Name, fieldTypeElem.String())
			case reflect.Slice:
				if fieldTypeElem.Elem().Kind() != reflect.String {
					return errors.New("slice仅支持[]string")
				}

				strValue, ok := tableRowValue.(string)
				if !ok {
					return errors.New("slice仅支持[]string")
				}

				strParts := strings.Split(strValue, element.SplitWith)
				if strParts == nil || len(strParts) == 0 {
					return nil
				}

				valSlice := fieldValueElem
				if valSlice.IsNil() {
					valSlice = reflect.MakeSlice(fieldTypeElem, 0, 0)
				}

				for _, strPart := range strParts {
					valSlice = reflect.Append(valSlice, reflect.ValueOf(strPart))
				}

				fieldValueElem.Set(valSlice)
			default:
				return fmt.Errorf("字段: %s 列: %s 不支持的类型: %s",
					fieldName, element.Name, fieldTypeElem.String())
			}
		default:
			return errors.New("不支持的元素类型")
		}
	}

	return nil
}