avro/typeconverter_test.go at main · hamba/avro · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package avro_test

import (
	"fmt"
	"math/big"

	"github.com/hamba/avro/v2"
)

var (
	boolConverter = avro.TypeConversionFuncs{
		AvroType: avro.Boolean,
		DecoderTypeConversion: func(in any, _ avro.Schema) (any, error) {
			b := in.(bool)
			if b {
				return "yes", nil
			} else {
				return "no", nil
			}
		},
	}

	floatToAvroInt = func(in any, _ avro.Schema) (any, error) {
		switch v := in.(type) {
		case float32:
			if float32(int(v)) != v {
				return 0, fmt.Errorf("%v is not an integer", in)
			}
			return int(v), nil
		case float64:
			if float64(int(v)) != v {
				return 0, fmt.Errorf("%v is not an integer", in)
			}
			return int(v), nil
		case *float64:
			if float64(int(*v)) != *v {
				return 0, fmt.Errorf("%v is not an integer", *v)
			}
			return int(*v), nil
		}
		return in, nil
	}

	intConverter = avro.TypeConversionFuncs{
		AvroType:              avro.Int,
		EncoderTypeConversion: floatToAvroInt,
	}

	fixedDecimalConverter = avro.TypeConversionFuncs{
		AvroType:        avro.Fixed,
		AvroLogicalType: avro.Decimal,
		EncoderTypeConversion: func(in any, _ avro.Schema) (any, error) {
			switch v := in.(type) {
			case string:
				val, _ := new(big.Rat).SetString(v)
				return val, nil
			}
			return in, nil
		},
		DecoderTypeConversion: func(in any, _ avro.Schema) (any, error) {
			r := in.(*big.Rat)
			f, _ := r.Float64()
			return f, nil
		},
	}

	unionConverter = avro.TypeConversionFuncs{
		AvroType: avro.Union,
		EncoderTypeConversion: func(in any, schema avro.Schema) (any, error) {
			union := schema.(*avro.UnionSchema)
			if union.Contains(avro.String) {
				return fmt.Sprintf("%v", in), nil
			}
			if union.Contains(avro.Int) {
				return floatToAvroInt(in, schema)
			}
			return in, nil
		},
		DecoderTypeConversion: func(in any, _ avro.Schema) (any, error) {
			switch v := in.(type) {
			case bool:
				if v {
					return "yes", nil
				} else {
					return "no", nil
				}
			case *big.Rat:
				f, _ := v.Float64()
				return f, nil
			}

			return in, nil
		},
	}
)

func nonConverter(typ avro.Type) avro.TypeConversionFuncs {
	return avro.TypeConversionFuncs{
		AvroType: typ,
	}
}

func errorConverter(typ avro.Type, err error) avro.TypeConversionFuncs {
	return avro.TypeConversionFuncs{
		AvroType: typ,
		EncoderTypeConversion: func(in any, _ avro.Schema) (any, error) {
			return nil, err
		},
		DecoderTypeConversion: func(in any, _ avro.Schema) (any, error) {
			return nil, err
		},
	}
}