当前位置:首页 > 开发 > 数据库 > 正文

redis执行lua lua执行redis

发表于: 2015-04-15   作者:ronin47   来源:转载   浏览次数:
摘要: redis-2.6支持通过EVAL命令来执行lua脚本,对lua脚本的支持扩展了redis的应用场景,redis支持路脚本需要做2件事 redis能执行lua脚本 在lua脚本里能执行redis的命令 接下来,我将通过一个简单的实例来解析redis如何完成上述两个工作的。   构建一个简单的redis #define DICT_SIZE 100 struct re

redis-2.6支持通过EVAL命令来执行lua脚本,对lua脚本的支持扩展了redis的应用场景,redis支持路脚本需要做2件事

  1. redis能执行lua脚本
  2. 在lua脚本里能执行redis的命令

接下来,我将通过一个简单的实例来解析redis如何完成上述两个工作的。

 

构建一个简单的redis

#define DICT_SIZE 100
struct redisDict {
  char* key[DICT_SIZE];
  char* value[DICT_SIZE];
  int  idx;
};

static void setCommand(const char *key, const char *value)
{
    /* ignore memory issue for simple */
    if (dict.idx + 1 <= DICT_SIZE) {
    dict.key[dict.idx] = (char *)malloc(strlen(key) + 1);
    strcpy(dict.key[dict.idx], key);

    dict.value[dict.idx] = (char *)malloc(strlen(value) + 1);
    strcpy(dict.value[dict.idx], value);

    dict.idx += 1;
  }
}

static const char *getCommand(const char *key)
{
  int j;
  for (j = 0; j <= dict.idx; j++) {
    if (strcmp(dict.key[j], key) == 0) {
      return dict.value[j];
    }
  }
  return "KeyNotFound";
}

上述代码实现了一个伪redis,支持setCommand、getCommand。

C调用lua脚本

具体例子参考http://lua-users.org/wiki/SimpleLuaApiExample

/*
 * All Lua contexts are held in this structure. We work with it almost
 * all the time.
 */
lua_State *L = luaL_newstate();

luaL_openlibs(L); /* Load Lua libraries */

/* Load the file containing the script we are going to run */
status = luaL_loadfile(L, "script.lua");

/* Ask Lua to run our little script */
result = lua_pcall(L, 0, LUA_MULTRET, 0);

上述代码片段中,其中script.lua是一个lua脚本。redis里稍有不同,redis里的脚本是通过EVAL命令传递到服务器端,redis将脚本拼成一个lua函数,然后调用loadbuffer,而这里从文件执行脚本调用的loadfile。

lua调用C函数

下面的lua代码里,调用的是redis的setCommand和getCommand。

redis.call("set", "foo", "bar");

return redis.call("get", "foo");

要想lua脚本能调用C代码,需要现在lua环境注册对应的C函数,参考redis的scriptingInit函数。

static int call(lua_State *L)
{
  int argc = lua_gettop(L);
  const char *cmd = lua_tostring(L, 1);
  const char *key = lua_tostring(L, 2);
  if (strcmp(cmd, "set") == 0) {
    assert(argc == 3);
    const char *value = lua_tostring(L, 3);
    setCommand(key, value);
   return 0;
  } else if (strcmp(cmd, "get") == 0) {
    assert(argc == 2);
    lua_pushstring(L, getCommand(key));
    return 1;
  }

  lua_pushstring(L, "Invalid Command");
  return 1;
}

static void scriptingInit()
{

  L = luaL_newstate();

  luaL_openlibs(L);

  /* Register the redis commands table and fields */
  lua_newtable(L);

  /* redis.call */
  lua_pushstring(L, "call");
  lua_pushcfunction(L, call);
  lua_settable(L, -3);

  /* Finally set the table as 'redis' global var. */
  lua_setglobal(L, "redis");
}

完整示例代码

 1
 2
redis . call ( " set" , " foo" , " bar" );
return redis . call ( " get" , " foo" );
来自CODE的代码片
script.lua

 

   1
   2
   3
   4
   5
   6
   7
   8
   9
  10
  11
  12
  13
  14
  15
  16
  17
  18
  19
  20
  21
  22
  23
  24
  25
  26
  27
  28
  29
  30
  31
  32
  33
  34
  35
  36
  37
  38
  39
  40
  41
  42
  43
  44
  45
  46
  47
  48
  49
  50
  51
  52
  53
  54
  55
  56
  57
  58
  59
  60
  61
  62
  63
  64
  65
  66
  67
  68
  69
  70
  71
  72
  73
  74
  75
  76
  77
  78
  79
  80
  81
  82
  83
  84
  85
  86
  87
  88
  89
  90
  91
  92
  93
  94
  95
  96
  97
  98
  99
 100
 101
 102
 103
 104
 105
 106
 107
 108
 109
 110
 111
 112
 113
 114
 115
 116
