MaysWind
113 lines
3.7 KiB
Go
113 lines
3.7 KiB
Go
package duplicatechecker
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/mayswind/ezbookkeeping/pkg/errs"
|
|
"github.com/mayswind/ezbookkeeping/pkg/settings"
|
|
)
|
|
|
|
// DuplicateCheckerContainer contains the current duplicate checker
|
|
type DuplicateCheckerContainer struct {
|
|
current DuplicateChecker
|
|
}
|
|
|
|
// Initialize a duplicate checker container singleton instance
|
|
var (
|
|
Container = &DuplicateCheckerContainer{}
|
|
)
|
|
|
|
// InitializeDuplicateChecker initializes the current duplicate checker according to the config
|
|
func InitializeDuplicateChecker(config *settings.Config) error {
|
|
if config.DuplicateCheckerType == settings.InMemoryDuplicateCheckerType {
|
|
checker, err := NewInMemoryDuplicateChecker(config)
|
|
Container.current = checker
|
|
|
|
return err
|
|
}
|
|
|
|
return errs.ErrInvalidDuplicateCheckerType
|
|
}
|
|
|
|
// SetDuplicateChecker sets the current duplicate checker
|
|
func SetDuplicateChecker(checker DuplicateChecker) {
|
|
Container.current = checker
|
|
}
|
|
|
|
// IsEnabled returns whether the duplicate checker is enabled
|
|
func (c *DuplicateCheckerContainer) IsEnabled() bool {
|
|
return c.current != nil
|
|
}
|
|
|
|
// GetSubmissionRemark returns whether the same submission has been processed and related remark by the current duplicate checker
|
|
func (c *DuplicateCheckerContainer) GetSubmissionRemark(checkerType DuplicateCheckerType, uid int64, identification string) (bool, string) {
|
|
if c.current == nil {
|
|
return false, ""
|
|
}
|
|
|
|
return c.current.GetSubmissionRemark(checkerType, uid, identification)
|
|
}
|
|
|
|
// SetSubmissionRemark saves the identification and remark by the current duplicate checker
|
|
func (c *DuplicateCheckerContainer) SetSubmissionRemark(checkerType DuplicateCheckerType, uid int64, identification string, remark string) {
|
|
if c.current == nil {
|
|
return
|
|
}
|
|
|
|
c.current.SetSubmissionRemark(checkerType, uid, identification, remark)
|
|
}
|
|
|
|
// SetSubmissionRemarkWithCustomExpiration saves the identification and remark by the current duplicate checker with custom expiration time
|
|
func (c *DuplicateCheckerContainer) SetSubmissionRemarkWithCustomExpiration(checkerType DuplicateCheckerType, uid int64, identification string, remark string, expiration time.Duration) {
|
|
if c.current == nil {
|
|
return
|
|
}
|
|
|
|
c.current.SetSubmissionRemarkWithCustomExpiration(checkerType, uid, identification, remark, expiration)
|
|
}
|
|
|
|
// RemoveSubmissionRemark removes the identification and remark by the current duplicate checker
|
|
func (c *DuplicateCheckerContainer) RemoveSubmissionRemark(checkerType DuplicateCheckerType, uid int64, identification string) {
|
|
if c.current == nil {
|
|
return
|
|
}
|
|
|
|
c.current.RemoveSubmissionRemark(checkerType, uid, identification)
|
|
}
|
|
|
|
// 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) {
|
|
if c.current == nil {
|
|
return false, ""
|
|
}
|
|
|
|
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) {
|
|
if c.current == nil {
|
|
return
|
|
}
|
|
|
|
c.current.RemoveCronJobRunningInfo(jobName)
|
|
}
|
|
|
|
// GetFailureCount returns the failure count of the specified failure key
|
|
func (c *DuplicateCheckerContainer) GetFailureCount(failureKey string) uint32 {
|
|
if c.current == nil {
|
|
return 0
|
|
}
|
|
|
|
return c.current.GetFailureCount(failureKey)
|
|
}
|
|
|
|
// IncreaseFailureCount increases the failure count of the specified failure key
|
|
func (c *DuplicateCheckerContainer) IncreaseFailureCount(failureKey string) uint32 {
|
|
if c.current == nil {
|
|
return 0
|
|
}
|
|
|
|
return c.current.IncreaseFailureCount(failureKey)
|
|
}
|