You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.0 KiB
Go
61 lines
1.0 KiB
Go
package crdt
|
|
|
|
import (
|
|
"fmt"
|
|
"constraints"
|
|
)
|
|
|
|
func max[N constraints.Ordered](a, b N) N {
|
|
if a >= b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
func NewGCounter[K comparable]() GCounter[K] {
|
|
return GCounter[K]{slots: make(map[K]int)}
|
|
}
|
|
|
|
type GCounter[K comparable] struct {
|
|
slots map[K]int `json:"slots"`
|
|
}
|
|
|
|
func (g GCounter[K]) Incr(slot K) error {
|
|
var zero K
|
|
if slot == zero {
|
|
return fmt.Errorf("gcounter refuses incr on the zero-value of its key")
|
|
}
|
|
|
|
g.slots[slot]++
|
|
return nil
|
|
}
|
|
|
|
func (g GCounter[K]) Add(slot K, delta int) error {
|
|
var zero K
|
|
if slot == zero {
|
|
return fmt.Errorf("gcounter refuses add on the zero-value of its key")
|
|
}
|
|
|
|
if delta < 0 {
|
|
return fmt.Errorf("gcounters cannot go down, use a pncounter instead")
|
|
}
|
|
g.slots[slot] += delta
|
|
return nil
|
|
}
|
|
|
|
func (g *GCounter[K]) Merge(dest *GCounter[K]) {
|
|
for slot, count := range g.slots {
|
|
v := max(count, dest.slots[slot])
|
|
dest.slots[slot] = v
|
|
g.slots[slot] = v
|
|
}
|
|
}
|
|
|
|
func (g *GCounter[K]) Total() int {
|
|
var n int
|
|
for _, count := range g.slots {
|
|
n += count
|
|
}
|
|
return n
|
|
}
|