/
unary_snowflake.go
69 lines (55 loc) · 1.65 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
63
64
65
66
67
68
69
// Copyright (C) 2019-2024, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package snowball
import "fmt"
var _ Unary = (*unarySnowflake)(nil)
func newUnarySnowflake(beta int) unarySnowflake {
return unarySnowflake{
beta: beta,
}
}
// 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) RecordSuccessfulPoll() {
sf.confidence++
sf.finalized = sf.finalized || sf.confidence >= sf.beta
}
// RecordPollPreference fails to reach an alpha threshold to increase our
// confidence, so this calls RecordUnsuccessfulPoll to reset the confidence
// counter.
func (sf *unarySnowflake) RecordPollPreference() {
sf.RecordUnsuccessfulPoll()
}
func (sf *unarySnowflake) RecordUnsuccessfulPoll() {
sf.confidence = 0
}
func (sf *unarySnowflake) Finalized() bool {
return sf.finalized
}
func (sf *unarySnowflake) Extend(beta int, choice int) Binary {
return &binarySnowflake{
binarySlush: binarySlush{preference: choice},
confidence: sf.confidence,
beta: beta,
finalized: sf.finalized,
}
}
func (sf *unarySnowflake) Clone() Unary {
newSnowflake := *sf
return &newSnowflake
}
func (sf *unarySnowflake) String() string {
return fmt.Sprintf("SF(Confidence = %d, Finalized = %v)",
sf.confidence,
sf.finalized)
}