forked from golang/hotime
154 lines
2.7 KiB
Go
154 lines
2.7 KiB
Go
package cache
|
|
|
|
import (
|
|
"github.com/garyburd/redigo/redis"
|
|
. "go.hoteas.com/hotime"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type CacheRedis struct {
|
|
Host string
|
|
Pwd string
|
|
Time int64
|
|
conn redis.Conn
|
|
tag int64
|
|
Error
|
|
}
|
|
|
|
//唯一标志
|
|
func (this *CacheRedis) GetTag() int64 {
|
|
|
|
if this.tag == int64(0) {
|
|
this.tag = time.Now().UnixNano()
|
|
}
|
|
return this.tag
|
|
}
|
|
|
|
func (this *CacheRedis) reCon() bool {
|
|
var err error
|
|
this.conn, err = redis.Dial("tcp", this.Host)
|
|
if err != nil {
|
|
this.conn = nil
|
|
this.Error.SetError(err)
|
|
return false
|
|
}
|
|
|
|
_, err = this.conn.Do("AUTH", this.Pwd)
|
|
if err != nil {
|
|
this.conn = nil
|
|
this.Error.SetError(err)
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
func (this *CacheRedis) del(key string) {
|
|
del := strings.Index(key, "*")
|
|
if del != -1 {
|
|
val, err := redis.Strings(this.conn.Do("KEYS", key))
|
|
if err != nil {
|
|
return
|
|
}
|
|
this.conn.Send("MULTI")
|
|
for i, _ := range val {
|
|
this.conn.Send("DEL", val[i])
|
|
}
|
|
this.conn.Do("EXEC")
|
|
} else {
|
|
_, err := this.conn.Do("DEL", key)
|
|
if err != nil {
|
|
this.Error.SetError(err)
|
|
_, err = this.conn.Do("PING")
|
|
if err != nil {
|
|
if this.reCon() {
|
|
_, err = this.conn.Do("DEL", key)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//key value ,时间为时间戳
|
|
func (this *CacheRedis) set(key string, value string, time int64) {
|
|
_, err := this.conn.Do("SET", key, value, "EX", ObjToStr(time))
|
|
if err != nil {
|
|
|
|
this.Error.SetError(err)
|
|
_, err = this.conn.Do("PING")
|
|
if err != nil {
|
|
if this.reCon() {
|
|
_, err = this.conn.Do("SET", key, value, "EX", ObjToStr(time))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (this *CacheRedis) get(key string) *Obj {
|
|
reData := &Obj{}
|
|
var err error
|
|
reData.Data, err = redis.String(this.conn.Do("GET", key))
|
|
if err != nil {
|
|
reData.Data = nil
|
|
if !strings.Contains(err.Error(), "nil returned") {
|
|
this.Error.SetError(err)
|
|
_, err = this.conn.Do("PING")
|
|
if err != nil {
|
|
if this.reCon() {
|
|
reData.Data, err = redis.String(this.conn.Do("GET", key))
|
|
}
|
|
}
|
|
|
|
return reData
|
|
}
|
|
}
|
|
return reData
|
|
}
|
|
|
|
func (this *CacheRedis) Cache(key string, data ...interface{}) *Obj {
|
|
reData := &Obj{}
|
|
if this.conn == nil {
|
|
re := this.reCon()
|
|
if !re {
|
|
return reData
|
|
}
|
|
}
|
|
//查询缓存
|
|
if len(data) == 0 {
|
|
|
|
reData = this.get(key)
|
|
return reData
|
|
|
|
}
|
|
tim := int64(0)
|
|
//删除缓存
|
|
if len(data) == 1 && data[0] == nil {
|
|
this.del(key)
|
|
return reData
|
|
}
|
|
//添加缓存
|
|
if len(data) == 1 {
|
|
|
|
if this.Time == 0 {
|
|
this.Time = Config.GetInt64("cacheShortTime")
|
|
}
|
|
|
|
tim += this.Time
|
|
}
|
|
if len(data) == 2 {
|
|
this.Error.SetError(nil)
|
|
tempt := ObjToInt64(data[1], &this.Error)
|
|
if tempt > tim {
|
|
|
|
tim = tempt
|
|
} else if this.GetError() == nil {
|
|
|
|
tim = tim + tempt
|
|
}
|
|
}
|
|
|
|
this.set(key, ObjToStr(data[0]), tim)
|
|
|
|
return reData
|
|
|
|
}
|