forked from mirror/redis
5e2cce4853 | ||
---|---|---|
README.md | ||
commands.go | ||
connpool.go | ||
multi.go | ||
parser.go | ||
pipeline.go | ||
pubsub.go | ||
redis.go | ||
redis_test.go | ||
request.go | ||
request_test.go |
README.md
Readme
Redis client for Golang.
Supports:
- Redis 2.6 commands except QUIT command.
- Pub/sub.
- Transactions.
- Pipelining.
- Connection pool.
- TLS connections.
- Thread safety.
Installation
Install:
go get github.com/vmihailenco/redis
Getting Client instance
Example 1:
import "github.com/vmihailenco/redis"
address := "localhost:6379"
password := "secret"
db := 0
redisClient := redis.NewTCPClient(address, password, db)
Example 2:
import "github.com/vmihailenco/redis"
openConn := func() (io.ReadWriteCloser, error) {
fmt.Println("Connecting...")
return net.Dial("tcp", ":6379")
}
closeConn := func(conn io.ReadWriteCloser) error {
fmt.Println("Disconnecting...")
return nil
}
initConn := func(client *redis.Client) error {
auth := client.Auth("key")
if auth.Err() != nil {
return auth.Err()
}
ping := client.Ping()
if ping.Err() != nil {
return ping.Err()
}
return nil
}
redisClient := redis.NewClient(openConn, closeConn, initConn)
Both closeConn
and initConn
functions can be nil
.
Running commands
set := redisClient.Set("key", "hello")
if set.Err() != nil {
panic(set.Err())
}
ok := set.Val()
get := redisClient.Get("key")
if get.Err() != nil && get.Err() != redis.Nil {
panic(get.Err())
}
val := get.Val()
Pipelining
Client has ability to run commands in batches:
reqs, err := redisClient.Pipelined(func(c *redis.PipelineClient) {
c.Set("key1", "hello1") // queue command SET
c.Set("key2", "hello2") // queue command SET
})
if err != nil {
panic(err)
}
for _, req := range reqs {
// ...
}
Or:
pipeline, err := redisClient.PipelineClient()
if err != nil {
panic(err)
}
defer pipeline.Close()
setReq := pipeline.Set("key1", "hello1") // queue command SET
getReq := pipeline.Get("key2") // queue command GET
reqs, err := pipeline.RunQueued() // run queued commands
if err != nil {
panic(err)
}
for _, req := range reqs {
// ...
}
if setReq.Err() != nil {
panic(setReq.Err())
}
if getReq.Err() != nil && getReq.Err() != redis.Nil {
panic(getReq.Err())
}
Multi/Exec
Example:
func transaction(multi *redis.MultiClient) ([]redis.Req, error) {
get := multiClient.Get("key")
if get.Err() != nil {
panic(get.Err())
}
reqs, err = multiClient.Exec(func() {
multi.Set("key", get.Val() + "1")
})
if err == redis.Nil {
return transaction()
}
return reqs, err
}
multiClient, err := redisClient.MultiClient()
if err != nil {
panic(err)
}
defer multiClient.Close()
watch := multiClient.Watch("key")
if watch.Err() != nil {
panic(watch.Err())
}
reqs, err := transaction(multiClient)
if err != nil {
panic(err)
}
for _, req := range reqs {
// ...
}
Pub/sub
Publish:
pub := redisClient.Publish("mychannel", "hello")
if pub.Err() != nil {
panic(pub.Err())
}
Subscribe:
pubsub, err := redisClient.PubSubClient()
if err != nil {
panic(err)
}
defer pubsub.Close()
ch, err := pubsub.Subscribe("mychannel")
if err != nil {
panic(err)
}
go func() {
for msg := range ch {
if msg.Err != nil {
panic(err)
}
message := msg.Message
}
}
Thread safety
Commands are thread safe. Following code is correct:
for i := 0; i < 1000; i++ {
go func() {
redisClient.Incr("key")
}()
}
Custom commands
Example:
func Get(client *redis.Client, key string) *redis.BulkReq {
req := redis.NewBulkReq("GET", key)
client.Process(req)
return req
}
get := Get(redisClient, "key")
if get.Err() != nil && get.Err() != redis.Nil {
panic(get.Err())
}
Connection pool
Client uses connection pool with default capacity of 10 connections. To change pool capacity:
redisClient.ConnPool.(*redis.MultiConnPool).MaxCap = 1
Look and feel
Some corner cases:
SORT list LIMIT 0 2 ASC
client.Sort("list", redis.Sort{Offset: 0, Count: 2, Order: "ASC"})
ZRANGEBYSCORE zset -inf +inf WITHSCORES LIMIT 0 2
client.ZRangeByScoreWithScores("zset", "-inf", "+inf", 0, 2)
ZINTERSTORE out 2 zset1 zset2 WEIGHTS 2 3 AGGREGATE SUM
client.ZInterStore("out", 2, redis.ZStore{Weights: []int64{2, 3}}, "zset1", "zset2")
Contributing
Configure Redis to allow maximum 10 clients:
maxclients 10
Run tests:
go test -gocheck.v
Run benchmarks:
go test -gocheck.b