pkg/timerx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

提交

Browse Source
This commit is contained in:
Yun
2023-09-02 12:19:27 +08:00
parent 1d37696815
commit 9622d9f5b9
4 changed files with 108 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lockx/lockx.go
+14 -8
View File
@@ -30,12 +30,18 @@ func NewGlobalLock(ctx context.Context, red *redis.Client, uniqueKey string) *gl
func (g *globalLock) Lock() bool {
script := `
return redis.call('set',KEYS[1],ARGV[1],'EX',ARGV[2])
local token = redis.call('get',KEYS[1])
if token == false
then
return redis.call('set',KEYS[1],ARGV[1],'EX',ARGV[2])
end
return 'ERROR'
`
resp, err := g.redis.Eval(g.ctx, script, []string{g.uniqueKey}, g.value, 10).Result()
if resp != "OK" {
log.Println("globalLock Lock", resp, err)
_ = err
// log.Println("globalLock Lock", resp, err, g.uniqueKey, g.value)
}
return resp == "OK"
}
@@ -58,15 +64,15 @@ func (g *globalLock) Unlock() bool {
local token = redis.call('get',KEYS[1])
if token == ARGV[1]
then
redis.call('del',KEYS[1])
return 'OK'
redis.call('del',KEYS[1])
return 'OK'
end
return 'ERROR'
return 'ERROR'
`
resp, err := g.redis.Eval(g.ctx, script, []string{g.uniqueKey}, g.value).Result()
if resp != "OK" {
log.Println("globalLock Unlock", resp, err)
log.Println("globalLock Unlock", resp, err, g.uniqueKey, g.value)
}
return resp == "OK"
}
@@ -98,12 +104,12 @@ func (g *globalLock) refresh() bool {
redis.call('set',KEYS[1],ARGV[1],'EX',ARGV[2])
return 'OK'
end
return 'ERROR'
return 'ERROR'
`
resp, err := g.redis.Eval(g.ctx, script, []string{g.uniqueKey}, g.value, 5).Result()
if resp != "OK" {
log.Println("globalLock refresh", resp, err)
log.Println("globalLock refresh", resp, err, g.uniqueKey, g.value)
}
return resp == "OK"
}
lockx/lockx_test.go
+19 -8
View File
@@ -11,7 +11,21 @@ import (
var Redis *redis.Client
func TestMain(m *testing.M) {
// func TestMain(m *testing.M) {
// client := redis.NewClient(&redis.Options{
// Addr: "127.0.0.1" + ":" + "6379",
// Password: "", // no password set
// DB: 0, // use default DB
// })
// if client == nil {
// fmt.Println("redis init error")
// return
// }
// // fmt.Println("ffff")
// Redis = client
// }
func TestLockx(t *testing.T) {
client := redis.NewClient(&redis.Options{
Addr: "127.0.0.1" + ":" + "6379",
Password: "", // no password set
@@ -21,22 +35,19 @@ func TestMain(m *testing.M) {
fmt.Println("redis init error")
return
}
Redis = client
}
func TestLockx(t *testing.T) {
fmt.Println("begin")
ctx := context.Background()
ctx, cancel := context.WithCancel(ctx)
defer cancel()
lock := lockx.NewGlobalLock(ctx, Redis, "lockx:test")
lock := lockx.NewGlobalLock(ctx, client, "lockx:test")
if !lock.Lock() {
t.Log("lock error")
fmt.Println("lock error")
}
defer lock.Unlock()
lock.Refresh()
t.Log("doing")
fmt.Println("ssss")
}