Golang的memcache如何簡單實現(xiàn)

下面由golang教程欄目給大家介紹Golang簡單的memcache實現(xiàn)方法，希望對需要的朋友有所幫助！

這兩天在做項目的過程中遇到了一個訪問全局變量的問題場景：編寫一個方法，獲取id對應(yīng)的token值，token需要緩存起來（全局變量內(nèi)存緩存），如果獲取不到或者token的時間過期，那么發(fā)送http請求到其他端去取，然后緩存起來，然后再返回，那么代碼如下：

code.go:

package person  import ( 	"time" )  var gAccId2Token map[int]interface{} = make(map[int]interface{})  func GetTokenByAccountId(accountId uint, acl string) (map[string]interface{}, error) { 	//get token from cache 	if token, ok := gAccId2Token[accountId]; ok { 		if token != nil { 			now := time.Now().Unix() 			if int(now) < int(token.(map[string]interface{})["expireDate"].(float64)) { 				return token.(map[string]interface{}), nil 			} 		} 	}  	token, err := getTokenByHttpUrl(apiUrl) 	if err != nil { 		return map[string]interface{}{}, err 	}  	gAccId2Token[accountId] = token  	return token.(map[string]interface{}), nil }

那么問題來了：

1.由于gAccId2Token變量是全局變量，那么會出現(xiàn)同時讀寫的情況，則會可能出現(xiàn)讀寫不一致的情況。

2.就本例來看，獲取id=2的token，而緩存的token已經(jīng)過期了，那么就會發(fā)送http請求去獲取，之后寫緩存，假設(shè)寫緩存的時間很長，而在這段時間內(nèi)，又恰好有大量請求來獲取id=2的token，由于token都過期了，就會出現(xiàn)大量請求http服務(wù)端的問題，不僅沒有起到獲取緩存的目的，又增大了后端的壓力，同時又有多個寫緩存的操作，而golang的map應(yīng)該不是原子的，那么大量寫內(nèi)存也可能會造成crash的問題。

因此，我們需要對讀寫操作進行加鎖：

memcache.go：

package person  import ( 	"sync" 	"time" )  type memoryCache struct { 	lock  *sync.RWMutex 	items map[interface{}]interface{} }  func (mc *memoryCache) set(key interface{}, value interface{}) error { 	mc.lock.Lock() 	defer mc.lock.Unlock() 	mc.items[key] = value 	return nil }  func (mc *memoryCache) get(key interface{}) interface{} { 	mc.lock.RLock() 	defer mc.lock.RUnlock()  	if val, ok := mc.items[key]; ok { 		return val 	} 	return nil }  var gAccId2Token *memoryCache = &memoryCache{ 		lock:  new(sync.RWMutex), 		items: make(map[interface{}]interface{}), 	}  func GetTokenByAccountId(accountId uint, acl string) (map[string]interface{}, error) { 	//get token from cache 	token := gAccId2Token.get(accountId) 	if token != nil { 		now := time.Now().Unix() 		if int(now) < int(token.(map[string]interface{})["expireDate"].(float64)) { 			return token.(map[string]interface{}), nil 		} 	}  	token, err := getTokenByHttpUrl(apiUrl) 	if err != nil { 		return map[string]interface{}{}, err 	}  	gAccId2Token.set(accountId, token)  	return token.(map[string]interface{}), nil }

幾點說明：

1.為寫操作上了全局鎖，一旦Lock()之后，其他lock便不能上鎖，直到釋放鎖Unlock()之后才行，也就是說保證寫操作的原子性。

2.而為讀操作上了讀鎖，那么可以有多個線程Rlock()對一個區(qū)域枷鎖，從而保證區(qū)域是可讀的，直到所有讀鎖都RUnlock()之后，才可以上寫鎖。

3.將map的key和value的類型都定義成為interface{}類型，interface{}可以接收任何類型，就像是Java中的Object。

4.interface{}類型轉(zhuǎn)換的方法，value.(type)，即將value轉(zhuǎn)換成為type類型，例如：value.(int)，value.(map[string]interface{})等等。