forked from tw-bc-group/fabric
-
Notifications
You must be signed in to change notification settings - Fork 0
/
uint64_encoding.go
51 lines (45 loc) · 1.59 KB
/
uint64_encoding.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
/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package util
import (
"encoding/binary"
"math"
"github.com/golang/protobuf/proto"
)
// EncodeReverseOrderVarUint64 returns a byte-representation for a uint64 number such that
// the number is first subtracted from MaxUint64 and then all the leading 0xff bytes
// are trimmed and replaced by the number of such trimmed bytes. This helps in reducing the size.
// In the byte order comparison this encoding ensures that EncodeReverseOrderVarUint64(A) > EncodeReverseOrderVarUint64(B),
// If B > A
func EncodeReverseOrderVarUint64(number uint64) []byte {
bytes := make([]byte, 8)
binary.BigEndian.PutUint64(bytes, math.MaxUint64-number)
numFFBytes := 0
for _, b := range bytes {
if b != 0xff {
break
}
numFFBytes++
}
size := 8 - numFFBytes
encodedBytes := make([]byte, size+1)
encodedBytes[0] = proto.EncodeVarint(uint64(numFFBytes))[0]
copy(encodedBytes[1:], bytes[numFFBytes:])
return encodedBytes
}
// DecodeReverseOrderVarUint64 decodes the number from the bytes obtained from function 'EncodeReverseOrderVarUint64'.
// Also, returns the number of bytes that are consumed in the process
func DecodeReverseOrderVarUint64(bytes []byte) (uint64, int) {
s, _ := proto.DecodeVarint(bytes)
numFFBytes := int(s)
decodedBytes := make([]byte, 8)
realBytesNum := 8 - numFFBytes
copy(decodedBytes[numFFBytes:], bytes[1:realBytesNum+1])
numBytesConsumed := realBytesNum + 1
for i := 0; i < numFFBytes; i++ {
decodedBytes[i] = 0xff
}
return (math.MaxUint64 - binary.BigEndian.Uint64(decodedBytes)), numBytesConsumed
}