forked from ava-labs/coreth
-
Notifications
You must be signed in to change notification settings - Fork 0
/
syncable.go
93 lines (76 loc) · 2.54 KB
/
syncable.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
// (c) 2021-2022, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package message
import (
"context"
"fmt"
"github.com/MetalBlockchain/metalgo/ids"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/crypto"
"github.com/MetalBlockchain/metalgo/snow/engine/snowman/block"
)
var _ block.StateSummary = &SyncSummary{}
// SyncSummary provides the information necessary to sync a node starting
// at the given block.
type SyncSummary struct {
BlockNumber uint64 `serialize:"true"`
BlockHash common.Hash `serialize:"true"`
BlockRoot common.Hash `serialize:"true"`
AtomicRoot common.Hash `serialize:"true"`
summaryID ids.ID
bytes []byte
acceptImpl func(SyncSummary) (block.StateSyncMode, error)
}
func NewSyncSummaryFromBytes(summaryBytes []byte, acceptImpl func(SyncSummary) (block.StateSyncMode, error)) (SyncSummary, error) {
summary := SyncSummary{}
if codecVersion, err := Codec.Unmarshal(summaryBytes, &summary); err != nil {
return SyncSummary{}, err
} else if codecVersion != Version {
return SyncSummary{}, fmt.Errorf("failed to parse syncable summary due to unexpected codec version (%d != %d)", codecVersion, Version)
}
summary.bytes = summaryBytes
summaryID, err := ids.ToID(crypto.Keccak256(summaryBytes))
if err != nil {
return SyncSummary{}, err
}
summary.summaryID = summaryID
summary.acceptImpl = acceptImpl
return summary, nil
}
func NewSyncSummary(blockHash common.Hash, blockNumber uint64, blockRoot common.Hash, atomicRoot common.Hash) (SyncSummary, error) {
summary := SyncSummary{
BlockNumber: blockNumber,
BlockHash: blockHash,
BlockRoot: blockRoot,
AtomicRoot: atomicRoot,
}
bytes, err := Codec.Marshal(Version, &summary)
if err != nil {
return SyncSummary{}, err
}
summary.bytes = bytes
summaryID, err := ids.ToID(crypto.Keccak256(bytes))
if err != nil {
return SyncSummary{}, err
}
summary.summaryID = summaryID
return summary, nil
}
func (s SyncSummary) Bytes() []byte {
return s.bytes
}
func (s SyncSummary) Height() uint64 {
return s.BlockNumber
}
func (s SyncSummary) ID() ids.ID {
return s.summaryID
}
func (s SyncSummary) String() string {
return fmt.Sprintf("SyncSummary(BlockHash=%s, BlockNumber=%d, BlockRoot=%s, AtomicRoot=%s)", s.BlockHash, s.BlockNumber, s.BlockRoot, s.AtomicRoot)
}
func (s SyncSummary) Accept(context.Context) (block.StateSyncMode, error) {
if s.acceptImpl == nil {
return block.StateSyncSkipped, fmt.Errorf("accept implementation not specified for summary: %s", s)
}
return s.acceptImpl(s)
}