Locker: add option function locker with time to live, for handling if service down and key still exist

Author: WanMuhafidzFaldi

candiutils/locker.go

@@ -25,9 +25,13 @@ func NewRedisLocker(pool *redis.Pool) *RedisLocker {
 	return &RedisLocker{pool: pool}
 }
 
-func (r *RedisLocker) IsLocked(key string) bool {
+func (r *RedisLocker) IsLocked(key string, expiration ...time.Duration) bool {
 	conn := r.pool.Get()
 	incr, _ := redis.Int64(conn.Do("INCR", key))
+
+	if len(expiration) > 0 {
+		conn.Do("EXPIRE", key, int(expiration[0].Seconds()))
+	}
 	conn.Close()
 
 	return incr > 1
@@ -122,6 +126,9 @@ func (r *RedisLocker) Lock(key string, timeout time.Duration) (unlockFunc func()
 // IsLocked method
 func (NoopLocker) IsLocked(string) bool { return false }
 
+// IsLocked method
+func (NoopLocker) IsLockedWithTTL(string, time.Duration) bool { return false }
+
 // HasBeenLocked method
 func (NoopLocker) HasBeenLocked(string) bool { return false }
 

codebase/interfaces/locker.go

@@ -6,6 +6,7 @@ type (
 	// Locker abstraction, lock concurrent process
 	Locker interface {
 		IsLocked(key string) bool
+		IsLockedWithTTL(key string, timeout time.Duration) bool
 		HasBeenLocked(key string) bool
 		Unlock(key string)
 		Reset(key string)