mirror of
https://github.com/haiwen/ccnet-server.git
synced 2025-04-27 18:25:06 +00:00
82 lines
1.7 KiB
C
82 lines
1.7 KiB
C
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
|
|
#if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)
|
|
#include <event2/event.h>
|
|
#include <event2/event_compat.h>
|
|
#include <event2/event_struct.h>
|
|
#else
|
|
#include <event.h>
|
|
#endif
|
|
|
|
#include <sys/time.h>
|
|
|
|
#ifdef CCNET_LIB
|
|
#include "libccnet_utils.h"
|
|
#define timeval_from_msec ccnet_util_timeval_from_msec
|
|
#else
|
|
#include "utils.h"
|
|
#endif
|
|
|
|
#include "timer.h"
|
|
|
|
struct CcnetTimer
|
|
{
|
|
struct event event;
|
|
struct timeval tv;
|
|
TimerCB func;
|
|
void *user_data;
|
|
uint8_t inCallback;
|
|
};
|
|
|
|
static void
|
|
timer_callback (evutil_socket_t fd, short event, void *vtimer)
|
|
{
|
|
int more;
|
|
struct CcnetTimer *timer = vtimer;
|
|
|
|
timer->inCallback = 1;
|
|
more = (*timer->func) (timer->user_data);
|
|
timer->inCallback = 0;
|
|
|
|
if (more)
|
|
evtimer_add (&timer->event, &timer->tv);
|
|
else
|
|
ccnet_timer_free (&timer);
|
|
}
|
|
|
|
void
|
|
ccnet_timer_free (CcnetTimer **ptimer)
|
|
{
|
|
CcnetTimer *timer;
|
|
|
|
/* zero out the argument passed in */
|
|
g_return_if_fail (ptimer);
|
|
|
|
timer = *ptimer;
|
|
*ptimer = NULL;
|
|
|
|
/* destroy the timer directly or via the command queue */
|
|
if (timer && !timer->inCallback)
|
|
{
|
|
event_del (&timer->event);
|
|
g_free (timer);
|
|
}
|
|
}
|
|
|
|
CcnetTimer*
|
|
ccnet_timer_new (TimerCB func,
|
|
void *user_data,
|
|
uint64_t interval_milliseconds)
|
|
{
|
|
CcnetTimer *timer = g_new0 (CcnetTimer, 1);
|
|
|
|
timer->tv = timeval_from_msec (interval_milliseconds);
|
|
timer->func = func;
|
|
timer->user_data = user_data;
|
|
|
|
evtimer_set (&timer->event, timer_callback, timer);
|
|
evtimer_add (&timer->event, &timer->tv);
|
|
|
|
return timer;
|
|
}
|