Golang实现小型任务提醒工具实例

time.AfterFunc

robfig/cron

time.AfterFunc

package mainimport (    "fmt"    "time")// Task 定义了任务的数据结构type Task struct {    ID          string    `json:"id"`           // 任务唯一标识符    Title       string    `json:"title"`        // 任务标题    Description string    `json:"description"`  // 任务详细描述    DueTime     time.Time `json:"due_time"`     // 提醒时间    IsCompleted bool      `json:"is_completed"` // 任务是否已完成}// String 方法让Task在打印时更具可读性func (t Task) String() string {    status := "待办"    if t.IsCompleted {        status = "已完成"    }    return fmt.Sprintf("ID: %sn标题: %sn描述: %sn提醒时间: %sn状态: %sn",        t.ID, t.Title, t.Description, t.DueTime.Format("2006-01-02 15:04:05"), status)}

json:"..."

DueTime

time.Time

String()

fmt.Println(task)

string

uuid

encoding/json

[]Task

[]Task

package mainimport (    "encoding/json"    "io/ioutil"    "os"    "path/filepath")const dataFile = "tasks.json"// saveTasks 将任务列表保存到JSON文件func saveTasks(tasks []Task) error {    data, err := json.MarshalIndent(tasks, "", "  ") // 使用MarshalIndent让JSON文件更易读    if err != nil {        return fmt.Errorf("序列化任务失败: %w", err)    }    // 确保数据文件所在的目录存在    dir := filepath.Dir(dataFile)    if dir != "" && dir != "." { // 如果dataFile不是在当前目录，需要创建目录        if _, err := os.Stat(dir); os.IsNotExist(err) {            if err := os.MkdirAll(dir, 0755); err != nil {                return fmt.Errorf("创建数据目录失败: %w", err)            }        }    }    err = ioutil.WriteFile(dataFile, data, 0644)    if err != nil {        return fmt.Errorf("写入任务文件失败: %w", err)    }    return nil}// loadTasks 从JSON文件加载任务列表func loadTasks() ([]Task, error) {    if _, err := os.Stat(dataFile); os.IsNotExist(err) {        // 文件不存在，返回空列表而不是错误        return []Task{}, nil    }    data, err := ioutil.ReadFile(dataFile)    if err != nil {        return nil, fmt.Errorf("读取任务文件失败: %w", err)    }    var tasks []Task    err = json.Unmarshal(data, &tasks)    if err != nil {        return nil, fmt.Errorf("反序列化任务失败: %w", err)    }    return tasks, nil}

json.MarshalIndent

json.Marshal

ioutil.WriteFile

ioutil.ReadFile

loadTasks

time.AfterFunc

time.AfterFunc

time.AfterFunc

package mainimport (    "sync"    "time")// scheduler 用于管理所有任务的调度type Scheduler struct {    tasks      []Task    mu         sync.Mutex // 保护tasks切片的并发访问    stopChans  map[string]chan struct{} // 存储每个任务的停止信号通道}func NewScheduler(initialTasks []Task) *Scheduler {    s := &Scheduler{        tasks:      initialTasks,        stopChans:  make(map[string]chan struct{}),    }    return s}// ScheduleTask 为单个任务设置提醒func (s *Scheduler) ScheduleTask(task Task) {    if task.IsCompleted {        return // 已完成的任务无需调度    }    now := time.Now()    if task.DueTime.Before(now) {        // 提醒时间已过，可以直接触发提醒或标记为过期        fmt.Printf("【过期提醒】任务 '%s' (ID: %s) 提醒时间已过: %sn", task.Title, task.ID, task.DueTime.Format("15:04:05"))        return    }    duration := task.DueTime.Sub(now)    if duration <= 0 { // 避免负数或零导致time.AfterFunc立即执行        duration = time.Millisecond // 至少等待一毫秒    }    stopChan := make(chan struct{})    s.mu.Lock()    s.stopChans[task.ID] = stopChan // 记录停止通道，以便取消    s.mu.Unlock()    fmt.Printf("【调度】任务 '%s' (ID: %s) 将在 %s 后提醒。n", task.Title, task.ID, duration.String())    go func(t Task, sc chan struct{}) {        select {        case <-time.After(duration):            fmt.Printf("n? 【任务提醒】时间到了！n%sn", t.String())            // 提醒后，可以考虑自动标记为已完成或需要用户手动确认            // 这里为了简化，我们只打印提醒        case <-sc:            fmt.Printf("【取消调度】任务 '%s' (ID: %s) 的提醒已被取消。n", t.Title, t.ID)        }        s.mu.Lock()        delete(s.stopChans, t.ID) // 任务完成后从调度器中移除        s.mu.Unlock()    }(task, stopChan)}// CancelTask 取消一个任务的提醒func (s *Scheduler) CancelTask(taskID string) {    s.mu.Lock()    defer s.mu.Unlock()    if ch, ok := s.stopChans[taskID]; ok {        close(ch) // 发送停止信号        delete(s.stopChans, taskID)    }}// StartAllTasksScheduler 启动所有任务的调度func (s *Scheduler) StartAllTasksScheduler() {    s.mu.Lock()    defer s.mu.Unlock()    for _, task := range s.tasks {        s.ScheduleTask(task)    }}// AddTaskToScheduler 添加新任务并调度func (s *Scheduler) AddTaskToScheduler(task Task) {    s.mu.Lock()    s.tasks = append(s.tasks, task)    s.mu.Unlock()    s.ScheduleTask(task)}

Scheduler

sync.Mutex

stopChans

select

time.After

stopChan

time.AfterFunc

flag

spf13/cobra

go run main.go add -t "买菜" -d "晚上做饭用" -time "2023-10-27 18:00:00"

go run main.go list

go run main.go complete -id 

go run main.go delete -id 

main

main

package mainimport (    "bufio"    "fmt"    "os"    "strings"    "time"    "github.com/google/uuid" // 引入uuid库生成唯一ID)var scheduler *Schedulervar tasks []Task // 全局的任务列表func main() {    // 加载现有任务    loadedTasks, err := loadTasks()    if err != nil {        fmt.Printf("加载任务失败: %vn", err)        os.Exit(1)    }    tasks = loadedTasks    scheduler = NewScheduler(tasks)    scheduler.StartAllTasksScheduler() // 启动所有任务的调度    fmt.Println("任务提醒工具已启动。输入 'help' 查看命令。")    reader := bufio.NewReader(os.Stdin)    for {        fmt.Print("> ")        input, _ := reader.ReadString('n')        input = strings.TrimSpace(input)        parts := strings.Fields(input)        if len(parts) == 0 {            continue        }        command := parts[0]        args := parts[1:]        switch command {        case "add":            handleAddCommand(args)        case "list":            handleListCommand()        case "complete":            handleCompleteCommand(args)        case "delete":            handleDeleteCommand(args)        case "help":            printHelp()        case "exit":            fmt.Println("退出任务提醒工具。")            // 在退出前保存所有任务            if err := saveTasks(tasks); err != nil {                fmt.Printf("保存任务失败: %vn", err)            }            return        default:            fmt.Println("未知命令。输入 'help' 查看命令。")        }    }}func printHelp() {    fmt.Println(`可用命令:  add