support periodically cleaning up expired tokens

pkg/duplicatechecker/in_memory_duplicate_checker.go

@@ -2,6 +2,8 @@ package duplicatechecker
 
 import (
 	"fmt"
+	"sync"
+	"time"
 
 	"github.com/patrickmn/go-cache"
 
@@ -11,6 +13,8 @@ import (
 // InMemoryDuplicateChecker represents in-memory duplicate checker
 type InMemoryDuplicateChecker struct {
 	cache *cache.Cache
+
+	mutex sync.Mutex
 }
 
 // NewInMemoryDuplicateChecker returns a new in-memory duplicate checker
@@ -38,6 +42,33 @@ func (c *InMemoryDuplicateChecker) SetSubmissionRemark(checkerType DuplicateChec
 	c.cache.Set(c.getCacheKey(checkerType, uid, identification), remark, cache.DefaultExpiration)
 }
 
+// GetOrSetCronJobRunningInfo returns the running info when the cron job is running or saves the running info by the current duplicate checker
+func (c *InMemoryDuplicateChecker) GetOrSetCronJobRunningInfo(jobName string, runningInfo string, runningInterval time.Duration) (bool, string) {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+
+	existedRunningInfo, found := c.cache.Get(c.getCacheKey(DUPLICATE_CHECKER_TYPE_BACKGROUND_CRON_JOB, 0, jobName))
+
+	if found {
+		return true, existedRunningInfo.(string)
+	}
+
+	expiration := runningInterval
+
+	if expiration > 1*time.Second {
+		expiration = expiration - 1*time.Second
+	}
+
+	c.cache.Set(c.getCacheKey(DUPLICATE_CHECKER_TYPE_BACKGROUND_CRON_JOB, 0, jobName), runningInfo, expiration)
+
+	return false, ""
+}
+
+// RemoveCronJobRunningInfo removes the running info of the cron job by the current duplicate checker
+func (c *InMemoryDuplicateChecker) RemoveCronJobRunningInfo(jobName string) {
+	c.cache.Delete(c.getCacheKey(DUPLICATE_CHECKER_TYPE_BACKGROUND_CRON_JOB, 0, jobName))
+}
+
 func (c *InMemoryDuplicateChecker) getCacheKey(checkerType DuplicateCheckerType, uid int64, identification string) string {
 	return fmt.Sprintf("%d|%d|%s", checkerType, uid, identification)
 }