github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-02-02 16:28:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: redislock use stringx.randn replace randomStr func (#1220)

Browse Source
This commit is contained in:
an 2021-11-10 21:14:21 +08:00 committed by GitHub
parent d3bfa16813
commit 3a510a9138
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
core/stores/redis/redis_test.go
View File

@ -3,6 +3,7 @@ package redis
import (
"crypto/tls"
"errors"
"github.com/tal-tech/go-zero/core/stringx"
"io"
"strconv"
"testing"
@ -186,7 +187,7 @@ func TestRedis_Hscan(t *testing.T) {
key := "hash:test"
fieldsAndValues := make(map[string]string)
for i := 0; i < 1550; i++ {
fieldsAndValues["filed_"+strconv.Itoa(i)] = randomStr(i)
fieldsAndValues["filed_"+strconv.Itoa(i)] = stringx.Randn(i)
}
err := client.Hmset(key, fieldsAndValues)
assert.Nil(t, err)
@ -550,7 +551,7 @@ func TestRedis_Sscan(t *testing.T) {
key := "list"
var list []string
for i := 0; i < 1550; i++ {
list = append(list, randomStr(i))
list = append(list, stringx.Randn(i))
}
lens, err := client.Sadd(key, list)
assert.Nil(t, err)

11
core/stores/redis/redislock.go
View File

@ -1,6 +1,7 @@
package redis
import (
"github.com/tal-tech/go-zero/core/stringx"
"math/rand"
"strconv"
"sync/atomic"
@ -45,7 +46,7 @@ func NewRedisLock(store *Redis, key string) *RedisLock {
return &RedisLock{
store: store,
key: key,
id: randomStr(randomLen),
id: stringx.Randn(randomLen),
}
}
@ -92,11 +93,3 @@ func (rl *RedisLock) Release() (bool, error) {
func (rl *RedisLock) SetExpire(seconds int) {
atomic.StoreUint32(&rl.seconds, uint32(seconds))
}
func randomStr(n int) string {
b := make([]byte, n)
for i := range b {
b[i] = letters[rand.Intn(len(letters))]
}
return string(b)
}