support periodically cleaning up expired tokens

pkg/duplicatechecker/duplicate_checker.go

@@ -1,7 +1,11 @@
 package duplicatechecker
 
+import "time"
+
 // DuplicateChecker is common duplicate checker interface
 type DuplicateChecker interface {
 	GetSubmissionRemark(checkerType DuplicateCheckerType, uid int64, identification string) (bool, string)
 	SetSubmissionRemark(checkerType DuplicateCheckerType, uid int64, identification string, remark string)
+	GetOrSetCronJobRunningInfo(jobName string, runningInfo string, runningInterval time.Duration) (bool, string)
+	RemoveCronJobRunningInfo(jobName string)
 }

pkg/duplicatechecker/duplicate_checker_container.go

@@ -1,6 +1,8 @@
 package duplicatechecker
 
 import (
+	"time"
+
 	"github.com/mayswind/ezbookkeeping/pkg/errs"
 	"github.com/mayswind/ezbookkeeping/pkg/settings"
 )
@@ -36,3 +38,13 @@ func (c *DuplicateCheckerContainer) GetSubmissionRemark(checkerType DuplicateChe
 func (c *DuplicateCheckerContainer) SetSubmissionRemark(checkerType DuplicateCheckerType, uid int64, identification string, remark string) {
 	c.Current.SetSubmissionRemark(checkerType, uid, identification, remark)
 }
+
+// GetOrSetCronJobRunningInfo returns the running info when the cron job is running or saves the running info by the current duplicate checker
+func (c *DuplicateCheckerContainer) GetOrSetCronJobRunningInfo(jobName string, runningInfo string, runningInterval time.Duration) (bool, string) {
+	return c.Current.GetOrSetCronJobRunningInfo(jobName, runningInfo, runningInterval)
+}
+
+// RemoveCronJobRunningInfo removes the running info of the cron job by the current duplicate checker
+func (c *DuplicateCheckerContainer) RemoveCronJobRunningInfo(jobName string) {
+	c.Current.RemoveCronJobRunningInfo(jobName)
+}

pkg/duplicatechecker/duplicate_checker_type.go

@@ -5,9 +5,9 @@ type DuplicateCheckerType uint8
 
 // Types of uuid
 const (
-	DUPLICATE_CHECKER_TYPE_DEFAULT         DuplicateCheckerType = 0
-	DUPLICATE_CHECKER_TYPE_NEW_ACCOUNT     DuplicateCheckerType = 1
-	DUPLICATE_CHECKER_TYPE_NEW_CATEGORY    DuplicateCheckerType = 2
-	DUPLICATE_CHECKER_TYPE_NEW_TRANSACTION DuplicateCheckerType = 3
-	DUPLICATE_CHECKER_TYPE_NEW_TEMPLATE    DuplicateCheckerType = 4
+	DUPLICATE_CHECKER_TYPE_BACKGROUND_CRON_JOB DuplicateCheckerType = 0
+	DUPLICATE_CHECKER_TYPE_NEW_ACCOUNT         DuplicateCheckerType = 1
+	DUPLICATE_CHECKER_TYPE_NEW_CATEGORY        DuplicateCheckerType = 2
+	DUPLICATE_CHECKER_TYPE_NEW_TRANSACTION     DuplicateCheckerType = 3
+	DUPLICATE_CHECKER_TYPE_NEW_TEMPLATE        DuplicateCheckerType = 4
 )

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)
 }