This repository has been archived by the owner on Sep 26, 2019. It is now read-only.
/
protocol_fix_190_merkle.go
76 lines (65 loc) · 1.82 KB
/
protocol_fix_190_merkle.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
// Copyright 2019 The go-pttai Authors
// This file is part of the go-pttai library.
//
// The go-pttai library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-pttai library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-pttai library. If not, see <http://www.gnu.org/licenses/>.
package service
import (
"github.com/ailabstw/go-pttai/common"
"github.com/ailabstw/go-pttai/pttdb"
"github.com/syndtr/goleveldb/leveldb"
)
func (pm *BaseProtocolManager) Fix190Merkle() error {
isFixed, err := pm.isFix190Merkle()
if err != nil {
return err
}
if isFixed {
return nil
}
err = pm.ForceReconstructMerkle()
if err != nil {
return err
}
err = pm.setFix190Merkle()
if err != nil {
return err
}
return nil
}
func (pm *BaseProtocolManager) isFix190Merkle() (bool, error) {
key, err := pm.marshalFix190MerkleKey()
if err != nil {
return false, err
}
_, err = dbMeta.Get(key)
if err == leveldb.ErrNotFound {
return false, nil
}
if err != nil {
return false, err
}
return true, nil
}
func (pm *BaseProtocolManager) marshalFix190MerkleKey() ([]byte, error) {
entityID := pm.Entity().GetID()
return common.Concat([][]byte{DBFix190Prefix, entityID[:]})
}
func (pm *BaseProtocolManager) setFix190Merkle() error {
key, err := pm.marshalFix190MerkleKey()
if err != nil {
return err
}
dbMeta.Put(key, pttdb.ValueTrue)
return nil
}