hotime/cache/cache_db.go

package cache

import (
	. "../common"
	"database/sql"
	"encoding/json"
	"strings"
	"time"
)

type HoTimeDBInterface interface {
	GetPrefix() string
	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 {
	TimeOut    int64
	DbSet      bool
	SessionSet bool
	Db         HoTimeDBInterface
	*Error
	ContextBase
	isInit bool
}

func (this *CacheDb) GetError() *Error {

	return this.Error

}

func (this *CacheDb) SetError(err *Error) {
	this.Error = err
}

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='" + that.Db.GetPrefix() + "cached'")
		if len(res) != 0 {
			that.isInit = true
			return
		}

		_, e := that.Db.Exec("CREATE TABLE `" + that.Db.GetPrefix() + "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 = '` + that.Db.GetPrefix() + `cached'`)

		if len(res) != 0 {
			that.isInit = true
			return
		}
		_, e := that.Db.Exec(`CREATE TABLE "` + that.Db.GetPrefix() + `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.TimeOut == 0 {
			//that.Time = Config.GetInt64("cacheLongTime")
		}
		tim += that.TimeOut
	}
	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}
}