46 lines
774 B
Go
46 lines
774 B
Go
|
package memory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"forge.cadoles.com/cadoles/bouncer/internal/lock"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
ErrTimeout = errors.New("timeout")
|
||
|
|
)
|
||
|
|
|
||
|
|
type Locker struct {
|
||
|
|
lock chan struct{}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithLock implements lock.Locker.
|
||
|
|
func (l *Locker) WithLock(ctx context.Context, key string, timeout time.Duration, fn func(ctx context.Context) error) error {
|
||
|
|
select {
|
||
|
|
case l.lock <- struct{}{}:
|
||
|
|
defer func() {
|
||
|
|
<-l.lock
|
||
|
|
}()
|
||
|
|
if err := fn(ctx); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
case <-ctx.Done():
|
||
|
|
return errors.WithStack(ctx.Err())
|
||
|
|
|
||
|
|
case <-time.After(timeout):
|
||
|
|
return errors.WithStack(ErrTimeout)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLocker() *Locker {
|
||
|
|
return &Locker{
|
||
|
|
lock: make(chan struct{}, 1),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ lock.Locker = &Locker{}
|