Golang 实现消息定时分发的 gosd 库使用指南

gosd 是一个用于 Golang 的消息调度和定时分发到通道的插件库，它可以帮助你实现消息的延迟发送和定时调度功能。

安装 gosd

go get github.com/yourusername/gosd

基本使用示例

1. 创建调度器实例

package main

import (
	"fmt"
	"time"
	
	"github.com/yourusername/gosd"
)

func main() {
	// 创建调度器实例
	scheduler := gosd.NewScheduler()
	
	// 确保在程序退出前关闭调度器
	defer scheduler.Stop()
	
	// 创建一个消息通道
	msgChan := make(chan string, 10)
	
	// 启动调度器
	go scheduler.Run()
	
	// 添加定时任务
	scheduler.Schedule(time.Now().Add(2*time.Second), "Hello after 2 seconds", msgChan)
	scheduler.Schedule(time.Now().Add(5*time.Second), "Hello after 5 seconds", msgChan)
	
	// 接收消息
	for i := 0; i < 2; i++ {
		msg := <-msgChan
		fmt.Println("Received:", msg)
	}
}

2. 周期性任务

func periodicExample() {
	scheduler := gosd.NewScheduler()
	defer scheduler.Stop()
	
	msgChan := make(chan string, 10)
	
	go scheduler.Run()
	
	// 每隔1秒发送一次消息，共发送5次
	scheduler.SchedulePeriodic(1*time.Second, 5, "Periodic message", msgChan)
	
	for i := 0; i < 5; i++ {
		msg := <-msgChan
		fmt.Println("Periodic:", msg, "at", time.Now().Format("15:04:05"))
	}
}

高级功能

1. 自定义消息类型

type CustomMessage struct {
	ID      int
	Content string
}

func customTypeExample() {
	scheduler := gosd.NewScheduler()
	defer scheduler.Stop()
	
	customChan := make(chan CustomMessage, 10)
	
	go scheduler.Run()
	
	// 发送自定义类型消息
	scheduler.Schedule(time.Now().Add(3*time.Second), 
		CustomMessage{ID: 1, Content: "Custom content"}, 
		customChan)
	
	msg := <-customChan
	fmt.Printf("Received custom message: %+v\n", msg)
}

2. 取消已调度的任务

func cancelExample() {
	scheduler := gosd.NewScheduler()
	defer scheduler.Stop()
	
	msgChan := make(chan string, 10)
	
	go scheduler.Run()
	
	// 添加任务并获取任务ID
	taskID := scheduler.Schedule(time.Now().Add(10*time.Second), "This will be cancelled", msgChan)
	
	// 取消任务
	scheduler.Cancel(taskID)
	
	// 验证是否取消
	select {
	case msg := <-msgChan:
		fmt.Println("Unexpected message:", msg)
	case <-time.After(11 * time.Second):
		fmt.Println("Message was successfully cancelled")
	}
}

最佳实践

1. 资源管理：确保在不再需要调度器时调用 Stop() 方法释放资源
2. 通道缓冲：为消息通道设置适当的缓冲大小以防止阻塞
3. 错误处理：考虑添加错误处理通道来接收调度过程中的错误
4. 性能考量：对于大量任务，考虑使用优先级队列优化

扩展功能

如果需要更复杂的功能，可以扩展 gosd：

// 扩展调度器添加优先级功能
type PriorityScheduler struct {
	*gosd.Scheduler
	priorityQueue *PriorityQueue
}

func NewPriorityScheduler() *PriorityScheduler {
	return &PriorityScheduler{
		Scheduler:     gosd.NewScheduler(),
		priorityQueue: NewPriorityQueue(),
	}
}

// 重写Schedule方法支持优先级
func (ps *PriorityScheduler) ScheduleWithPriority(when time.Time, priority int, msg interface{}, ch interface{}) int {
	// 实现优先级调度逻辑
	// ...
}

gosd 提供了一个简单而强大的基础，你可以根据实际需求进行扩展和定制。