blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-11-05 21:02:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Release cron lock

Browse Source 
Signed-off-by: imalasong <imalasong@qq.com>
This commit is contained in:
imalasong 2024-04-21 08:02:34 +08:00
parent 42332cdd26
commit 9bc6a07cbf
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
internal/tools/cron_task.go
View File

@ -103,6 +103,15 @@ func netlock(rdb redis.UniversalClient, key string, ttl time.Duration) bool {
return ok return ok
} }
// netUnlock release redis lock.
func netUnlock(rdb redis.UniversalClient, key string) error {
_, err := rdb.Del(context.Background(), key).Result()
if err != nil {
return errs.Wrap(err)
}
return nil
}
func cronWrapFunc(config *CronTaskConfig, rdb redis.UniversalClient, key string, fn func()) func() { func cronWrapFunc(config *CronTaskConfig, rdb redis.UniversalClient, key string, fn func()) func() {
enableCronLocker := config.CronTask.EnableCronLocker enableCronLocker := config.CronTask.EnableCronLocker
return func() { return func() {
@ -114,6 +123,9 @@ func cronWrapFunc(config *CronTaskConfig, rdb redis.UniversalClient, key string,
// when acquire redis lock, call fn(). // when acquire redis lock, call fn().
if netlock(rdb, key, 5*time.Second) { if netlock(rdb, key, 5*time.Second) {
defer func() {
_ = netUnlock(rdb, key)
}()
fn() fn()
} }
} }

6
internal/tools/cron_task_test.go
View File

@ -38,6 +38,12 @@ func TestDisLock(t *testing.T) {
// set different key // set different key
assert.Equal(t, true, netlock(rdb, "cron-2", 2*time.Second)) assert.Equal(t, true, netlock(rdb, "cron-2", 2*time.Second))
// set key and release
assert.Equal(t, true, netlock(rdb, "cron-3", 5*time.Second))
assert.Equal(t, nil, netUnlock(rdb, "cron-3"))
//retrieve set key
assert.Equal(t, true, netlock(rdb, "cron-3", 5*time.Second))
} }
//func TestCronWrapFunc(t *testing.T) { //func TestCronWrapFunc(t *testing.T) {