/
confirmmsg.go
66 lines (54 loc) · 1.3 KB
/
confirmmsg.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
// Copyright (c) 2017-2019 The Elastos Foundation
// Use of this source code is governed by an MIT
// license that can be found in the LICENSE file.
//
package msg
import (
"bytes"
"io"
"github.com/elastos/Elastos.ELA/common"
"github.com/elastos/Elastos.ELA/core/types/payload"
"github.com/elastos/Elastos.ELA/p2p"
)
// Ensure BlockMsg implement p2p.Message interface.
var _ p2p.Message = (*ConfirmMsg)(nil)
type ConfirmMsg struct {
Confirm *payload.Confirm
Height uint64
}
func NewConfirmMsg(confirm *payload.Confirm, height uint64) *ConfirmMsg {
return &ConfirmMsg{Confirm: confirm, Height: height}
}
func (msg *ConfirmMsg) CMD() string {
return CmdConfirm
}
func (msg *ConfirmMsg) MaxLength() uint32 {
buffer := new(bytes.Buffer)
if err := msg.Serialize(buffer); err != nil {
return 0
}
return uint32(buffer.Len())
}
func (msg *ConfirmMsg) Serialize(w io.Writer) error {
err := msg.Confirm.Serialize(w)
if err != nil {
return err
}
if err := common.WriteUint64(w, msg.Height); err != nil {
return err
}
return nil
}
func (msg *ConfirmMsg) Deserialize(r io.Reader) error {
if msg.Confirm == nil {
msg.Confirm = &payload.Confirm{}
}
err := msg.Confirm.Deserialize(r)
if err != nil {
return err
}
if msg.Height, err = common.ReadUint64(r); err != nil {
return err
}
return nil
}