下面是一個簡單的示例,展示如何使用MySQL和Go語言開發一個任務調度系統:
安裝MySQL數據庫和Go語言開發環境。
創建一個MySQL數據庫,用于存儲任務和調度信息。可以使用以下命令創建一個名為task_scheduler
的數據庫:
CREATE DATABASE task_scheduler;
tasks
的表,用于存儲任務信息。可以使用以下命令創建該表:CREATE TABLE tasks (
id INT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(255) NOT NULL,
description TEXT,
schedule_time DATETIME NOT NULL,
status ENUM('pending', 'running', 'completed') DEFAULT 'pending',
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
);
創建一個Go語言的項目,并導入MySQL數據庫驅動程序。
創建一個Task
結構體,用于存儲任務信息。可以使用以下代碼創建該結構體:
type Task struct {
ID int
Name string
Description string
ScheduleTime time.Time
Status string
CreatedAt time.Time
UpdatedAt time.Time
}
TaskScheduler
結構體,用于執行任務調度。可以使用以下代碼創建該結構體:type TaskScheduler struct {
DB *sql.DB
}
AddTask
方法,用于添加新的任務到數據庫。可以使用以下代碼實現該方法:func (ts *TaskScheduler) AddTask(task *Task) error {
stmt, err := ts.DB.Prepare("INSERT INTO tasks(name, description, schedule_time) VALUES(?, ?, ?)")
if err != nil {
return err
}
defer stmt.Close()
_, err = stmt.Exec(task.Name, task.Description, task.ScheduleTime)
if err != nil {
return err
}
return nil
}
GetPendingTasks
方法,用于獲取待執行的任務列表。可以使用以下代碼實現該方法:func (ts *TaskScheduler) GetPendingTasks() ([]*Task, error) {
tasks := []*Task{}
rows, err := ts.DB.Query("SELECT id, name, description, schedule_time, status, created_at, updated_at FROM tasks WHERE status = 'pending'")
if err != nil {
return nil, err
}
defer rows.Close()
for rows.Next() {
task := &Task{}
err = rows.Scan(&task.ID, &task.Name, &task.Description, &task.ScheduleTime, &task.Status, &task.CreatedAt, &task.UpdatedAt)
if err != nil {
return nil, err
}
tasks = append(tasks, task)
}
return tasks, nil
}
UpdateTaskStatus
方法,用于更新任務的狀態。可以使用以下代碼實現該方法:func (ts *TaskScheduler) UpdateTaskStatus(task *Task, status string) error {
stmt, err := ts.DB.Prepare("UPDATE tasks SET status = ? WHERE id = ?")
if err != nil {
return err
}
defer stmt.Close()
_, err = stmt.Exec(status, task.ID)
if err != nil {
return err
}
return nil
}
main
函數,用于初始化數據庫連接和調度任務的循環。可以使用以下代碼創建該函數:func main() {
db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/task_scheduler")
if err != nil {
log.Fatal(err)
}
defer db.Close()
scheduler := &TaskScheduler{DB: db}
for {
tasks, err := scheduler.GetPendingTasks()
if err != nil {
log.Println(err)
continue
}
for _, task := range tasks {
// 執行任務
log.Println("Running task:", task.Name)
// 更新任務狀態為運行中
err = scheduler.UpdateTaskStatus(task, "running")
if err != nil {
log.Println(err)
continue
}
// 模擬任務執行
time.Sleep(5 * time.Second)
// 更新任務狀態為已完成