zTC1/TC1/timed_task/timed_task.c

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<stdbool.h>
#include<time.h>

#include"main.h"
#include"user_gpio.h"
#include"timed_task/timed_task.h"
#include"http_server/web_log.h"

pTimedTask task_top = NULL;
int task_count = 0;

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

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

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

bool AddTaskWeek(pTimedTask task)
{
    int day_sec = 86400;
    time_t now = time(NULL);
    int today_weekday = (now / day_sec + 3) % 7 + 1; //1970-01-01 星期五
    int next_day = task->weekday - today_weekday;
    bool next_day_is_today = next_day == 0 && task->prs_time % day_sec > now % day_sec;
    next_day = next_day > 0 || next_day_is_today ? next_day : next_day + 7;
    task->prs_time = (now - now % day_sec) + (next_day * day_sec) + task->prs_time % day_sec;

    return AddTaskSingle(task);
}

bool AddTask(pTimedTask task)
{
    if (task->weekday == 0) return AddTaskSingle(task);
    return AddTaskWeek(task);
}

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->prs_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->prs_time)
        {
            pre_tsk->next = tmp_tsk->next;
            free(tmp_tsk);
            task_count--;
            return true;
        }
        tmp_tsk = tmp_tsk->next;
    }
    return false;
}

void ProcessSingle()
{
    UserRelaySet(task_top->socket_idx, task_top->on);
    DelFirstTask();
}

void ProcessWeek()
{
    //TODO 添加下一周的定时任务
    ProcessSingle();
}

void ProcessTask()
{
    task_log("process task time[%ld] socket_idx[%d] on[%d]",
        task_top->prs_time, task_top->socket_idx, task_top->on);
    if (task_top->weekday == 0) ProcessSingle();
    else ProcessWeek();
}

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

        sprintf(tmp_str, "{'timestamp':%ld,'prs_time':'%s','socket_idx':%d,'on':%d},",
            tmp_tsk->prs_time, buffer, tmp_tsk->socket_idx, tmp_tsk->on);
        tmp_str += strlen(tmp_str);
        tmp_tsk = tmp_tsk->next;
    }
    if (task_count > 0) --tmp_str;
    *tmp_str = ']';
    return str;
}