Kala - Golang高性能轻量级任务调度库

Kala是一个用Go编写的轻量级任务调度库，它提供了简单易用的API来执行定时任务。下面我将详细介绍Kala的使用方法和示例代码。

安装Kala

go get github.com/ajvb/kala

基本使用

1. 创建调度器

package main

import (
	"fmt"
	"time"

	"github.com/ajvb/kala/job"
	"github.com/ajvb/kala/utils/iso8601"
)

func main() {
	// 创建调度器
	scheduler := job.NewScheduler()
	
	// 启动调度器
	scheduler.Start()
	defer scheduler.Stop()
	
	// 创建任务
	job := &job.Job{
		Name:     "test_job",
		Command:  "echo 'Hello, Kala!'",
		Schedule: iso8601.MustParse("R/2015-06-04T19:25:16.828696-07:00/PT10S"),
	}
	
	// 添加任务到调度器
	scheduler.AddJob(job)
	
	// 等待一段时间查看任务执行
	time.Sleep(30 * time.Second)
}

2. 使用自定义处理函数

package main

import (
	"fmt"
	"time"

	"github.com/ajvb/kala/job"
	"github.com/ajvb/kala/utils/iso8601"
)

func myTask() {
	fmt.Println("Custom task executed at:", time.Now().Format(time.RFC3339))
}

func main() {
	scheduler := job.NewScheduler()
	scheduler.Start()
	defer scheduler.Stop()

	// 创建自定义任务
	customJob := &job.Job{
		Name: "custom_function_job",
		Schedule: iso8601.MustParse("PT5S"), // 每5秒执行一次
		JobFunction: func() {
			myTask()
		},
	}

	scheduler.AddJob(customJob)

	// 等待一段时间查看任务执行
	time.Sleep(20 * time.Second)
}

高级功能

1. 一次性任务

oneTimeJob := &job.Job{
	Name:    "one_time_job",
	Command: "echo 'This will run once'",
	Schedule: &iso8601.OneTimeSchedule{
		RunAt: time.Now().Add(10 * time.Second),
	},
}
scheduler.AddJob(oneTimeJob)

2. 带重试机制的任务

retryJob := &job.Job{
	Name:     "retry_job",
	Command:  "some_command_that_might_fail",
	Schedule: iso8601.MustParse("PT1M"), // 每分钟执行一次
	Retries:  3,                         // 重试3次
}
scheduler.AddJob(retryJob)

3. 获取任务状态

// 获取所有任务
jobs := scheduler.GetAllJobs()
for _, j := range jobs {
	fmt.Printf("Job: %s, NextRun: %v\n", j.Name, j.NextRunAt)
}

// 获取单个任务
job, exists := scheduler.GetJob("test_job")
if exists {
	fmt.Println("Job found:", job.Name)
}

4. 删除任务

scheduler.DeleteJob("job_to_delete")

性能优化技巧

1. 批量添加任务：当需要添加大量任务时，可以使用AddJobs方法批量添加

jobs := []*job.Job{
	{Name: "job1", Command: "cmd1", Schedule: iso8601.MustParse("PT10S")},
	{Name: "job2", Command: "cmd2", Schedule: iso8601.MustParse("PT20S")},
}
scheduler.AddJobs(jobs)

1. 使用内存存储：Kala默认使用内存存储，对于不需要持久化的场景，这是最高效的选择

2. 合理设置调度间隔：避免设置过于频繁的调度间隔，减少系统开销

实际应用示例

下面是一个完整的Web服务示例，通过HTTP API管理任务：

package main

import (
	"encoding/json"
	"fmt"
	"net/http"
	"time"

	"github.com/ajvb/kala/job"
	"github.com/ajvb/kala/utils/iso8601"
)

var scheduler = job.NewScheduler()

func main() {
	scheduler.Start()
	defer scheduler.Stop()

	http.HandleFunc("/jobs", listJobs)
	http.HandleFunc("/jobs/add", addJob)
	http.HandleFunc("/jobs/delete", deleteJob)

	fmt.Println("Server started at :8080")
	http.ListenAndServe(":8080", nil)
}

func listJobs(w http.ResponseWriter, r *http.Request) {
	jobs := scheduler.GetAllJobs()
	json.NewEncoder(w).Encode(jobs)
}

func addJob(w http.ResponseWriter, r *http.Request) {
	var newJob struct {
		Name     string `json:"name"`
		Command  string `json:"command"`
		Schedule string `json:"schedule"`
	}
	
	if err := json.NewDecoder(r.Body).Decode(&newJob); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	job := &job.Job{
		Name:    newJob.Name,
		Command: newJob.Command,
		Schedule: iso8601.MustParse(newJob.Schedule),
	}

	scheduler.AddJob(job)
	w.WriteHeader(http.StatusCreated)
}

func deleteJob(w http.ResponseWriter, r *http.Request) {
	name := r.URL.Query().Get("name")
	if name == "" {
		http.Error(w, "name parameter is required", http.StatusBadRequest)
		return
	}

	scheduler.DeleteJob(name)
	w.WriteHeader(http.StatusOK)
}

总结

Kala是一个简单而强大的Go任务调度库，具有以下特点：

• 轻量级，无外部依赖
• 支持ISO8601时间格式
• 支持一次性任务和周期性任务
• 提供任务重试机制
• 易于集成到现有应用中

通过合理使用Kala，你可以轻松地为你的Go应用添加可靠的任务调度功能。