zTC1/TC1/timed_task/timed_task.c

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<stdbool.h>
#include<time.h>
#include"timed_task/timed_task.h"

pTimedTask task_top = NULL;
int task_count = 0;

bool AddTask(pTimedTask task)
{
    task_count++;
    if (task_top == NULL)
    {
        task->next = NULL;
        task_top = task;
        return true;
    }

    if (task->time <= task_top->time)
    {
        task->next = task_top;
        task_top = task;
        return true;
    }

    pTimedTask tmp = task_top;
    while (tmp)
    {
        if ((task->time > tmp->time && task->time <= tmp->next->time)
            || tmp->next == NULL)
        {
            task->next = tmp->next;
            tmp->next = task;
            return true;
        }
        tmp = tmp->next;
    }
    task_count--;
    return false;
}

bool DelFirstTask()
{
    if (task_top)
    {
        pTimedTask tmp = task_top;
        task_top = task_top->next;
        free(tmp);
        task_count--;
        return true;
    }
    return false;
}

bool DelTask(int time)
{
    if (task_top == NULL)
    {
        return false;
    }

    if (time == task_top->time)
    {
        pTimedTask tmp = task_top;
        task_top = task_top->next;
        free(tmp);
        task_count--;
        return true;
    }
    else if (task_top->next == NULL)
    {
        return false;
    }

    pTimedTask pre_tsk = task_top;
    pTimedTask tmp_tsk = task_top->next;
    while (tmp_tsk)
    {
        if (time == tmp_tsk->time)
        {
            pre_tsk->next = tmp_tsk->next;
            free(tmp_tsk);
            task_count--;
            return true;
        }
        tmp_tsk = tmp_tsk->next;
    }
    return false;
}

char* GetTaskStr()
{
    char* str = (char*)malloc(sizeof(char)*task_count * 52);
    pTimedTask tmp_tsk = task_top;
    char* tmp_str = str;
    tmp_str[0] = '[';
    tmp_str++;
    while (tmp_tsk)
    {
        char buffer[26];
        struct tm* tm_info;
        tm_info = localtime(&tmp_tsk->time);
        strftime(buffer, 26, "%Y-%m-%d %H:%M:%S", tm_info);

        sprintf(tmp_str, "{time:'%s',socket_index:%d,on:%d},",
            buffer, tmp_tsk->socket_idx, tmp_tsk->on);
        tmp_str += strlen(tmp_str);
        tmp_tsk = tmp_tsk->next;
    }
    *(--tmp_str) = ']';
    return str;
}