Go語言中怎麼透過Lua腳本操作Redis

前言

為了在我的一個基本函式庫中降低Redis的通訊成本，我將一系列作業封裝到LUA腳本中，借助Redis提供的EVAL指令來簡化操作。

EVAL能夠提供的特性：

• #可以在LUA腳本中封裝若干操作，如果有多條Redis指令，封裝好之後只需向Redis一次發送所有參數即可獲得結果

• Redis可以保證Lua腳本運行期間不會有其他命令插入執行，提供像資料庫事務一樣的原子性

• Redis會根據腳本的SHA值快取腳本，已經快取過的腳本不需要再次傳輸Lua程式碼，減少了通訊成本，此外在自己程式碼中改變Lua腳本，執行時Redis必定也會使用最新的程式碼。

導入常見的Go函式庫如 "github.com/go-redis/redis"，就可以實作下列程式碼。

產生一段Lua腳本

// KEYS: key for record
// ARGV: fieldName, currentUnixTimestamp, recordTTL
// Update expire field of record key to current timestamp, and renew key expiration
var updateRecordExpireScript = redis.NewScript(`
redis.call("EXPIRE", KEYS[1], ARGV[3])
redis.call("HSET", KEYS[1], ARGV[1], ARGV[2])
return 1
`)

該變數建立時，Lua程式碼不會被執行，也不需要有已儲存的Redis連線。

Redis提供的Lua腳本支持，預設有KEYS、ARGV兩個數組，KEYS代表腳本運行時傳入的若干鍵值，ARGV代表傳入的若干參數。由於Lua程式碼需要保持簡潔，難免難以讀懂，最好為這些參數寫一些註解

#注意：上面一段程式碼使用``跨行，`所在的行雖然空白回車，也會被認為是一行，報錯時不要看錯代碼行號。

執行一段Lua腳本

 updateRecordExpireScript.Run(c.Client, []string{recordKey(key)}, 
         expireField,
         time.Now().UTC().UnixNano(), int64(c.opt.RecordTTL/time.Second)).Err()

執行時，Run將會先透過EVALSHA嘗試透過快取執行腳本。如果沒有緩存，則使用EVAL運行，這時Lua腳本才會被整個傳入Redis。

Lua腳本的限制

• Redis不提供引入額外的包，例如os等，只有redis這一包可用。

• Lua腳本將會在一個函數中執行，所有變數必須使用local宣告

• return傳回多個值時，Redis將會只給你第一個

腳本中的型別限制

• #腳本回傳nil時，Go中得到的是err = redis.Nil（與Get找不到值相同）

• #腳本回傳false時，Go中得到的是nil，當腳本回傳true時， Go中得到的是int64型別的1

• 在腳本回傳{"ok": ...}時，Go中得到的是redis的status型別（true/false)

• 腳本回傳{"err": ...}時，Go中得到的是err值，也可以透過return redis.error_reply("My Error")達成

• 腳本傳回number型別時，Go中得到的是int64型別

• 在腳本中，如果需要將傳入的KEYS/ ARGV中的值從字串類型轉換為數字類型，則應使用to_number函數

#如果腳本運行了很久會發生什麼？

Lua腳本運行期間，為了避免被其他操作污染數據，這段期間將不能執行其它命令，一直等到執行完畢才可以繼續執行其它請求。當Lua腳本執行時間超過了lua-time-limit時，其他請求將會收到Busy錯誤，除非這些請求是SCRIPT KILL（殺掉腳本）或SHUTDOWN NOSAVE（不儲存結果直接關閉Redis）

更多內容參考以下地址，我這裡主要是根據使用Go的經驗提供一些總結。 https://redis.io/commands/eval

一段更「複雜」的腳本，它要求在取得一個key值時，如果該值訪問較多，就延長生存週期。另外也要比較更新時間，如果不需要更新，直接傳回取到的值，否則回傳redis.Nil

// KEYS: rec:key, key
// ARGV: currentUnixTimestamp, hotHit, recordTTL, ttl
// When there's a hit,
var fetchRecordScript = redis.NewScript(local value = redis.call("GET", KEYS[2]) if(value == nil) then return nil end local hit = redis.call("HINCRBY", KEYS[1], "hit", 1) redis.call("EXPIRE", KEYS[1], ARGV[3]) local minHotHit = tonumber(ARGV[2]) local keyTTL = tonumber(ARGV[4]) if(hit > minHotHit)then keyTTL = keyTTL * 2 end redis.call("EXPIRE", KEYS[2], keyTTL) local expire = tonumber(redis.call("HGET", KEYS[1], "expire")) local unixTime = tonumber(ARGV[1]) if(expire == nil or expire < unixTime) then return nil else return value end)
// KEYS: key for record
// ARGV: fieldName, currentUnixTimestamp, recordTTL
// Update expire field of record key to current timestamp, and renew key expiration
var updateRecordExpireScript = redis.NewScript(redis.call("EXPIRE", KEYS[1], ARGV[3]) redis.call("HSET", KEYS[1], ARGV[1], ARGV[2]) return 1)

以上是Go語言中怎麼透過Lua腳本操作Redis的詳細內容。更多資訊請關注PHP中文網其他相關文章！