-
Notifications
You must be signed in to change notification settings - Fork 4
/
statemarsh.go
110 lines (99 loc) · 2.16 KB
/
statemarsh.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
package state
import (
"encoding/binary"
"io"
)
func (t *Command) Marshal(w io.Writer) {
var b [8]byte
// ClientId
bs := b[:4]
utmp32 := t.ClientId
bs[0] = byte(utmp32)
bs[1] = byte(utmp32 >> 8)
bs[2] = byte(utmp32 >> 16)
bs[3] = byte(utmp32 >> 24)
w.Write(bs)
// OpId
bs = b[:4]
tmp32 := t.OpId
bs[0] = byte(tmp32)
bs[1] = byte(tmp32 >> 8)
bs[2] = byte(tmp32 >> 16)
bs[3] = byte(tmp32 >> 24)
w.Write(bs)
// Op
bs = b[:1]
b[0] = byte(t.Op)
w.Write(bs)
bs = b[:8]
// K
binary.LittleEndian.PutUint64(bs, uint64(t.K))
w.Write(bs)
// V
binary.LittleEndian.PutUint64(bs, uint64(t.V))
w.Write(bs)
}
func (t *Command) Unmarshal(r io.Reader) error {
var b [8]byte
bs := b[:4]
// ClientId
if _, err := io.ReadAtLeast(r, bs, 4); err != nil {
return err
}
t.ClientId = uint32((uint32(bs[0]) | (uint32(bs[1]) << 8) | (uint32(bs[2]) << 16) | (uint32(bs[3]) << 24)))
// OpId
bs = b[:4]
if _, err := io.ReadAtLeast(r, bs, 4); err != nil {
return err
}
//t.OpId = OperationId((uint32(bs[0]) | (uint32(bs[1]) << 8) | (uint32(bs[2]) << 16) | (uint32(bs[3]) << 24)))
t.OpId = int32((uint32(bs[0]) | (uint32(bs[1]) << 8) | (uint32(bs[2]) << 16) | (uint32(bs[3]) << 24)))
// Op
bs = b[:1]
if _, err := io.ReadFull(r, bs); err != nil {
return err
}
t.Op = Operation(b[0])
bs = b[:8]
// K
if _, err := io.ReadFull(r, bs); err != nil {
return err
}
t.K = Key(binary.LittleEndian.Uint64(bs))
// V
if _, err := io.ReadFull(r, bs); err != nil {
return err
}
t.V = Value(binary.LittleEndian.Uint64(bs))
return nil
}
func (t *Key) Marshal(w io.Writer) {
var b [8]byte
bs := b[:8]
binary.LittleEndian.PutUint64(bs, uint64(*t))
w.Write(bs)
}
func (t *Value) Marshal(w io.Writer) {
var b [8]byte
bs := b[:8]
binary.LittleEndian.PutUint64(bs, uint64(*t))
w.Write(bs)
}
func (t *Key) Unmarshal(r io.Reader) error {
var b [8]byte
bs := b[:8]
if _, err := io.ReadFull(r, bs); err != nil {
return err
}
*t = Key(binary.LittleEndian.Uint64(bs))
return nil
}
func (t *Value) Unmarshal(r io.Reader) error {
var b [8]byte
bs := b[:8]
if _, err := io.ReadFull(r, bs); err != nil {
return err
}
*t = Value(binary.LittleEndian.Uint64(bs))
return nil
}