redisgo 操作lua script

原创
2019/08/20 15:46
阅读数 883
  • lua script filename: test.lua
local key=KEYS[1]
local val=ARGV[1]
return redis.call('SET', key, val)
  • go code
package main

import (
	"fmt"
	"io/ioutil"
	"time"

	"github.com/gomodule/redigo/redis"
)

func CliPool() *redis.Pool {
	return &redis.Pool{
		MaxIdle:     10,
		MaxActive:   100,
		IdleTimeout: time.Duration(20) * time.Second,
		Dial: func() (redis.Conn, error) {
			c, err := redis.Dial("tcp", "127.0.0.1:6379")
			if err != nil {
				return nil, err
			}
			return c, nil
		},
	}
}
func main() {
	var data []byte
	var err error
	if data, err = ioutil.ReadFile("test.lua"); err != nil {
		fmt.Printf("read failed %v", err.Error())
		return
	}
	script := redis.NewScript(1, string(data))
	pool := CliPool()
	defer pool.Close()

	cli := pool.Get()
	script.Load(cli)

	arg := []string{"key", "value"}
	args := redis.Args{}.AddFlat(arg)
	var reply interface{}
	if reply, err = script.Do(cli, args...); err != nil {
		fmt.Printf("redis exec failed %v", err.Error())
		return
	}
	fmt.Printf("return %v", reply)
}

script.Do 的参数一定要搞对,否则很容易出错, 要么填arg[0], arg[1] 这种形式或者用redis.Args{}往里面填参数,其他方式容易出错. 要保证通过(s *Script) args(spec string, keysAndArgs []interface{}) 方法操作之后所有的参数都在一个slice 中

展开阅读全文
打赏
0
0 收藏
分享
加载中
您好,大佬,看到您发达redigo使用lua脚本操作,我比较困惑,想请教一下您,如果是使用redigo脚本进行批量插入操作,例如一个slice中有许多struct,并且struct中都是key,score,member值,这样应该怎么利用redigo进行脚本插入redis呢?
05/27 09:52
回复
举报
更多评论
打赏
2 评论
0 收藏
0
分享
返回顶部
顶部