diff options
| -rw-r--r-- | rate/ratelimit.go | 2 | ||||
| -rw-r--r-- | stmutil/context.go | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/rate/ratelimit.go b/rate/ratelimit.go index 69d6932..a44f383 100644 --- a/rate/ratelimit.go +++ b/rate/ratelimit.go @@ -37,7 +37,7 @@ func Every(interval time.Duration) Limit { func NewLimiter(rate Limit, burst numTokens) *Limiter { rl := &Limiter{ max: stm.NewVar(burst), - cur: stm.NewVar(burst), + cur: stm.NewBuiltinEqVar(burst), lastAdd: stm.NewVar(time.Now()), rate: rate, } diff --git a/stmutil/context.go b/stmutil/context.go index 315abf4..8a4d58d 100644 --- a/stmutil/context.go +++ b/stmutil/context.go @@ -23,7 +23,7 @@ func ContextDoneVar(ctx context.Context) (*stm.Var, func()) { if ctx.Err() != nil { // TODO: What if we had read-only Vars? Then we could have a global one for this that we // just reuse. - v := stm.NewVar(true) + v := stm.NewBuiltinEqVar(true) return v, func() {} } v := stm.NewVar(false) |