Golang轻量级内存任务队列Artifex使用指南

Artifex是一个基于内存的轻量级Golang任务队列库，它提供了简单易用的API来管理后台任务。下面我将详细介绍如何使用Artifex。

安装Artifex

首先安装Artifex库：

go get github.com/mborders/artifex

基本使用

1. 创建调度器

package main

import (
	"fmt"
	"time"
	
	"github.com/mborders/artifex"
)

func main() {
	// 创建调度器，设置工作线程数为2
	dispatcher, err := artifex.NewDispatcher(2, 2)
	if err != nil {
		panic(err)
	}
	defer dispatcher.Shutdown()

	// 启动调度器
	err = dispatcher.Start()
	if err != nil {
		panic(err)
	}
}

2. 添加即时任务

// 添加即时任务
err = dispatcher.Dispatch(func() {
    fmt.Println("This task runs immediately")
})
if err != nil {
    panic(err)
}

3. 添加延迟任务

// 添加延迟任务（3秒后执行）
err = dispatcher.DispatchAfter(func() {
    fmt.Println("This task runs after 3 seconds")
}, 3*time.Second)
if err != nil {
    panic(err)
}

4. 添加定时任务

// 添加定时任务（每5秒执行一次）
err = dispatcher.DispatchEvery(func() {
    fmt.Println("This task runs every 5 seconds")
}, 5*time.Second)
if err != nil {
    panic(err)
}

完整示例

package main

import (
	"fmt"
	"time"
	
	"github.com/mborders/artifex"
)

func main() {
	// 创建调度器
	dispatcher, err := artifex.NewDispatcher(2, 2)
	if err != nil {
		panic(err)
	}
	defer dispatcher.Shutdown()

	// 启动调度器
	err = dispatcher.Start()
	if err != nil {
		panic(err)
	}

	// 添加即时任务
	err = dispatcher.Dispatch(func() {
		fmt.Println("即时任务执行:", time.Now().Format("15:04:05"))
	})
	if err != nil {
		panic(err)
	}

	// 添加延迟任务
	err = dispatcher.DispatchAfter(func() {
		fmt.Println("延迟3秒任务执行:", time.Now().Format("15:04:05"))
	}, 3*time.Second)
	if err != nil {
		panic(err)
	}

	// 添加定时任务
	err = dispatcher.DispatchEvery(func() {
		fmt.Println("每5秒执行的任务:", time.Now().Format("15:04:05"))
	}, 5*time.Second)
	if err != nil {
		panic(err)
	}

	// 主程序等待20秒以便观察定时任务
	time.Sleep(20 * time.Second)
}

高级特性

1. 带参数的任务

// 带参数的任务
err = dispatcher.Dispatch(func(name string) {
    fmt.Println("Hello,", name)
}, "World")
if err != nil {
    panic(err)
}

2. 错误处理

// 错误处理
err = dispatcher.Dispatch(func() error {
    _, err := fmt.Println("This task might fail")
    return err
})
if err != nil {
    panic(err)
}

3. 动态调整工作池大小

// 调整工作池大小
err = dispatcher.SetPoolSize(4)
if err != nil {
    panic(err)
}

注意事项

1. Artifex是基于内存的队列，任务不会持久化，程序重启后任务会丢失
2. 适合轻量级、非关键性任务处理
3. 对于高可靠性要求的场景，建议使用Redis或RabbitMQ等持久化队列
4. 注意资源管理，避免内存泄漏

性能优化建议

1. 根据任务负载合理设置工作线程数
2. 避免在任务中执行长时间阻塞的操作
3. 对于CPU密集型任务，考虑使用更大的工作池
4. 对于IO密集型任务，可以适当增加工作池大小

Artifex是一个简单易用的内存任务队列，适合需要轻量级任务调度的场景。对于更复杂的需求，可以考虑其他更强大的任务队列系统。