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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
|
/*************************************************************************
> File Name: lua_plugin_cfunc.c
> Author:
> Created Time: 2024-08
> Encoding : UTF-8
************************************************************************/
/*************************************************************************
* version
* [ v0.1 ]
* 08-06
* 1. 实现函数
* void *lpm_ctx_new_func;
* void lpm_ctx_free_func;
*
* 08-13
* 1. 由于context结构体修改, 部分函数调用及处理逻辑需要同步修改
*
* 08-15
* 1. 实现函数
* void lpm_message_free_func
* void lpm_on_session_msg_func
************************************************************************/
#include "lua_plugin_manage_internal.h"
#include "stellar/session.h"
/*
* Function: lpm_ctx_new_func
* Input: | struct session * | sess | 会话信息
* | void * | plugin_env | 插件运行环境
* Output:
* Return: | NULL | 运行函数错误
* | pointer | 运行插件上的ctx_new函数成功, 返回context
* Description: 与C插件管理器保持一致的ctx_new_func
*/
void *lpm_ctx_new_func(
struct session *sess,
void *plugin_env)
{
if (__glibc_unlikely(!sess || !plugin_env))
return NULL;
struct lua_model *env = (struct lua_model *)plugin_env;
/* 获取插件ID并找到该插件 */
int plugin_id = session_get_current_plugin_id(sess);
// int plugin_id = 1;
struct lua_plugin *plugin = NULL;
while ((plugin = utarray_next(env->plugin_array, plugin)))
{
if (plugin->plugin_id == plugin_id)
break;
}
// LOGDEBUG("%d call plugin id %d, ref %d\n", plugin_id, plugin->plugin_id, plugin->ctx_new_ref);
if (!plugin || plugin->plugin_id != plugin_id)
/* 未找到该插件 */
return NULL;
/* 获取当前的线程id并找到该线程对应的state */
int thread_id = session_get_current_thread_id(sess);
if (thread_id > global_schema->state_count)
return NULL;
lua_State *state = global_schema->thread_state[thread_id];
struct lua_context *new_context = lua_context_new(state);
if (__glibc_unlikely(!new_context))
/* 创建新的context失败 */
return NULL;
struct lua_cdata param[3] = {0};
param[0].cdata_type = DATATYPE_POINTER;
param[0].cdata_pointer = sess;
param[1].cdata_type = DATATYPE_TABLE;
param[1].cdata_table = env->private_env_ref;
param[2].cdata_type = DATATYPE_CONTEXT;
param[2].cdata_context = new_context;
if (lua_chunk_execute(state, plugin->ctx_new_ref, 3, param, 0, NULL))
{
/* 脚本执行失败 */
lua_context_free(state, new_context);
return NULL;
}
return (void *)new_context;
}
/*
* Function: lpm_ctx_free_func
* Input: | struct session * | sess | 会话信息
* | void * | sess_ctx | 会话中的私有数据
* | void * | plugin_env | 插件运行环境
* Output:
* Return:
* Description: 与C插件管理器保持一致的ctx_free_func
*/
void lpm_ctx_free_func(
struct session *sess,
void *sess_ctx,
void *plugin_env)
{
if (__glibc_unlikely(!sess || !sess_ctx || !plugin_env))
return;
struct lua_model *env = (struct lua_model *)plugin_env;
/* 获取插件ID并找到该插件 */
int plugin_id = session_get_current_plugin_id(sess);
// int plugin_id = 1;
struct lua_plugin *plugin = NULL;
while ((plugin = utarray_next(env->plugin_array, plugin)))
{
if (plugin->plugin_id == plugin_id)
break;
}
if (!plugin || plugin->plugin_id != plugin_id)
/* 未找到该插件 */
return;
int thread_id = session_get_current_thread_id(sess);
if (thread_id > global_schema->state_count)
return;
lua_State *state = global_schema->thread_state[thread_id];
struct lua_cdata param[3] = {0};
param[0].cdata_type = DATATYPE_POINTER;
param[0].cdata_pointer = sess;
param[1].cdata_type = DATATYPE_CONTEXT;
param[1].cdata_context = (struct lua_context *)sess_ctx;
param[2].cdata_type = DATATYPE_TABLE;
param[2].cdata_table = env->private_env_ref;
lua_chunk_execute(state, plugin->ctx_free_ref, 3, param, 0, NULL);
lua_context_free(state, (struct lua_context *)sess_ctx);
return;
}
/*
* Function: lpm_message_free_func
* Input: | struct session * | sess | 释放message对应的session
* | void * | msg | 需要释放的message消息
* | void * | msg_free_arg | 释放函数需要存储的私有数据
* Output:
* Return:
* Description: 与C插件保持一致的session_msg_free_cb_func, 作为lua插件实现对应功能的通用函数
*/
void lpm_message_free_func(
struct session *sess,
void *msg,
void *msg_free_arg)
{
if (__glibc_unlikely(!sess || !msg || !msg_free_arg))
return;
struct lua_message_mq *plugin_mq = (struct lua_message_mq *)msg_free_arg;
int thread_id = session_get_current_thread_id(sess);
if (thread_id > global_schema->state_count)
return;
lua_State *state = global_schema->thread_state[thread_id];
struct lua_cdata param[3] = {0};
param[0].cdata_type = DATATYPE_POINTER;
param[0].cdata_pointer = sess;
param[1].cdata_type = DATATYPE_CONTEXT;
param[1].cdata_context = (struct lua_context *)msg;
param[2].cdata_type = DATATYPE_TABLE;
param[2].cdata_table = plugin_mq->mq_private_ref;
lua_chunk_execute(state, plugin_mq->freemessage_ref, 3, param, 0, NULL);
lua_context_free(state, (struct lua_context *)msg);
return;
}
/*
* Function: lpm_on_session_msg_func
* Input: | struct session * | sess | 会话信息
* | int | topic_id | 处理的message topic id
* | const void * | msg | 需要处理的消息信息
* | void * | sess_ctx | 会话中的私有数据
* | void * | plugin_env | 插件运行环境
* Output:
* Return:
* Description: 与C插件管理器保持一致的ctx_free_func
*/
void lpm_on_session_msg_func(
struct session *sess,
int topic_id,
const void *msg,
void *sess_ctx,
void *plugin_env)
{
if (__glibc_unlikely(!sess || !sess || !plugin_env))
return;
struct lua_context *sess_context = (struct lua_context *)sess_ctx;
struct lua_model *env = (struct lua_model *)plugin_env;
/* 获取插件ID并找到该插件 */
int plugin_id = session_get_current_plugin_id(sess);
// int plugin_id = 1;
struct lua_plugin *plugin = NULL;
while ((plugin = utarray_next(env->plugin_array, plugin)))
{
if (plugin->plugin_id == plugin_id)
break;
}
if (!plugin || plugin->plugin_id != plugin_id || !plugin->sub_topic_array)
/* 未找到该插件, 或者在该插件中未发现注册的topic_id */
return;
struct lua_plugin_mq *mq = NULL;
while ((mq = utarray_next(plugin->sub_topic_array, mq)))
{
if (mq->topic_id == topic_id)
break;
}
if (!mq || mq->topic_id != topic_id)
/* 未找到对应的消息处理函数 */
return;
/* 判断该消息是由C端插件产生的还是由Lua插件产生的 */
struct lua_message_mq *message_mq = search_message_mq_by_id(topic_id);
int thread_id = session_get_current_thread_id(sess);
if (thread_id > global_schema->state_count)
return;
lua_State *state = global_schema->thread_state[thread_id];
struct lua_cdata params[5] = {0};
params[0].cdata_type = DATATYPE_POINTER;
params[0].cdata_pointer = sess;
params[1].cdata_type = DATATYPE_INT;
params[1].cdata_int = topic_id;
if (!message_mq)
{
/* C端产生的直接使用指针 */
params[2].cdata_type = DATATYPE_POINTER;
params[2].cdata_pointer = (void *)msg;
}
else
{
/* lua端产生的为一个context结构 */
params[2].cdata_type = DATATYPE_CONTEXT;
params[2].cdata_context = (struct lua_context *)msg;
}
params[3].cdata_type = DATATYPE_CONTEXT;
params[3].cdata_context = sess_context;
params[4].cdata_type = DATATYPE_TABLE;
params[4].cdata_table = env->private_env_ref;
lua_chunk_execute(state, mq->onmessage_ref, 5, params, 0, NULL);
return;
}
|
