golang实现自定义驱动的Cache

近期在写 ActivedRouter项目的时候需求一个缓存模型，要求缓存模型支持不同驱动,例如:memory、file、redis、mysql，实现思路代码如下:

cache.go文件,定义缓存对外接口

//ActivedRouter
//Author:usher.yue
//Amail:usher.yue@gmail.com
//TencentQQ:4223665
//缓存驱动接口定义
//可以自定义扩展驱动类型 默认是 memory driver

package cache

import (
	"./driver"
)

//cache接口声明
type Cacher interface {
	GetStorage() driver.Containerer
	Set(k string,v interface{})
	Get(k string) (interface{},bool)
	Del(k string)
	Has(k string) bool
}

//create memory cache
//type "file" or "memory"
func Newcache(cacheType string) Cacher {
	switch cacheType {
	case "memory":
		return &driver.CacheImpl{Driver: driver.NewMapContainer()}
	case "file":
		{
			return &driver.CacheImpl{Driver: driver.NewFileContainer()}
		}
	case "mysql":
		{
			return &driver.CacheImpl{Driver: driver.NewMysqlContainer()}
		}
	case "redis":
		{
			return &driver.CacheImpl{Driver: driver.NewRedisContainer()}
		}
	case "mongodb":
		{
			return &driver.CacheImpl{Driver: driver.NewMongoContainer()}
		}
	}
	return nil
}

driver.go 实现驱动,这里我只利用Map实现了，具体其他驱动可自行实现

//ActivedRouter
//Author:usher.yue
//Amail:usher.yue@gmail.com
//TencentQQ:4223665
//缓存驱动接口定义
//可以自定义扩展驱动类型 默认是 memory cache

package driver

//容器接口声明
type Containerer interface {
	PushKVPair(k,v interface{}) Containerer
	EraseKVPair(k interface{}) Containerer
	PushKVMaps(maps ...map[string]interface{}) Containerer
	ResetKVPair(k string,v interface{}) Containerer
	ResetOrAddKVPair(k string,v interface{}) Containerer
	ResetKVPairs(kvMaps map[string]interface{}) Containerer
	ResetOrAddKVPairs(kvMaps map[string]interface{}) Containerer
	Exist(k interface{}) bool
	GetData() *map[string]interface{}
}

//基于内存实现的缓存
type CacheImpl struct {
	Driver Containerer //数据缓存驱动
}

func (self *CacheImpl) Exist(k interface{}) bool {

	return self.Driver.Exist(k)
}

func (self *CacheImpl) GetStorage() Containerer {
	return self.Driver
}

//set
func (self *CacheImpl) Set(k string,v interface{}) {
	self.Driver.PushKVPair(k,v)
}

//get
func (self *CacheImpl) Get(k string) (interface{},bool) {
	mapData := *self.Driver.GetData()
	val,ok := mapData[k]
	return val,ok
}

//erase
func (self *CacheImpl) Del(k string) {
	self.Driver.EraseKVPair(k)
}

//has
func (self *CacheImpl) Has(k string) bool {
	mapData := *self.Driver.GetData()
	_,ok := mapData[k]
	return ok
}

//Map实现的内存驱动
type MapContainer struct {
	data map[string]interface{}
}

//创建数据channer
func NewMapContainer() *MapContainer {
	return &MapContainer{data: make(map[string]interface{})}
}

func (this *MapContainer) PushKVPair(k,v interface{}) Containerer {
	if key,ok := k.(string); !ok {
		panic("key必须是string类型!")
	} else {
		this.data[key] = v
	}
	return this
}

func (this *MapContainer) Exist(k interface{}) bool {
	return true
}

func (this *MapContainer) EraseKVPair(k interface{}) Containerer {
	if key,ok := k.(string); !ok {
		panic("key必须是string类型!")
	} else {
		delete(this.data,key)
	}
	return this
}
func (this *MapContainer) PushKVMaps(maps ...map[string]interface{}) Containerer {
	for _,itemMap := range maps {
		for itemKey,itemValue := range itemMap {
			this.PushKVPair(itemKey,itemValue)
		}
	}
	return this
}
func (this *MapContainer) ResetKVPair(k string,v interface{}) Containerer {
	if _,ok := this.data[k]; ok {
		this.data[k] = v
	}
	return this
}
func (this *MapContainer) ResetOrAddKVPair(k string,v interface{}) Containerer {
	this.data[k] = v
	return this
}

func (this *MapContainer) ResetKVPairs(kvMaps map[string]interface{}) Containerer {
	for k,v := range kvMaps {
		if _,ok := this.data[k]; ok {
			this.data[k] = v
		}
	}
	return this
}

func (this *MapContainer) ResetOrAddKVPairs(kvMaps map[string]interface{}) Containerer {
	for k,v := range kvMaps {
		this.data[k] = v
	}
	return this
}

func (this *MapContainer) GetData() *map[string]interface{} {
	return &this.data
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!