pdxx
/
libserverframe
mirror of https://github.com/happyfish100/libserverframe.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add function sf_enable_thread_notify_ex

connection_manager
YuQing 2020-03-11 15:28:59 +08:00
parent 46dfd4ad01
commit 115b1e7750
4 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/sf_nio.c
View File

@ -109,10 +109,11 @@ static inline int set_write_event(struct fast_task_info *task)
return 0;
}
static inline int set_read_event(struct fast_task_info *task)
int sf_set_read_event(struct fast_task_info *task)
{
int result;
task->nio_stage = SF_NIO_STAGE_RECV;
if (task->event.callback == (IOEventCallback)sf_client_sock_read) {
return 0;
}
@ -166,7 +167,7 @@ static int sf_nio_deal_task(struct fast_task_info *task)
result = sf_nio_init(task);
break;
case SF_NIO_STAGE_RECV:
if ((result=set_read_event(task)) == 0)
if ((result=sf_set_read_event(task)) == 0)
{
sf_client_sock_read(task->event.fd,
IOEVENT_READ, task);
@ -546,8 +547,7 @@ int sf_client_sock_write(int sock, short event, void *arg)
if (task->offset >= task->length) {
task->offset = 0;
task->length = 0;
task->nio_stage = SF_NIO_STAGE_RECV;
if (set_read_event(task) != 0) {
if (sf_set_read_event(task) != 0) {
return -1;
}
break;

2
src/sf_nio.h
View File

@ -52,6 +52,8 @@ void sf_task_finish_clean_up(struct fast_task_info *pTask);
int sf_nio_notify(struct fast_task_info *pTask, const int stage);
int sf_set_read_event(struct fast_task_info *task);
void sf_task_switch_thread(struct fast_task_info *pTask,
const int new_thread_index);

13
src/sf_service.c
View File

@ -612,3 +612,16 @@ void sf_set_current_time()
g_sf_global_vars.up_time = g_current_time;
srand(g_sf_global_vars.up_time);
}
void sf_enable_thread_notify_ex(SFContext *sf_context, const bool enabled)
{
struct nio_thread_data *thread_data;
struct nio_thread_data *pDataEnd;
pDataEnd = sf_context->thread_data + sf_context->work_threads;
for (thread_data=sf_context->thread_data; thread_data<pDataEnd;
thread_data++)
{
thread_data->notify.enabled = enabled;
}
}

5
src/sf_service.h
View File

@ -56,6 +56,11 @@ void sf_accept_loop_ex(SFContext *sf_context, const bool block);
#define sf_accept_loop() sf_accept_loop_ex(&g_sf_context, true)
void sf_enable_thread_notify_ex(SFContext *sf_context, const bool enabled);
#define sf_enable_thread_notify(enabled) \
sf_enable_thread_notify_ex(&g_sf_context, enabled)
#ifdef __cplusplus
}
#endif