Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
xindoo
redis
提交
2bd13cf0
R
redis
项目概览
xindoo
/
redis
通知
2
Star
2
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
R
redis
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
2bd13cf0
编写于
6月 05, 2016
作者:
Y
Yossi Gottlieb
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Allow passing arguments to modules on load.
上级
550fa7e1
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
46 addition
and
13 deletion
+46
-13
src/config.c
src/config.c
+15
-2
src/module.c
src/module.c
+24
-10
src/server.h
src/server.h
+7
-1
未找到文件。
src/config.c
浏览文件 @
2bd13cf0
...
...
@@ -153,6 +153,19 @@ void resetServerSaveParams(void) {
server
.
saveparamslen
=
0
;
}
void
queueLoadModule
(
sds
path
,
sds
*
argv
,
int
argc
)
{
struct
loadmodule
*
loadmod
=
zmalloc
(
sizeof
(
struct
loadmodule
)
+
sizeof
(
sds
)
*
argc
);
int
i
;
loadmod
->
path
=
sdsnew
(
path
);
loadmod
->
argc
=
argc
;
for
(
i
=
0
;
i
<
argc
;
i
++
)
{
loadmod
->
argv
[
i
]
=
sdsnew
(
argv
[
i
]);
}
listAddNodeTail
(
server
.
loadmodule_queue
,
loadmod
);
}
void
loadServerConfigFromString
(
char
*
config
)
{
char
*
err
=
NULL
;
int
linenum
=
0
,
totlines
,
i
;
...
...
@@ -632,8 +645,8 @@ void loadServerConfigFromString(char *config) {
"Allowed values: 'upstart', 'systemd', 'auto', or 'no'"
;
goto
loaderr
;
}
}
else
if
(
!
strcasecmp
(
argv
[
0
],
"loadmodule"
)
&&
argc
=
=
2
)
{
listAddNodeTail
(
server
.
loadmodule_queue
,
sdsnew
(
argv
[
1
])
);
}
else
if
(
!
strcasecmp
(
argv
[
0
],
"loadmodule"
)
&&
argc
>
=
2
)
{
queueLoadModule
(
argv
[
1
],
&
argv
[
2
],
argc
-
2
);
}
else
if
(
!
strcasecmp
(
argv
[
0
],
"sentinel"
))
{
/* argc == 1 is handled by main() as we need to enter the sentinel
* mode ASAP. */
...
...
src/module.c
浏览文件 @
2bd13cf0
...
...
@@ -2897,11 +2897,11 @@ void moduleLoadFromQueue(void) {
listRewind
(
server
.
loadmodule_queue
,
&
li
);
while
((
ln
=
listNext
(
&
li
)))
{
s
ds
modulepath
=
ln
->
value
;
if
(
moduleLoad
(
modulepath
)
==
C_ERR
)
{
s
truct
loadmodule
*
loadmod
=
ln
->
value
;
if
(
moduleLoad
(
loadmod
->
path
,(
void
**
)
loadmod
->
argv
,
loadmod
->
argc
)
==
C_ERR
)
{
serverLog
(
LL_WARNING
,
"Can't load module from %s: server aborting"
,
module
path
);
loadmod
->
path
);
exit
(
1
);
}
}
...
...
@@ -2915,8 +2915,8 @@ void moduleFreeModuleStructure(struct RedisModule *module) {
/* Load a module and initialize it. On success C_OK is returned, otherwise
* C_ERR is returned. */
int
moduleLoad
(
const
char
*
path
)
{
int
(
*
onload
)(
void
*
);
int
moduleLoad
(
const
char
*
path
,
void
**
module_argv
,
int
module_argc
)
{
int
(
*
onload
)(
void
*
,
void
**
,
int
);
void
*
handle
;
RedisModuleCtx
ctx
=
REDISMODULE_CTX_INIT
;
...
...
@@ -2925,14 +2925,14 @@ int moduleLoad(const char *path) {
serverLog
(
LL_WARNING
,
"Module %s failed to load: %s"
,
path
,
dlerror
());
return
C_ERR
;
}
onload
=
(
int
(
*
)(
void
*
))(
unsigned
long
)
dlsym
(
handle
,
"RedisModule_OnLoad"
);
onload
=
(
int
(
*
)(
void
*
,
void
**
,
int
))(
unsigned
long
)
dlsym
(
handle
,
"RedisModule_OnLoad"
);
if
(
onload
==
NULL
)
{
serverLog
(
LL_WARNING
,
"Module %s does not export RedisModule_OnLoad() "
"symbol. Module not loaded."
,
path
);
return
C_ERR
;
}
if
(
onload
((
void
*
)
&
ctx
)
==
REDISMODULE_ERR
)
{
if
(
onload
((
void
*
)
&
ctx
,
module_argv
,
module_argc
)
==
REDISMODULE_ERR
)
{
if
(
ctx
.
module
)
moduleFreeModuleStructure
(
ctx
.
module
);
dlclose
(
handle
);
serverLog
(
LL_WARNING
,
...
...
@@ -3006,16 +3006,30 @@ int moduleUnload(sds name) {
/* Redis MODULE command.
*
* MODULE LOAD <path> */
* MODULE LOAD <path>
[args...]
*/
void
moduleCommand
(
client
*
c
)
{
char
*
subcmd
=
c
->
argv
[
1
]
->
ptr
;
if
(
!
strcasecmp
(
subcmd
,
"load"
)
&&
c
->
argc
==
3
)
{
if
(
moduleLoad
(
c
->
argv
[
2
]
->
ptr
)
==
C_OK
)
if
(
!
strcasecmp
(
subcmd
,
"load"
)
&&
c
->
argc
>=
3
)
{
sds
*
argv
=
NULL
;
int
argc
=
0
;
int
i
;
if
(
c
->
argc
>
3
)
{
argc
=
c
->
argc
-
3
;
argv
=
zmalloc
(
sizeof
(
sds
)
*
argc
);
for
(
i
=
0
;
i
<
argc
;
i
++
)
{
argv
[
i
]
=
(
sds
)
c
->
argv
[
i
+
3
]
->
ptr
;
}
}
if
(
moduleLoad
(
c
->
argv
[
2
]
->
ptr
,(
void
**
)
argv
,
argc
)
==
C_OK
)
addReply
(
c
,
shared
.
ok
);
else
addReplyError
(
c
,
"Error loading the extension. Please check the server logs."
);
if
(
argv
)
zfree
(
argv
);
}
else
if
(
!
strcasecmp
(
subcmd
,
"unload"
)
&&
c
->
argc
==
3
)
{
if
(
moduleUnload
(
c
->
argv
[
2
]
->
ptr
)
==
C_OK
)
addReply
(
c
,
shared
.
ok
);
...
...
src/server.h
浏览文件 @
2bd13cf0
...
...
@@ -683,6 +683,12 @@ struct saveparam {
int
changes
;
};
struct
loadmodule
{
sds
path
;
int
argc
;
sds
argv
[];
};
struct
sharedObjectsStruct
{
robj
*
crlf
,
*
ok
,
*
err
,
*
emptybulk
,
*
czero
,
*
cone
,
*
cnegone
,
*
pong
,
*
space
,
*
colon
,
*
nullbulk
,
*
nullmultibulk
,
*
queued
,
...
...
@@ -1156,7 +1162,7 @@ extern dictType modulesDictType;
/* Modules */
void
moduleInitModulesSystem
(
void
);
int
moduleLoad
(
const
char
*
path
);
int
moduleLoad
(
const
char
*
path
,
void
**
argv
,
int
argc
);
void
moduleLoadFromQueue
(
void
);
int
*
moduleGetCommandKeysViaAPI
(
struct
redisCommand
*
cmd
,
robj
**
argv
,
int
argc
,
int
*
numkeys
);
moduleType
*
moduleTypeLookupModuleByID
(
uint64_t
id
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录