Redis之并发写入
那么开始总结下方案吧：
- 1. SETNX key value //key存在就不做任何操作，返回0；不存在操作成功返回1
复制代码
这种方式通过对需要操作的key加锁来保证并发操作的串行化。这里我们以Golang代码为例来举例说明该操作。先看多个协程写同一个key的情况。代码如下：
package main
import (
"fmt"
"github.com/garyburd/redigo/redis"
"runtime"
"sync"
"time"
)
var w sync.WaitGroup
func newRdsPool(server, auth string) *redis.Pool {
return &redis.Pool{
MaxIdle: 100,
MaxActive: 30,
IdleTimeout: 60 * time.Second,
Dial: func() (redis.Conn, error) {
c, err := redis.Dial("tcp", server)
if err != nil {
return nil, err
}
if auth == "" {
return c, err
}
if _, err := c.Do("AUTH", auth); err != nil {
c.Close()
return nil, err
}
return c, err
},
TestOnBorrow: func(c redis.Conn, t time.Time) error {
_, err := c.Do("PING")
return err
},
}
}
func g1(r redis.Conn) {
for i := 0; i < 2; i++ {
if _, err := redis.String(r.Do("set", "hello", "1")); err != nil {
fmt.Println(err)
}
time.Sleep(10 * time.Millisecond)
}
w.Done()
}
func g2(r redis.Conn) {
for i := 0; i < 2; i++ {
if _, err := redis.String(r.Do("set", "hello", "2")); err != nil {
fmt.Println(err)
}
time.Sleep(10 * time.Millisecond)
}
w.Done()
}
func main() {
w.Add(2)
runtime.GOMAXPROCS(runtime.NumCPU())
var rc1 redis.Conn = newRdsPool(`127.0.0.1:6379`, ``).Get()
var rc2 redis.Conn = newRdsPool(`127.0.0.1:6379`, ``).Get()
defer rc1.Close()
defer rc2.Close()
go g1(rc1)
go g2(rc2)
w.Wait()
}执行上面的代码之后，hello的值在1和2之间徘徊。希望出现的是如果协程1在操作时候，协程2就放弃操作，也即让操作串行化。这样就需要有一个锁来保证不能同时让两个协程进去临界区。setnx = set if not exists 不存在返回1，存在返回0。通过这个机制可以判断当前的lock是否已经被设置了。lock必须给一个过期时间，因为很有可能goroutine1在do work的时候出现panic，这样就导致goroutine2一直在尝试获取锁。
package main
import (
"fmt"
"github.com/garyburd/redigo/redis"
"runtime"
"sync"
"time"
)
var w sync.WaitGroup
func newRdsPool(server, auth string) *redis.Pool {
return &redis.Pool{
MaxIdle: 100,
MaxActive: 30,
IdleTimeout: 60 * time.Second,
Dial: func() (redis.Conn, error) {
c, err := redis.Dial("tcp", server)
if err != nil {
return nil, err
}
if auth == "" {
return c, err
}
if _, err := c.Do("AUTH", auth); err != nil {
c.Close()
return nil, err
}
return c, err
},
TestOnBorrow: func(c redis.Conn, t time.Time) error {
_, err := c.Do("PING")
return err
},
}
}
func g1(r redis.Conn) {
var lock int64
var lock_timeout int64 = 2
var lock_time int64
var now int64
for lock != 1 {
now = time.Now().Unix()
lock_time = now + lock_timeout
lock, err1 := redis.Int64(r.Do("setnx", "foo", lock_time))
lockValue1, err2 := redis.Int64(r.Do("get", "foo"))
if lock == 1 && err1 == nil {
break
} else {
if now > lockValue1 && err2 == nil {
lockValue2, err3 := redis.Int64(r.Do("getset", "foo", lock_time))
if err3 == nil && now > lockValue2 {
break
} else {
fmt.Println(`g1 not get lock`)
time.Sleep(1000 * time.Millisecond)
}
} else {
fmt.Println(`g1 not get lock`)
time.Sleep(1000 * time.Millisecond)
}
}
}
for i := 0; i < 5; i++ {
if _, err := redis.String(r.Do("set", "hello", "1")); err != nil {
fmt.Println(err)
}
fmt.Println(`g1 now work... `)
time.Sleep(1 * time.Second)
}
if time.Now().Unix() < lock_time {
if _, err4 := redis.Int64(r.Do("del", "foo")); err4 != nil {
fmt.Println(err4)
}
}
w.Done()
}
func g2(r redis.Conn) {
var lock int64
var lock_timeout int64 = 2
var lock_time int64
var now int64
for lock != 1 {
now = time.Now().Unix()
lock_time = now + lock_timeout
lock, err1 := redis.Int64(r.Do("setnx", "foo", lock_time))
lockValue1, err2 := redis.Int64(r.Do("get", "foo"))
if lock == 1 && err1 == nil {
break
} else {
if now > lockValue1 && err2 == nil {
lockValue2, err3 := redis.Int64(r.Do("getset", "foo", lock_time))
if err3 == nil && now > lockValue2 {
break
} else {
fmt.Println(`g2 not get lock`)
time.Sleep(1000 * time.Millisecond)
}
} else {
fmt.Println(`g2 not get lock`)
time.Sleep(1000 * time.Millisecond)
}
}
}
for i := 0; i < 5; i++ {
if _, err := redis.String(r.Do("set", "hello", "2")); err != nil {
fmt.Println(err)
}
fmt.Println(`g2 now work... `)
time.Sleep(1 * time.Second)
}
if time.Now().Unix() < lock_time {
if _, err4 := redis.Int64(r.Do("del", "foo")); err4 != nil {
fmt.Println(err4)
}
}
w.Done()
}
func main() {
w.Add(2)
runtime.GOMAXPROCS(runtime.NumCPU())
var rc1 redis.Conn = newRdsPool(`127.0.0.1:6379`, ``).Get()
var rc2 redis.Conn = newRdsPool(`127.0.0.1:6379`, ``).Get()
defer rc1.Close()
defer rc2.Close()
go g1(rc1)
go g2(rc2)
w.Wait()
}上面的代码给出了两个goroutine通过锁达到串行化操作同一个key的效果。 2. MULTI、DISCARD、 EXEC、WATCH // redis事务 3. 将对key的操作的值都放到一个list里面
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2017-08-29，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
相关产品与服务
云数据库 Redis
腾讯云数据库 Redis（TencentDB for Redis）是腾讯云打造的兼容 Redis 协议的缓存和存储服务。丰富的数据结构能帮助您完成不同类型的业务场景开发。支持主从热备，提供自动容灾切换、数据备份、故障迁移、实例监控、在线扩容、数据回档等全套的数据库服务。