/* gcc -o test_lua lua.c -llua -lm -ldl */
#include <lua.h>
#include <lauxlib.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>
 
#define DICT_SIZE 100
struct redisDict {
char * key [ DICT_SIZE ];
char * value [ DICT_SIZE ];
int idx ;
};
 
struct redisDict dict ;
lua_State * L ;
 
static void setCommand ( const char * key , const char * value )
{
/* ignore memory issue for simple */
if ( dict . idx + 1 <= DICT_SIZE ) {
dict . key [ dict . idx ] = ( char * ) malloc ( strlen ( key ) + 1 );
strcpy ( dict . key [ dict . idx ], key );
 
dict . value [ dict . idx ] = ( char * ) malloc ( strlen ( value ) + 1 );
strcpy ( dict . value [ dict . idx ], value );
 
dict . idx += 1 ;
}
}
 
static const char * getCommand ( const char * key )
{
int j ;
for ( j = 0 ; j <= dict . idx ; j ++ ) {
if ( strcmp ( dict . key [ j ], key ) == 0 ) {
return dict . value [ j ];
}
}
return "KeyNotFound" ;
}
 
static int call ( lua_State * L )
{
int argc = lua_gettop ( L );
const char * cmd = lua_tostring ( L , 1 );
const char * key = lua_tostring ( L , 2 );
if ( strcmp ( cmd , "set" ) == 0 ) {
assert ( argc == 3 );
const char * value = lua_tostring ( L , 3 );
setCommand ( key , value );
return 0 ;
} else if ( strcmp ( cmd , "get" ) == 0 ) {
assert ( argc == 2 );
lua_pushstring ( L , getCommand ( key ));
return 1 ;
}
 
lua_pushstring ( L , "Invalid Command" );
return 1 ;
}
 
static void scriptingRun ( const char * filename )
{
int status = luaL_loadfile ( L , filename );
if ( status ) {
fprintf ( stderr , "Couldn't load file: %s \n " , lua_tostring ( L , - 1 ));
exit ( 1 );
}
 
/* Ask Lua to run our little script */
int result = lua_pcall ( L , 0 , LUA_MULTRET , 0 );
if ( result ) {
fprintf ( stderr , "Failed to run script: %s \n " , lua_tostring ( L , - 1 ));
exit ( 1 );
}
 
/* Get the returned value at the top of the stack (index -1) */
const char * value = lua_tostring ( L , - 1 );
 
printf ( "%s \n " , value );
 
lua_pop ( L , 1 ); /* Take the returned value out of the stack */
lua_close ( L ); /* Cya, Lua */
}
 
static void scriptingInit ()
{
/*
* All Lua contexts are held in this structure. We work with it almost
* all the time.
*/
L = luaL_newstate ();
 
/* Load Lua libraries */
luaL_openlibs ( L );
 
/* Register the redis commands table and fields */
lua_newtable ( L );
 
/* redis.call */
lua_pushstring ( L , "call" );
lua_pushcfunction ( L , call );
lua_settable ( L , - 3 );
 
/* Finally set the table as 'redis' global var. */
lua_setglobal ( L , "redis" );
}
 
int main ( void )
{
scriptingInit ();
scriptingRun ( "script.lua" );
return 0 ;
}

redis执行lua lua执行redis

  • 0

    开心

    开心

  • 0

    板砖

    板砖

  • 0

    感动

    感动

  • 0

    有用

    有用

  • 0

    疑问

    疑问

  • 0

    难过

    难过

  • 0

    无聊

    无聊

  • 0

    震惊

    震惊

编辑推荐
2.6版本之后支持嵌入Lua脚本,客户端使用Lua脚本,直接在服务器端原子的执行多条命令 Lua脚本执行过
刚开始接触redis的时候为了了解redis的执行流程,粗略的翻了一下redis的源码然后画了一个草图。下面
此处我说的HTTP服务主要指如访问京东网站时我们看到的热门搜索、用户登录、实时价格、实时库存、服
Redis提供了通过eval命令来执行Lua脚本。下面通过几个小例子来讲述如何在Redis服务端执行Lua脚本。
1 #include "lua.hpp" 2 3 #include <iostream> 4 using namespace std; 5 6 #pragma comment
1、 下载OpenResty和Redis OpenResty下载地址:wget http://openresty.org/download/ngx_openresty-1
1、下载OpenResty和Redis OpenResty下载地址:wget http://openresty.org/download/ngx_openresty-1
1、下载OpenResty和Redis OpenResty下载地址:wget http://openresty.org/download/ngx_openresty-1
Lua称为解释型语言的原因:Lua允许在运行源代码之前,先将源代码预编译为一种中间形式。区别解释型
10 Redis
百度百科 - Redis 征服Redis (JAVAEye) Redis 的安装配置介绍 java使用redis简单说明 [ri' des] 瑞
版权所有 IT知识库 CopyRight © 2009-2015 IT知识库 IT610.com , All Rights Reserved. 京ICP备09083238号