package aria2 import ( "github.com/HFO4/cloudreve/pkg/aria2/rpc" "sync" ) // Notifier aria2实践通知处理 type Notifier struct { Subscribes sync.Map } // Subscribe 订阅事件通知 func (notifier *Notifier) Subscribe(target chan StatusEvent, gid string) { notifier.Subscribes.Store(gid, target) } // Unsubscribe 取消订阅事件通知 func (notifier *Notifier) Unsubscribe(gid string) { notifier.Subscribes.Delete(gid) } // Notify 发送通知 func (notifier *Notifier) Notify(events []rpc.Event, status int) { for _, event := range events { if target, ok := notifier.Subscribes.Load(event.Gid); ok { target.(chan StatusEvent) <- StatusEvent{ GID: event.Gid, Status: status, } } } } // OnDownloadStart 下载开始 func (notifier *Notifier) OnDownloadStart(events []rpc.Event) { notifier.Notify(events, Downloading) } // OnDownloadPause 下载暂停 func (notifier *Notifier) OnDownloadPause(events []rpc.Event) { notifier.Notify(events, Paused) } // OnDownloadStop 下载停止 func (notifier *Notifier) OnDownloadStop(events []rpc.Event) { notifier.Notify(events, Canceled) } // OnDownloadComplete 下载完成 func (notifier *Notifier) OnDownloadComplete(events []rpc.Event) { notifier.Notify(events, Complete) } // OnDownloadError 下载出错 func (notifier *Notifier) OnDownloadError(events []rpc.Event) { notifier.Notify(events, Error) } // OnBtDownloadComplete BT下载完成 func (notifier *Notifier) OnBtDownloadComplete(events []rpc.Event) { notifier.Notify(events, Complete) }