161 lines
3.6 KiB
Go
161 lines
3.6 KiB
Go
package cache
|
|
|
|
import (
|
|
. "../common"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type HoTimeDBInterface interface {
|
|
Query(query string, args ...interface{}) []Map
|
|
Exec(query string, args ...interface{}) (sql.Result, *Error)
|
|
Get(table string, qu ...interface{}) Map
|
|
Select(table string, qu ...interface{}) []Map
|
|
Delete(table string, data map[string]interface{}) int64
|
|
Update(table string, data Map, where Map) int64
|
|
Insert(table string, data map[string]interface{}) int64
|
|
GetType() string
|
|
}
|
|
|
|
type CacheDb struct {
|
|
Time int64
|
|
Db HoTimeDBInterface
|
|
Error
|
|
ContextBase
|
|
isInit bool
|
|
}
|
|
|
|
func (that *CacheDb) initDbTable() {
|
|
if that.isInit {
|
|
return
|
|
}
|
|
if that.Db.GetType() == "mysql" {
|
|
|
|
dbNames := that.Db.Query("SELECT DATABASE()")
|
|
|
|
if len(dbNames) == 0 {
|
|
return
|
|
}
|
|
dbName := dbNames[0].GetString("DATABASE()")
|
|
res := that.Db.Query("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA='" + dbName + "' AND TABLE_NAME='cached'")
|
|
if len(res) != 0 {
|
|
that.isInit = true
|
|
return
|
|
}
|
|
|
|
_, e := that.Db.Exec("CREATE TABLE `cached` ( `id` int(11) unsigned NOT NULL AUTO_INCREMENT, `ckey` varchar(60) DEFAULT NULL, `cvalue` varchar(2000) DEFAULT NULL, `time` bigint(20) DEFAULT NULL, `endtime` bigint(20) DEFAULT NULL, PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=198740 DEFAULT CHARSET=utf8")
|
|
if e.GetError() == nil {
|
|
that.isInit = true
|
|
}
|
|
|
|
}
|
|
|
|
if that.Db.GetType() == "sqlite" {
|
|
res := that.Db.Query(`select * from sqlite_master where type = 'table' and name = 'cached'`)
|
|
|
|
if len(res) != 0 {
|
|
that.isInit = true
|
|
return
|
|
}
|
|
_, e := that.Db.Exec(`CREATE TABLE "cached" (
|
|
"id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
"ckey" TEXT(60),
|
|
"cvalue" TEXT(2000),
|
|
"time" integer,
|
|
"endtime" integer
|
|
);`)
|
|
if e.GetError() == nil {
|
|
that.isInit = true
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
//获取Cache键只能为string类型
|
|
func (that *CacheDb) get(key string) interface{} {
|
|
|
|
cached := that.Db.Get("cached", "*", Map{"ckey": key})
|
|
|
|
if cached == nil {
|
|
return nil
|
|
}
|
|
//data:=cacheMap[key];
|
|
if cached.GetInt64("endtime") <= time.Now().Unix() {
|
|
|
|
that.Db.Delete("cached", Map{"id": cached.GetString("id")})
|
|
return nil
|
|
}
|
|
|
|
data := Map{}
|
|
data.JsonToMap(cached.GetString("cvalue"))
|
|
|
|
return data.Get("data")
|
|
}
|
|
|
|
//key value ,时间为时间戳
|
|
func (that *CacheDb) set(key string, value interface{}, tim int64) {
|
|
|
|
bte, _ := json.Marshal(Map{"data": value})
|
|
|
|
num := that.Db.Update("cached", Map{"cvalue": string(bte), "time": time.Now().UnixNano(), "endtime": tim}, Map{"ckey": key})
|
|
if num == int64(0) {
|
|
that.Db.Insert("cached", Map{"cvalue": string(bte), "time": time.Now().UnixNano(), "endtime": tim, "ckey": key})
|
|
}
|
|
|
|
//随机执行删除命令
|
|
if Rand(1000) > 950 {
|
|
that.Db.Delete("cached", Map{"endtime[<]": time.Now().Unix()})
|
|
}
|
|
}
|
|
|
|
func (that *CacheDb) delete(key string) {
|
|
|
|
del := strings.Index(key, "*")
|
|
//如果通配删除
|
|
if del != -1 {
|
|
key = Substr(key, 0, del)
|
|
that.Db.Delete("cached", Map{"ckey": key + "%"})
|
|
|
|
} else {
|
|
that.Db.Delete("cached", Map{"ckey": key})
|
|
}
|
|
}
|
|
|
|
func (that *CacheDb) Cache(key string, data ...interface{}) *Obj {
|
|
|
|
that.initDbTable()
|
|
|
|
if len(data) == 0 {
|
|
return &Obj{Data: that.get(key)}
|
|
}
|
|
tim := time.Now().Unix()
|
|
|
|
if len(data) == 1 && data[0] == nil {
|
|
that.delete(key)
|
|
return &Obj{Data: nil}
|
|
}
|
|
|
|
if len(data) == 1 {
|
|
if that.Time == 0 {
|
|
//that.Time = Config.GetInt64("cacheLongTime")
|
|
}
|
|
tim += that.Time
|
|
}
|
|
if len(data) == 2 {
|
|
that.SetError(nil)
|
|
tempt := ObjToInt64(data[1], &that.Error)
|
|
|
|
if tempt > tim {
|
|
tim = tempt
|
|
} else if that.GetError() == nil {
|
|
|
|
tim = tim + tempt
|
|
}
|
|
}
|
|
that.set(key, data[0], tim)
|
|
return &Obj{Data: nil}
|
|
}
|