-
Notifications
You must be signed in to change notification settings - Fork 669
/
codec_registry.go
116 lines (99 loc) · 2.68 KB
/
codec_registry.go
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
114
115
116
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package avm
import (
"math"
"reflect"
"github.com/ava-labs/avalanchego/codec"
"github.com/ava-labs/avalanchego/codec/linearcodec"
"github.com/ava-labs/avalanchego/codec/reflectcodec"
"github.com/ava-labs/avalanchego/utils/logging"
"github.com/ava-labs/avalanchego/utils/timer"
"github.com/ava-labs/avalanchego/utils/wrappers"
"github.com/ava-labs/avalanchego/vms/secp256k1fx"
)
const (
codecVersion = 0
)
var (
_ codec.Registry = &codecRegistry{}
_ secp256k1fx.VM = &fxVM{}
)
type codecRegistry struct {
codecs []codec.Registry
index int
typeToIndex map[reflect.Type]int
}
func (cr *codecRegistry) RegisterType(val interface{}) error {
valType := reflect.TypeOf(val)
cr.typeToIndex[valType] = cr.index
errs := wrappers.Errs{}
for _, c := range cr.codecs {
errs.Add(c.RegisterType(val))
}
return errs.Err
}
// NewCodecs returns the genesis codec and the normal codec for the provided
// feature extensions.
func NewCodecs(fxs []Fx) (codec.Manager, codec.Manager, error) {
return newCustomCodecs(
make(map[reflect.Type]int),
&timer.Clock{},
logging.NoLog{},
fxs,
)
}
func newCustomCodecs(
typeToFxIndex map[reflect.Type]int,
clock *timer.Clock,
log logging.Logger,
fxs []Fx,
) (codec.Manager, codec.Manager, error) {
gc := linearcodec.New(reflectcodec.DefaultTagName, 1<<20)
c := linearcodec.NewDefault()
gcm := codec.NewManager(math.MaxInt32)
cm := codec.NewDefaultManager()
errs := wrappers.Errs{}
errs.Add(
c.RegisterType(&BaseTx{}),
c.RegisterType(&CreateAssetTx{}),
c.RegisterType(&OperationTx{}),
c.RegisterType(&ImportTx{}),
c.RegisterType(&ExportTx{}),
cm.RegisterCodec(codecVersion, c),
gc.RegisterType(&BaseTx{}),
gc.RegisterType(&CreateAssetTx{}),
gc.RegisterType(&OperationTx{}),
gc.RegisterType(&ImportTx{}),
gc.RegisterType(&ExportTx{}),
gcm.RegisterCodec(codecVersion, gc),
)
if errs.Errored() {
return nil, nil, errs.Err
}
vm := &fxVM{
typeToFxIndex: typeToFxIndex,
clock: clock,
log: log,
}
for i, fx := range fxs {
vm.codecRegistry = &codecRegistry{
codecs: []codec.Registry{gc, c},
index: i,
typeToIndex: vm.typeToFxIndex,
}
if err := fx.Initialize(vm); err != nil {
return nil, nil, err
}
}
return gcm, cm, nil
}
type fxVM struct {
typeToFxIndex map[reflect.Type]int
clock *timer.Clock
log logging.Logger
codecRegistry codec.Registry
}
func (vm *fxVM) Clock() *timer.Clock { return vm.clock }
func (vm *fxVM) CodecRegistry() codec.Registry { return vm.codecRegistry }
func (vm *fxVM) Logger() logging.Logger { return vm.log }