-
Notifications
You must be signed in to change notification settings - Fork 669
/
unary_snowflake.go
62 lines (49 loc) · 1.42 KB
/
unary_snowflake.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
52
53
54
55
56
57
58
59
60
61
62
// Copyright (C) 2019-2022, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package snowball
import (
"fmt"
)
var _ UnarySnowflake = (*unarySnowflake)(nil)
// unarySnowflake is the implementation of a unary snowflake instance
type unarySnowflake struct {
// beta is the number of consecutive successful queries required for
// finalization.
beta int
// confidence tracks the number of successful polls in a row that have
// returned the preference
confidence int
// finalized prevents the state from changing after the required number of
// consecutive polls has been reached
finalized bool
}
func (sf *unarySnowflake) Initialize(beta int) {
sf.beta = beta
}
func (sf *unarySnowflake) RecordSuccessfulPoll() {
sf.confidence++
sf.finalized = sf.finalized || sf.confidence >= sf.beta
}
func (sf *unarySnowflake) RecordUnsuccessfulPoll() {
sf.confidence = 0
}
func (sf *unarySnowflake) Finalized() bool {
return sf.finalized
}
func (sf *unarySnowflake) Extend(beta int, choice int) BinarySnowflake {
return &binarySnowflake{
binarySlush: binarySlush{preference: choice},
confidence: sf.confidence,
beta: beta,
finalized: sf.finalized,
}
}
func (sf *unarySnowflake) Clone() UnarySnowflake {
newSnowflake := *sf
return &newSnowflake
}
func (sf *unarySnowflake) String() string {
return fmt.Sprintf("SF(Confidence = %d, Finalized = %v)",
sf.confidence,
sf.finalized)
}