Golang Command命令模式任务调度示例
在软件设计中，命令模式是一种行为型设计模式，它将请求封装为独立的对象，从而让使用者可以根据不同的请求将客户端参数化，同时支持请求的排队、记录日志以及撤销操作。在任务调度场景中，我们通常会有多种不同类型的任务，调度器不需要关心每个任务的具体实现细节，只需要调用统一的执行方法即可，这种模式非常适合用命令模式来实现。
命令模式的核心角色
在命令模式的实现中，通常包含以下几个核心角色：
- 命令接口（Command）：定义所有命令的统一执行方法，是任务调度的通用入口。
- 具体命令（Concrete Command）：实现命令接口，绑定具体的任务接收者和执行逻辑。
- 调用者（Invoker）：也就是任务调度器，负责存储和执行命令，不需要了解命令的具体实现。
- 接收者（Receiver）：实际执行任务逻辑的对象，包含具体的业务处理方法。
Go语言实现示例
下面我们用Go语言实现一个简单的任务调度系统，包含三种不同的任务：打印日志任务、发送邮件任务、数据备份任务，调度器可以统一调度这些任务执行。
1. 定义命令接口
首先定义所有命令需要实现的统一接口，包含一个Execute方法用于执行命令：
// Command 命令接口，所有具体命令都需要实现该接口
type Command interface {
Execute() error
}2. 定义接收者
接收者是实际执行任务逻辑的对象，我们定义三个不同的接收者对应三种任务：
// LogReceiver 日志打印任务接收者
type LogReceiver struct {
Content string
}
// PrintLog 实际执行日志打印的方法
func (r *LogReceiver) PrintLog() error {
fmt.Printf("打印日志: %s\n", r.Content)
return nil
}
// MailReceiver 邮件发送任务接收者
type MailReceiver struct {
To string
Subject string
Body string
}
// SendMail 实际执行邮件发送的方法
func (r *MailReceiver) SendMail() error {
fmt.Printf("发送邮件到: %s, 主题: %s, 内容: %s\n", r.To, r.Subject, r.Body)
return nil
}
// BackupReceiver 数据备份任务接收者
type BackupReceiver struct {
SourcePath string
TargetPath string
}
// BackupData 实际执行数据备份的方法
func (r *BackupReceiver) BackupData() error {
fmt.Printf("备份数据: 从 %s 到 %s\n", r.SourcePath, r.TargetPath)
return nil
}3. 定义具体命令
具体命令实现Command接口，将接收者和对应的执行方法绑定，每个命令对应一种任务类型：
// LogCommand 日志打印命令
type LogCommand struct {
receiver *LogReceiver
}
// Execute 执行日志打印命令
func (c *LogCommand) Execute() error {
return c.receiver.PrintLog()
}
// MailCommand 邮件发送命令
type MailCommand struct {
receiver *MailReceiver
}
// Execute 执行邮件发送命令
func (c *MailCommand) Execute() error {
return c.receiver.SendMail()
}
// BackupCommand 数据备份命令
type BackupCommand struct {
receiver *BackupReceiver
}
// Execute 执行数据备份命令
func (c *BackupCommand) Execute() error {
return c.receiver.BackupData()
}4. 定义调度器（调用者）
调度器负责存储待执行的命令，并统一调用命令的Execute方法，不需要关心每个命令的具体实现：
// Scheduler 任务调度器
type Scheduler struct {
commands []Command
}
// AddCommand 向调度器添加待执行的命令
func (s *Scheduler) AddCommand(cmd Command) {
s.commands = append(s.commands, cmd)
}
// Run 执行所有添加到调度器的命令
func (s *Scheduler) Run() {
for _, cmd := range s.commands {
if err := cmd.Execute(); err != nil {
fmt.Printf("任务执行失败: %v\n", err)
}
}
// 执行完成后清空命令队列
s.commands = nil
}5. 完整调用示例
下面的代码演示了如何创建不同的任务，添加到调度器后统一执行：
package main
import "fmt"
// Command 命令接口
type Command interface {
Execute() error
}
// LogReceiver 日志打印任务接收者
type LogReceiver struct {
Content string
}
func (r *LogReceiver) PrintLog() error {
fmt.Printf("打印日志: %s\n", r.Content)
return nil
}
// MailReceiver 邮件发送任务接收者
type MailReceiver struct {
To string
Subject string
Body string
}
func (r *MailReceiver) SendMail() error {
fmt.Printf("发送邮件到: %s, 主题: %s, 内容: %s\n", r.To, r.Subject, r.Body)
return nil
}
// BackupReceiver 数据备份任务接收者
type BackupReceiver struct {
SourcePath string
TargetPath string
}
func (r *BackupReceiver) BackupData() error {
fmt.Printf("备份数据: 从 %s 到 %s\n", r.SourcePath, r.TargetPath)
return nil
}
// LogCommand 日志打印命令
type LogCommand struct {
receiver *LogReceiver
}
func (c *LogCommand) Execute() error {
return c.receiver.PrintLog()
}
// MailCommand 邮件发送命令
type MailCommand struct {
receiver *MailReceiver
}
func (c *MailCommand) Execute() error {
return c.receiver.SendMail()
}
// BackupCommand 数据备份命令
type BackupCommand struct {
receiver *BackupReceiver
}
func (c *BackupCommand) Execute() error {
return c.receiver.BackupData()
}
// Scheduler 任务调度器
type Scheduler struct {
commands []Command
}
func (s *Scheduler) AddCommand(cmd Command) {
s.commands = append(s.commands, cmd)
}
func (s *Scheduler) Run() {
for _, cmd := range s.commands {
if err := cmd.Execute(); err != nil {
fmt.Printf("任务执行失败: %v\n", err)
}
}
s.commands = nil
}
func main() {
// 创建调度器
scheduler := &Scheduler{}
// 创建日志打印任务并添加到调度器
logReceiver := &LogReceiver{Content: "系统启动完成，当前时间: 2024-05-20 10:00:00"}
logCmd := &LogCommand{receiver: logReceiver}
scheduler.AddCommand(logCmd)
// 创建邮件发送任务并添加到调度器
mailReceiver := &MailReceiver{
To: "admin@ipipp.com",
Subject: "系统启动通知",
Body: "系统已正常启动，所有服务运行正常",
}
mailCmd := &MailCommand{receiver: mailReceiver}
scheduler.AddCommand(mailCmd)
// 创建数据备份任务并添加到调度器
backupReceiver := &BackupReceiver{
SourcePath: "/data/app",
TargetPath: "/backup/app_20240520",
}
backupCmd := &BackupCommand{receiver: backupReceiver}
scheduler.AddCommand(backupCmd)
// 执行所有任务
fmt.Println("开始执行任务调度...")
scheduler.Run()
fmt.Println("所有任务执行完成")
}模式优势与扩展
使用命令模式实现任务调度有以下几个明显优势：
- 解耦调度器和任务实现：调度器不需要依赖具体的任务类型，只需要面向Command接口编程，新增任务时只需要新增具体的命令和接收者，不需要修改调度器代码，符合开闭原则。
- 支持任务扩展：后续如果需要新增任务类型，比如定时清理缓存任务，只需要新增对应的接收者和命令实现Command接口即可，原有代码不需要改动。
- 便于功能扩展：可以在命令执行前后添加统一的逻辑，比如记录任务执行时间、统计任务执行次数，只需要修改命令的Execute方法或者新增装饰器即可。
如果需要支持任务的撤销操作，只需要在Command接口中新增Undo方法，在具体命令中实现对应的撤销逻辑，调度器调用Undo方法即可完成撤销，这也是命令模式的典型扩展场景。