snippet/redis/redis.go

package redis

import (
	"encoding/json"
	"time"

	redigo "github.com/gomodule/redigo/redis"
)

var pool *redigo.Pool

// Config 配置
type Config struct {
	Host        string `yml:"host" json:"host"`
	Password    string `yml:"password" json:"password"`
	Database    int    `yml:"database" json:"database"`
	MaxIdle     int    `yml:"max_idle" json:"max_idle"`
	MaxActive   int    `yml:"max_active" json:"max_active"`
	IdleTimeout int    `yml:"idle_timeout" json:"idle_timeout"` //second
}

// Init init
func Init(opts *Config) error {
	if opts == nil {
		opts = &Config{}
	}

	if opts.Host == "" {
		opts.Host = "127.0.0.1:6379"
	}
	if opts.MaxIdle == 0 {
		opts.MaxIdle = 1
	}
	if opts.MaxActive == 0 {
		opts.MaxActive = 10
	}
	if opts.IdleTimeout == 0 {
		opts.IdleTimeout = 600
	}

	pool = &redigo.Pool{
		MaxActive:   opts.MaxActive,
		MaxIdle:     opts.MaxIdle,
		IdleTimeout: time.Second * time.Duration(opts.IdleTimeout),
		Dial: func() (redigo.Conn, error) {
			return redigo.Dial("tcp", opts.Host,
				redigo.DialDatabase(opts.Database),
				redigo.DialPassword(opts.Password),
			)
		},
		TestOnBorrow: func(conn redigo.Conn, t time.Time) error {
			if time.Since(t) < time.Minute {
				return nil
			}
			_, err := conn.Do("PING")
			return err
		},
	}
	return nil
}

// GetBytes 获取一个字节数组值
func GetBytes(key *string) (*[]byte, error) {
	conn := pool.Get()
	defer conn.Close()

	data, err := redigo.Bytes(conn.Do("GET", *key))
	return &data, err
}

// Get 获取一个值
func Get(key string) interface{} {
	conn := pool.Get()
	defer conn.Close()

	var data []byte
	var err error
	if data, err = redigo.Bytes(conn.Do("GET", key)); err != nil {
		return nil
	}
	var reply interface{}
	if err = json.Unmarshal(data, &reply); err != nil {
		return nil
	}

	return reply
}

// 集合Set增加元素
func SetAdd(key string, data ...interface{}) error {
	conn := pool.Get()
	defer conn.Close()
	var err error
	for _, d := range data {
		_, e := conn.Do("SADD", key, d)
		if e != nil {
			err = e
			break
		}
	}
	return err
}

// 集合Set删除元素
func SetRem(key string, data ...interface{}) error {
	conn := pool.Get()
	defer conn.Close()

	var err error
	for _, d := range data {
		_, e := conn.Do("SREM", key, d)
		if e != nil {
			err = e
			break
		}
	}

	return err
}

// 集合Set判断是否存在成员member，结果.(int64)==1表示存在
func SetIsMember(key string, member interface{}) (interface{}, error) {
	conn := pool.Get()
	defer conn.Close()

	return conn.Do("SISMEMBER", key, member)
}

// 设置一个值
func Set(key string, val interface{}, timeout time.Duration) error {
	data, err := json.Marshal(val)
	if err != nil {
		return err
	}
	return SetBytes(&key, &data, timeout)
}

func SetBytes(key *string, data *[]byte, timeout time.Duration) error {
	conn := pool.Get()
	defer conn.Close()

	_, err := conn.Do("SETEX", *key, int64(timeout/time.Second), *data)
	return err
}

// IsExist 判断key是否存在
func IsExist(key string) bool {
	conn := pool.Get()
	defer conn.Close()

	a, _ := conn.Do("EXISTS", key)
	i := a.(int64)
	return i > 0
}

// Delete 删除
func Delete(key string) error {
	conn := pool.Get()
	defer conn.Close()

	if _, err := conn.Do("DEL", key); err != nil {
		return err
	}

	return nil
}

// Expire 失效时间配置
func Expire(key string, t int64) error {
	conn := pool.Get()
	defer conn.Close()

	if _, err := conn.Do("expire", key, t); err != nil {
		return err
	}
	return nil
}