diamondburned/arikawa
1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-11-30 18:53:30 +00:00
Code Issues Releases Activity

API: Fix atomic SIGSEGV in ARM

Browse source
This commit is contained in:
diamondburned 2020-12-31 10:24:51 -08:00
parent f1f052180b
commit 2f6ab0d451
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
api/rate/rate.go
View file

@ -31,7 +31,8 @@ type Limiter struct {
Prefix string
global int64 // atomic guarded, unixnano
// global is a pointer to prevent ARM-compatibility alignment.
global *int64 // atomic guarded, unixnano
bucketMu sync.Mutex
buckets map[string]*bucket
@ -62,6 +63,7 @@ func newBucket() *bucket {
func NewLimiter(prefix string) *Limiter {
return &Limiter{
Prefix: prefix,
global: new(int64),
buckets: map[string]*bucket{},
CustomLimits: []*CustomRateLimit{},
}
@ -111,7 +113,7 @@ func (l *Limiter) Acquire(ctx context.Context, path string) error {
until = b.reset
} else {
// maybe global rate limit has it
until = time.Unix(0, atomic.LoadInt64(&l.global))
until = time.Unix(0, atomic.LoadInt64(l.global))
}
if until.After(now) {
@ -181,8 +183,8 @@ func (l *Limiter) Release(path string, headers http.Header) error {
at := time.Now().Add(time.Duration(i) * time.Second)
if global != "" { // probably true
atomic.StoreInt64(&l.global, at.UnixNano())
if global != "" { // probably "true"
atomic.StoreInt64(l.global, at.UnixNano())
} else {
b.reset = at
}