gf/database/gdb/gdb_converter.go

// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.

package gdb

import (
	"reflect"

	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/internal/json"
	"github.com/gogf/gf/v2/util/gconv"
)

// iVal is used for type assert api for Val().
type iVal interface {
	Val() any
}

var (
	// converter is the internal type converter for gdb.
	converter = gconv.NewConverter()
)

func init() {
	converter.RegisterAnyConverterFunc(
		sliceTypeConverterFunc,
		reflect.TypeOf([]string{}),
		reflect.TypeOf([]float32{}),
		reflect.TypeOf([]float64{}),
		reflect.TypeOf([]int{}),
		reflect.TypeOf([]int32{}),
		reflect.TypeOf([]int64{}),
		reflect.TypeOf([]uint{}),
		reflect.TypeOf([]uint32{}),
		reflect.TypeOf([]uint64{}),
	)
}

// GetConverter returns the internal type converter for gdb.
func GetConverter() gconv.Converter {
	return converter
}

func sliceTypeConverterFunc(from any, to reflect.Value) (err error) {
	v, ok := from.(iVal)
	if !ok {
		return nil
	}
	fromVal := v.Val()
	switch x := fromVal.(type) {
	case []byte:
		dst := to.Addr().Interface()
		err = json.Unmarshal(x, dst)
	case string:
		dst := to.Addr().Interface()
		err = json.Unmarshal([]byte(x), dst)
	default:
		fromType := reflect.TypeOf(fromVal)
		switch fromType.Kind() {
		case reflect.Slice:
			convertOption := gconv.ConvertOption{
				SliceOption:  gconv.SliceOption{ContinueOnError: true},
				MapOption:    gconv.MapOption{ContinueOnError: true},
				StructOption: gconv.StructOption{ContinueOnError: true},
			}
			dv, err := converter.ConvertWithTypeName(fromVal, to.Type().String(), convertOption)
			if err != nil {
				return err
			}
			to.Set(reflect.ValueOf(dv))
		default:
			err = gerror.Newf(
				`unsupported type converting from type "%T" to type "%T"`,
				fromVal, to,
			)
		}
	}
	return err
}