在golang中使用lua脚本

zooeymoon · · 8285 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

lua基础

注释:

-- this is a comment

声明全局变量:

x = 123

声明本地变量:

local y = 456

方法声明:

function hello_world()
 return "Hello World"
end

迭代:

for i = 1, 10 do
   print(i)
end

条件:

if x == 123 then
    print("x is the magic number")
else
    print("I have no idea what x is")
end

字符串连接:

print("Hello" .. " World")

作为数组使用一个table — 数组的索引从1开始:

data_types = {1.0, 123, "redis", true, false, hello_world}
print(data_types[3]) -- the output is "redis"

作为hash使用一个table:

languages = {lua = 1993, javascript = 1995, python = 1991, ruby =1995}
print("Lua was created in " .. languages["lua"])
print("JavaScript was created in " .. languages.javascript)

redis脚本执行是一种原子操作,所以在执行期间redis服务会被阻塞。默认脚本操作会有5秒的执行时间
当一个lua脚本执行超时不会自动停止,它会开始对每一个命令回复一个busy状态,表明脚本正在运行。使用命令SCRIPT KILL或者SHUTDOWN NOSAVE终止脚本执行才能使redis服务回到正常状态

举个栗子

package main

import (
    "fmt"
    "github.com/go-redis/redis"
)

var Client *redis.Client

func init() {
    Client = redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "", // no password set
        DB:       0,  // use default DB
    })
}

//noinspection GoInvalidCompositeLiteral
func main() {

    Client.FlushAll()

    Client.Set("foo", "bar" , 0)

    var luaScript = redis.NewScript(`return redis.call("GET" , KEYS[1])`)

    n , err := luaScript.Run(Client , []string{"foo"}).Result()
    if err != nil {
        panic(err)
    }

    fmt.Println(n , err)

}

另一个栗子

package main

import (
    "fmt"
    "github.com/go-redis/redis"
)

var Client *redis.Client

func init() {
    Client = redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "", // no password set
        DB:       0,  // use default DB
    })
}

//noinspection GoInvalidCompositeLiteral
func main() {

    Client.FlushAll()

    foo := []redis.Z{
        {
            1732, "George Washington",
        },
        {
            1809, "Abraham Lincoln",
        },
        {
            1858, "Theodore Roosevelt",
        },
    }

    Client.ZAdd("presidents", foo...)

    var luaScript = redis.NewScript(`
        local elements = redis.call("ZRANGE" , KEYS[1] , 0 , 0) 
        redis.call("ZREM" , KEYS[1] , elements[1])
        return elements[1]
    `)

    n , err := luaScript.Run(Client , []string{"presidents"} , 1).Result()
    if err != nil {
        panic(err)
    }

    fmt.Println(n , err)

}

使用evalsha来操作lua

当执行脚本多次的时候,可以使用SCRIPT LOADEVALSHA代替EVAL有可能节省带宽。命令SCRIPT LOAD缓存一个lua脚本,同时返回一个sha1的标识符。命令EVALSHA基于sha1的标识符执行脚本,通过SCRIPT LOAD返回。使用EVALSHA只有标识符通过网络传输,而不是一个lua代码块

继续举个栗子

package main

import (
    "fmt"
    "github.com/go-redis/redis"
)

var Client *redis.Client

func init() {
    Client = redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "", // no password set
        DB:       0,  // use default DB
    })
}

//noinspection GoInvalidCompositeLiteral
func main() {

    Client.FlushAll()

    Client.Set("foo" , "bar" , -1)

    var luaScript = `return redis.call("INFO")`
    result ,err := Client.ScriptLoad(luaScript).Result()   //返回的脚本会产生一个sha1哈希值,下次用的时候可以直接使用这个值,类似于
    if err != nil {
        panic(err)
    }

    foo :=Client.EvalSha(result ,[]string{})
    fmt.Println(foo.Val())
}

有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:zooeymoon

查看原文:在golang中使用lua脚本

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

8285 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传