util/BaseBinary.go at master · buguang01/util · GitHub

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
111
112
113
114
115
116
117
118
package util

import (
	"database/sql/driver"
	"errors"
	"fmt"
)

//BaseBinary DB上的2进制数据
type BaseBinary struct {
	Data          []uint8 //放数据的数组
	ArrayLen      int     //上面那个数组的长度
	OneDataBitNum uint8   //一个数据占多少个位，只能是1，2，4，8
}

func NewBinary(alen int) *BaseBinary {
	result := new(BaseBinary)
	result.Data = make([]uint8, alen)
	result.ArrayLen = alen
	result.OneDataBitNum = 1
	return result
}
func NewBinaryByLen(alen, dlen int) *BaseBinary {
	result := new(BaseBinary)

	result.Data = make([]uint8, alen)
	result.ArrayLen = alen
	result.OneDataBitNum = uint8(dlen)
	return result
}

func (this *BaseBinary) Init(d []uint8) {
	this.Data = d
	this.ArrayLen = len(this.Data)
}

func (this *BaseBinary) ContainKey(index int) int {
	dlen := int(this.OneDataBitNum)
	ai := index / (8 / dlen)
	bitnum := uint8(index % (8 / dlen) * dlen)
	if ai >= this.ArrayLen {
		return 0
	}

	d := this.Data[ai]
	if this.OneDataBitNum == 8 {
		return int(d)
	}
	var mask uint8 = (1 << this.OneDataBitNum) - 1
	return int((d & (mask << bitnum)) >> bitnum)
}

func (this *BaseBinary) UpData(index, val int) bool {
	bval := uint8(val)

	dlen := int(this.OneDataBitNum)
	ai := (index / (8 / dlen))
	bitnum := uint8(index % (8 / dlen) * dlen)
	if ai >= this.ArrayLen {
		return false
	}
	key := this.Data[ai]

	if this.OneDataBitNum == 8 {
		key = bval
	} else {
		var tmp uint8 = (1 << this.OneDataBitNum) - 1
		bval = bval & tmp
		tmp = tmp << bitnum
		tmp = key & ^tmp

		key = uint8(tmp + (bval << bitnum))
	}
	this.Data[ai] = key
	return true

}

func (this *BaseBinary) ToValuesJson() []interface{} {
	dlen := int(this.OneDataBitNum)
	result := make([]interface{}, this.ArrayLen*8/dlen)
	index := 0
	for k := 0; k < this.ArrayLen; k++ {
		key := this.Data[k]
		if this.OneDataBitNum == 8 {
			result[index] = key
			index++
		} else {

			for bitnum := uint8(0); bitnum < 8; bitnum = bitnum + this.OneDataBitNum {
				var mask uint8 = (1 << this.OneDataBitNum) - 1
				result[index] = (key & (mask << bitnum)) >> bitnum
				index++
			}
		}
	}
	return result
}

func (this *BaseBinary) String() string {
	return fmt.Sprintf("BaseBinary ArrayLen:%d,OneDataBitNum:%d Data:%+v", this.ArrayLen, this.OneDataBitNum, this.Data)
}

// 实现driver.Valuer接口
func (this BaseBinary) Value() (driver.Value, error) {
	return this.Data, nil
}

// 实现sql.Scanner接口
func (this *BaseBinary) Scan(val interface{}) (err error) {
	if d, ok := val.([]uint8); ok {
		this.Data = d
		this.ArrayLen = len(this.Data)
	} else {
		err = errors.New("sql.Scanner Scan val is not []uint8.")
	}
	return
}