Add function IsLockedTTLWithLimit checks if the key has been incremen…

candiutils/locker.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"time"
 
+	"github.com/golangid/candi/options"
 	"github.com/gomodule/redigo/redis"
 )
 
@@ -15,39 +16,37 @@ type (
 	// RedisLocker lock using redis
 	RedisLocker struct {
 		pool          *redis.Pool
-		lockeroptions LockerOptions
+		lockeroptions options.LockerOptions
 	}
 
 	// NoopLocker empty locker
 	NoopLocker struct{}
-
-	// Options for RedisLocker
-	LockerOptions struct {
-		Prefix string
-		TTL    time.Duration
-	}
-
-	// Option function type for setting options
-	LockerOption func(*LockerOptions)
 )
 
 // WithPrefix sets the prefix for keys
-func WithPrefixLocker(prefix string) LockerOption {
-	return func(o *LockerOptions) {
+func WithPrefixLocker(prefix string) options.LockerOption {
+	return func(o *options.LockerOptions) {
 		o.Prefix = prefix
 	}
 }
 
 // WithTTL sets the default TTL for keys
-func WithTTLLocker(ttl time.Duration) LockerOption {
-	return func(o *LockerOptions) {
+func WithTTLLocker(ttl time.Duration) options.LockerOption {
+	return func(o *options.LockerOptions) {
 		o.TTL = ttl
 	}
 }
 
+// WithLimit sets the limit for keys
+func WithLimitLocker(limit int) options.LockerOption {
+	return func(o *options.LockerOptions) {
+		o.Limit = limit
+	}
+}
+
 // NewRedisLocker constructor
-func NewRedisLocker(pool *redis.Pool, opts ...LockerOption) *RedisLocker {
-	lockeroptions := LockerOptions{
+func NewRedisLocker(pool *redis.Pool, opts ...options.LockerOption) *RedisLocker {
+	lockeroptions := options.LockerOptions{
 		Prefix: "LOCKFOR",
 		TTL:    0,
 	}
@@ -104,6 +103,28 @@ func (r *RedisLocker) IsLockedTTL(key string, TTL time.Duration) bool {
 	return incr > 1
 }
 
+func (r *RedisLocker) IsLockedWithOpts(key string, opts ...options.LockerOption) bool {
+	conn := r.pool.Get()
+	defer conn.Close()
+
+	lockOpt := r.lockeroptions
+	for _, opt := range opts {
+		opt(&lockOpt)
+	}
+
+	lockKey := fmt.Sprintf("%s:%s", r.lockeroptions.Prefix, key)
+	incr, err := redis.Int64(conn.Do("INCR", lockKey))
+	if err != nil {
+		return false
+	}
+
+	withLimit := lockOpt.Limit > 1
+	if lockOpt.TTL > 0 && !(withLimit && incr == 1) {
+		conn.Do("EXPIRE", lockKey, int(lockOpt.TTL.Seconds()))
+	}
+	return incr > int64(lockOpt.Limit)
+}
+
 func (r *RedisLocker) HasBeenLocked(key string) bool {
 	conn := r.pool.Get()
 	defer conn.Close()
@@ -215,6 +236,9 @@ func (NoopLocker) IsLocked(string) bool { return false }
 // IsLockedTTL method
 func (NoopLocker) IsLockedTTL(string, time.Duration) bool { return false }
 
+// IsLockedWithOpts method
+func (NoopLocker) IsLockedWithOpts(string, ...options.LockerOption) bool { return false }
+
 // HasBeenLocked method
 func (NoopLocker) HasBeenLocked(string) bool { return false }
 
@@ -227,10 +251,10 @@ func (NoopLocker) Reset(string) {}
 // Lock method
 func (NoopLocker) Lock(string, time.Duration) (func(), error) { return func() {}, nil }
 
-func (NoopLocker) Disconnect(context.Context) error { return nil }
-
 // GetPrefix method
 func (NoopLocker) GetPrefixLocker() string { return "" }
 
 // GetTTLLocker method
 func (NoopLocker) GetTTLLocker() time.Duration { return 0 }
+
+func (NoopLocker) Disconnect(context.Context) error { return nil }

codebase/interfaces/locker.go

@@ -1,12 +1,17 @@
 package interfaces
 
-import "time"
+import (
+	"time"
+
+	"github.com/golangid/candi/options"
+)
 
 type (
 	// Locker abstraction, lock concurrent process
 	Locker interface {
 		IsLocked(key string) bool
 		IsLockedTTL(key string, ttl time.Duration) bool
+		IsLockedWithOpts(key string, opts ...options.LockerOption) bool
 		HasBeenLocked(key string) bool
 		Unlock(key string)
 		Reset(key string)

options/locker.go

@@ -0,0 +1,15 @@
+package options
+
+import "time"
+
+type (
+	// Options for RedisLocker
+	LockerOptions struct {
+		Prefix string
+		TTL    time.Duration
+		Limit  int
+	}
+
+	// Option function type for setting options
+	LockerOption func(*LockerOptions)
+)