set the selected DB in lua

alicebob · Feb 17, 2021 · eca505a · eca505a
1 parent 653d2da
commit eca505a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/integration/script_test.go b/integration/script_test.go
@@ -196,6 +196,15 @@ func TestLua(t *testing.T) {
 			"EVAL", `return cjson.decode(1, 2)`, "0",
 		)
 	})
+
+	// selected DB gets passed on to lua
+	testRaw(t, func(c *client) {
+		c.Do("SELECT", "3")
+		c.Do("EVAL", "redis.call('SET', 'foo', 'bar')", "0")
+		c.Do("GET", "foo")
+		c.Do("SELECT", "0")
+		c.Do("GET", "foo")
+	})
 }
 
 func TestLuaCall(t *testing.T) {

diff --git a/lua.go b/lua.go
@@ -19,6 +19,7 @@ func mkLuaFuncs(srv *server.Server, c *server.Peer) map[string]lua.LGFunction {
 			pCtx.authenticated = true
 		}
 		pCtx.nested = true
+		pCtx.selectedDB = getCtx(c).selectedDB
 
 		return func(l *lua.LState) int {
 			top := l.GetTop()