libuv定时器

迈不过友情╰ 2022-01-21 10:53 345阅读 0赞

头文件：

#ifndef __MY_TIMER_LIST_H__
    #define __MY_TIMER_LIST_H__
    
    
    struct timer;
    struct timer*
    schedule(void(*on_timer)(void* udata), 
             void* udata, 
    		 int after_msec,
    		 int repeat_count);
    
    void 
    cancel_timer(struct timer* t);
    
    struct timer*
    schedule_once(void(*on_timer)(void* udata), 
                  void* udata, 
    			  int after_msec);
    #endif

c文件（实现文件）

#include <stdio.h>
    #include <string.h>
    #include <stdlib.h>
    
    #include "uv.h"
    #include "time_list.h"
    
    #define my_malloc malloc
    #define my_free free
    
    
    struct timer {
    	uv_timer_t uv_timer; // libuv timer handle
    	void(*on_timer)(void* udata);
    	void* udata;
    	int repeat_count; // -1一直循环;
    };
    
    static struct timer* 
    alloc_timer(void(*on_timer)(void* udata),
                void* udata, int repeat_count) {
    	struct timer* t = my_malloc(sizeof(struct timer));
    	memset(t, 0, sizeof(struct timer));
    
    	t->on_timer = on_timer;
    	t->repeat_count = repeat_count;
    	t->udata = udata;
    	uv_timer_init(uv_default_loop(), &t->uv_timer);
    	return t;
    }
    
    static void
    free_timer(struct timer* t) {
    	my_free(t);
    }
    
    static void
    on_uv_timer(uv_timer_t* handle) {
    	struct timer* t = handle->data;
    	if (t->repeat_count < 0) { // 不断的触发;
    		t->on_timer(t->udata);
    	}
    	else {
    		t->repeat_count --;
    		t->on_timer(t->udata);
    		if (t->repeat_count == 0) { // 函数time结束
    			uv_timer_stop(&t->uv_timer); // 停止这个timer
    			free_timer(t);
    		}
    	}
    	
    }
    
    struct timer*
    schedule(void(*on_timer)(void* udata),
             void* udata,
    		 int after_msec,
             int repeat_count) {
    	struct timer* t = alloc_timer(on_timer, udata, repeat_count);
    
    	// 启动一个timer;
    	t->uv_timer.data = t;
    	uv_timer_start(&t->uv_timer, on_uv_timer, after_msec, after_msec);
    	// end 
    	return t;
    }
    
    void
    cancel_timer(struct timer* t) {
    	if (t->repeat_count == 0) { // 全部触发完成，;
    		return;
    	}
    	uv_timer_stop(&t->uv_timer);
    	free_timer(t);
    }
    
    struct timer*
    schedule_once(void(*on_timer)(void* udata),
                  void* udata,
    			  int after_msec) {
    	return schedule(on_timer, udata, after_msec, 1);
    }