You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.1 KiB
36 lines
1.1 KiB
package jobs_gorm
|
|
|
|
import (
|
|
"fmt"
|
|
"go.dtapp.net/goredis"
|
|
"time"
|
|
)
|
|
|
|
// Lock 上锁
|
|
func (jobsGorm *JobsGorm) Lock(info Task, id any) string {
|
|
cacheName := fmt.Sprintf("cron:%v:%v", info.Type, id)
|
|
judgeCache := jobsGorm.Redis.NewStringOperation().Get(cacheName).UnwrapOr("")
|
|
if judgeCache != "" {
|
|
return judgeCache
|
|
}
|
|
jobsGorm.Redis.NewStringOperation().Set(cacheName, fmt.Sprintf("已在%v机器上锁成功", jobsGorm.outsideIp), goredis.WithExpire(time.Millisecond*time.Duration(info.Frequency)*3))
|
|
return ""
|
|
}
|
|
|
|
// Unlock Lock 解锁
|
|
func (jobsGorm *JobsGorm) Unlock(info Task, id any) {
|
|
cacheName := fmt.Sprintf("cron:%v:%v", info.Type, id)
|
|
jobsGorm.Redis.NewStringOperation().Del(cacheName)
|
|
}
|
|
|
|
// LockForever 永远上锁
|
|
func (jobsGorm *JobsGorm) LockForever(info Task, id any) string {
|
|
cacheName := fmt.Sprintf("cron:%v:%v", info.Type, id)
|
|
judgeCache := jobsGorm.Redis.NewStringOperation().Get(cacheName).UnwrapOr("")
|
|
if judgeCache != "" {
|
|
return judgeCache
|
|
}
|
|
jobsGorm.Redis.NewStringOperation().Set(cacheName, fmt.Sprintf("已在%v机器永远上锁成功", jobsGorm.outsideIp))
|
|
return ""
|
|
}
|