快捷導(dǎo)航

C語言實現(xiàn)多線程定時器實例講解

更新時間：2021年01月03日 09:36:48 作者：chegxy

在本篇文章里小編給各位分享的是一篇關(guān)于C語言實現(xiàn)多線程定時器實例講解內(nèi)容，有需要的朋友們可以參考學(xué)習(xí)下。

1. 大致功能介紹

實現(xiàn)任務(wù)列表，定時器會間隔一段時間遍歷列表發(fā)現(xiàn)要執(zhí)行的任務(wù)
任務(wù)列表中的所有任務(wù)并行執(zhí)行
每個任務(wù)都可以有自己的定時器，并且可以選擇是否要重復(fù)執(zhí)行
定義方便的任務(wù)函數(shù)實現(xiàn)接口
定時器可以由用戶自定義何時啟動和停止
提供等待功能，保證任務(wù)列表中的所有任務(wù)執(zhí)行完成
提供任務(wù)列表的傳參功能

2. API庫介紹

void setTick(int val);

設(shè)置定時間的間隔時間tick，若設(shè)置tick為1000，且任務(wù)的定時器時間為1000，則任務(wù)會在1秒后執(zhí)行，默認tick為1秒，最小tick時間為1us。

void addTimerTask(TimerTask task, int val, int autoreset, void *arg);

向任務(wù)列表注冊一個任務(wù)，并指定其定時時間val，以及是否要重復(fù)執(zhí)行autoreset，并可以指定參數(shù)的地址。
task需要按照頭文件提供的宏來編寫，例如：

TASK_START(test2, arg)

	//body
 Arg *temp = (Arg*)arg;
 temp->ret = temp->a + temp->b;
 printf("This is a test2\n");

TASK_END

TASK_START（name, arg）是任務(wù)頭，name是任務(wù)名，arg是參數(shù)地址，TASK_END是任務(wù)結(jié)尾。任務(wù)體內(nèi)可編寫正常的c語言代碼，并使用參數(shù)arg指針。

autoreset有兩個可選項：AUTORESET（重復(fù)執(zhí)行），NORESET（執(zhí)行一次）。

若沒有參數(shù)，可將arg參數(shù)設(shè)置為NULL。

void TimerWait();

用于等待任務(wù)列表中所有任務(wù)執(zhí)行完畢。

void TimerStop();

用于停止定時器。

void StartTimer();

用于啟動定時器。

3. 一個例子

#include <stdio.h>
#include "timer.h"

typedef struct Argument{
 int a;
 int b;
 int ret;
}Arg;

//任務(wù)1，打印語句
TASK_START(test1, arg)
 printf("This is a test1\n");
TASK_END

//任務(wù)2，計算arg中兩個數(shù)的和，打印語句
TASK_START(test2, arg)

 Arg *temp = (Arg*)arg;
 temp->ret = temp->a + temp->b;
 printf("This is a test2\n");

TASK_END

//任務(wù)3，打印語句
TASK_START(test3, arg)
 printf("This is a test3\n");
TASK_END

void main(){

 Arg arg;

	//設(shè)置tick 為 500ms
 setTick(500 * 1000);

	//添加任務(wù)1到任務(wù)列表，設(shè)置定時器時間為2.5s，重復(fù)執(zhí)行，無參數(shù)
 addTimerTask(test1, 5, AUTORESET, NULL);
 arg.a = 2; arg.b = 3;
 //添加任務(wù)2到任務(wù)列表，設(shè)置定時器時間為0.5s，不重復(fù)執(zhí)行，參數(shù)為arg
 addTimerTask(test2, 1, NORESET, &arg);
 //添加任務(wù)3到任務(wù)列表，設(shè)置定時器時間為1s，重復(fù)執(zhí)行，無參數(shù)
 addTimerTask(test3, 2, AUTORESET, NULL);

	//啟動定時器
 StartTimer();
 printf("Timer is started\n");
 //程序等待5秒
 sleep(5);
 //停止定時器
 TimerStop();
 //等待所有任務(wù)執(zhí)行完畢
 TimerWait();
 //打印任務(wù)二的計算結(jié)果
 printf("%d\n", arg.ret);

}

運行結(jié)果：

4. 庫文件源碼

timer.h:

#ifndef TIMER_H
#define TIMER_H
#include <unistd.h>
#include <stdio.h>
#include <pthread.h>
#include <stdlib.h>
#include <signal.h>
#define AUTORESET 1
#define NORESET 0
#define TASK_START(name, arg) void* name(void *arg){
#define TASK_END return NULL;} 
typedef void* (*TimerTask)(void* arg);
struct TaskItem{
 TimerTask task;
 int init_counter;
 int counter;
 pthread_t th;
 void *arg;
 void *ret;
 int flag;
 int autoreset;
 struct TaskItem *next;
 
};
void setTick(int val);
void* EventLoop(void* arg);
void addTimerTask(TimerTask task, int val, int autoreset, void *arg);
void TimerWait();
void TimerStop();
void StartTimer();
#endif //TIMER_H

timer.cpp

#include "timer.h"
#define STOPFLAG 0
#define RUNFLAG 1 
static int tick = 1000 * 1000;
static struct TaskItem head = {
 .next = NULL,
};
static pthread_t loop_thread;
static int flag = STOPFLAG;
static int tasknum = 0;

void setTick(int val){
 tick = val;
}
void* EventLoop(void* arg){
 
 struct TaskItem *task = head.next;
 struct TaskItem *pretask = &head;

 while(flag == RUNFLAG && tasknum > 0){
  
  while(task != NULL){
   if(task->counter == 0){ // it is time for doing task
    if(task->flag == STOPFLAG){ // task is not created
     if(0 != pthread_create(&(task->th), NULL, task->task, task->arg)){ // do a task
      printf("Failed to create user's task");
     }
     else{
      task->flag = RUNFLAG;
     }
    }
    else{
     if(0 != pthread_kill(task->th, 0)){ // current task is completed
      if(task->autoreset == AUTORESET){ // repeat execute
       task->counter = task->init_counter;
       task->flag = STOPFLAG;
      }
      else{ // delete a task
       pretask->next = task->next;
       free(task);
       task = pretask->next;
       tasknum--;
       continue;
      }
     }
    }
   }
   else{
    task->counter--;
   }
   pretask = pretask->next;
   task = task->next;
  }
  usleep(tick); // sleep a tick
  task = head.next;
  pretask = &head;
 }
 flag = STOPFLAG;
}
void addTimerTask(TimerTask task, int val, int autoreset, void *arg){
 struct TaskItem *node;
 node = (struct TaskItem*)malloc(sizeof(struct TaskItem));
 node->next = head.next;
 head.next = node;
 node->arg = arg;
 node->counter = val;
 node->init_counter = val;
 node->task = task;
 node->flag = STOPFLAG;
 node->autoreset = autoreset;
 tasknum++;
}
void TimerWait(){
 pthread_join(loop_thread, NULL);
}
void TimerStop(){
 flag = STOPFLAG;
}
void StartTimer(){
 flag = RUNFLAG;
 if(0 != pthread_create(&loop_thread, NULL, EventLoop, NULL)){
  printf("Failed to create loop task.\n");
 }
}