chain/bloom.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

package chain

type BloomFilter struct {
    bin []byte
}

func NewBloomFilter(bin []byte) *BloomFilter {
    if bin == nil {
        bin = make([]byte, 256)
    }

    return &BloomFilter{
        bin: bin,
    }
}

func (self *BloomFilter) Set(addr []byte) {
    if len(addr) < 8 {
        chainlogger.Warnf("err: bloom set to small: %x\n", addr)

        return
    }

    for _, i := range addr[len(addr)-8:] {
        self.bin[i] = 1
    }
}

func (self *BloomFilter) Search(addr []byte) bool {
    if len(addr) < 8 {
        chainlogger.Warnf("err: bloom search to small: %x\n", addr)

        return false
    }

    for _, i := range addr[len(addr)-8:] {
        if self.bin[i] == 0 {
            return false
        }
    }

    return true
}

func (self *BloomFilter) Bin() []byte {
    return self.bin
}