highperfocused/go-nostr
1
0
Fork 0
mirror of https://github.com/nbd-wtf/go-nostr.git synced 2025-07-01 11:02:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

pool: support CountMany() using hyperloglog.

Browse Source
This commit is contained in:
fiatjaf
2024-11-16 16:59:24 -03:00
parent 99e45035d5
commit 0d40b40c9c
7 changed files with 72 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nip45/helpers.go → nip45/hyperloglog/helpers.go
View File

@ -1,4 +1,4 @@
package nip45
package hyperloglog
import (
"math"

20
nip45/hll.go → nip45/hyperloglog/hll.go
View File

@ -1,4 +1,4 @@
package nip45
package hyperloglog
import (
"encoding/binary"
@ -18,13 +18,14 @@ func New() *HyperLogLog {
return hll
}
func (hll *HyperLogLog) Encode() string {
return hex.EncodeToString(hll.registers)
}
func (hll *HyperLogLog) Decode(enc string) error {
_, err := hex.Decode(hll.registers, []byte(enc))
return err
func (hll *HyperLogLog) GetRegisters() []byte { return hll.registers }
func (hll *HyperLogLog) SetRegisters(enc []byte) { hll.registers = enc }
func (hll *HyperLogLog) MergeRegisters(other []byte) {
for i, v := range other {
if v > hll.registers[i] {
hll.registers[i] = v
}
}
}
func (hll *HyperLogLog) Clear() {
@ -45,13 +46,12 @@ func (hll *HyperLogLog) Add(id string) {
}
}
func (hll *HyperLogLog) Merge(other *HyperLogLog) error {
func (hll *HyperLogLog) Merge(other *HyperLogLog) {
for i, v := range other.registers {
if v > hll.registers[i] {
hll.registers[i] = v
}
}
return nil
}
func (hll *HyperLogLog) Count() uint64 {

2
nip45/hll_test.go → nip45/hyperloglog/hll_test.go
View File

@ -1,4 +1,4 @@
package nip45
package hyperloglog
import (
"encoding/hex"