Function NormalizeVal

pgtest/framework.go:260–319

NormalizeVal normalizes values to the Doltgres type expects, so it can be used to convert the values using the given Doltgres type. This is used to normalize array types as the type conversion expects certain type values.

(typ *pgtype.Type, v any)

// convert the values using the given Doltgres type. This is used to normalize array
// types as the type conversion expects certain type values.
func NormalizeVal(typ *pgtype.Type, v any) any {
	switch strings.ToLower(typ.Name) {
	case "json":
		str, err := json.Marshal(v)
		if err != nil {
			panic(err)
		}
		return string(str)
	case "jsonb":
		bytes, err := defaultMap.Encode(typ.OID, pgtype.TextFormatCode, v, nil)
		if err != nil {
			panic(err)
		}
		var s string
		if err := defaultMap.Scan(typ.OID, pgtype.TextFormatCode, bytes, &s); err != nil {
			panic(err)
		}
		return s
	}

	switch val := v.(type) {
	case pgtype.Numeric:
		if val.NaN {
			return math.NaN()
		} else if val.InfinityModifier != pgtype.Finite {
			return math.Inf(int(val.InfinityModifier))
		} else if !val.Valid {
			return nil
		} else {
			d := decimalFromNumeric(val)
			return d
		}
	case pgtype.Time:
		// This value type is used for TIME type.
		var zero time.Time
		return zero.Add(time.Duration(val.Microseconds) * time.Microsecond)
	case pgtype.Interval:
		// This value type is used for INTERVAL type.
		// TODO(fan): Months
		var zero time.Time
		return zero.Add(time.Duration(val.Microseconds)*time.Microsecond).AddDate(0, 0, int(val.Days))
	case [16]byte:
		// This value type is used for UUID type.
		u, err := uuid.FromBytes(val[:])
		if err != nil {
			panic(err)
		}
		return u
	case []any:
		baseType := typ.Codec.(*pgtype.ArrayCodec).ElementType
		newVal := make([]any, len(val))
		for i, el := range val {
			newVal[i] = NormalizeVal(baseType, el)
		}
		return newVal
	}
	return v
}