mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 09:11:55 +00:00
134 lines
3.5 KiB
Go
134 lines
3.5 KiB
Go
package bbgo
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/c9s/bbgo/pkg/util"
|
|
)
|
|
|
|
var Notification = &Notifiability{
|
|
SymbolChannelRouter: NewPatternChannelRouter(nil),
|
|
SessionChannelRouter: NewPatternChannelRouter(nil),
|
|
ObjectChannelRouter: NewObjectChannelRouter(),
|
|
}
|
|
|
|
func RegisterCommand(application, command string, handler func(string)) {
|
|
Notification.RegisterCommand(application, command, handler)
|
|
}
|
|
|
|
func Notify(obj interface{}, args ...interface{}) {
|
|
Notification.Notify(obj, args...)
|
|
}
|
|
|
|
func NotifyTo(channel string, obj interface{}, args ...interface{}) {
|
|
Notification.NotifyTo(channel, obj, args...)
|
|
}
|
|
|
|
func SendPhoto(buffer *bytes.Buffer) {
|
|
Notification.SendPhoto(buffer)
|
|
}
|
|
|
|
func SendPhotoTo(channel string, buffer *bytes.Buffer) {
|
|
Notification.SendPhotoTo(channel, buffer)
|
|
}
|
|
|
|
type Notifier interface {
|
|
NotifyTo(channel string, obj interface{}, args ...interface{})
|
|
Notify(obj interface{}, args ...interface{})
|
|
SendPhotoTo(channel string, buffer *bytes.Buffer)
|
|
SendPhoto(buffer *bytes.Buffer)
|
|
RegisterCommand(command string, handler func(string))
|
|
ID() string
|
|
}
|
|
|
|
type NullNotifier struct{}
|
|
|
|
func (n *NullNotifier) NotifyTo(channel string, obj interface{}, args ...interface{}) {}
|
|
|
|
func (n *NullNotifier) Notify(obj interface{}, args ...interface{}) {}
|
|
|
|
func (n *NullNotifier) SendPhoto(buffer *bytes.Buffer) {}
|
|
|
|
func (n *NullNotifier) SendPhotoTo(channel string, buffer *bytes.Buffer) {}
|
|
|
|
func (n *NullNotifier) RegisterCommand(command string, handler func(string)) {}
|
|
|
|
func (n *NullNotifier) ID() string {
|
|
return "null"
|
|
}
|
|
|
|
type Notifiability struct {
|
|
notifiers []Notifier
|
|
SessionChannelRouter *PatternChannelRouter `json:"-"`
|
|
SymbolChannelRouter *PatternChannelRouter `json:"-"`
|
|
ObjectChannelRouter *ObjectChannelRouter `json:"-"`
|
|
}
|
|
|
|
// RouteSymbol routes symbol name to channel
|
|
func (m *Notifiability) RouteSymbol(symbol string) (channel string, ok bool) {
|
|
if m.SymbolChannelRouter != nil {
|
|
return m.SymbolChannelRouter.Route(symbol)
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
// RouteSession routes Session name to channel
|
|
func (m *Notifiability) RouteSession(session string) (channel string, ok bool) {
|
|
if m.SessionChannelRouter != nil {
|
|
return m.SessionChannelRouter.Route(session)
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
// RouteObject routes object to channel
|
|
func (m *Notifiability) RouteObject(obj interface{}) (channel string, ok bool) {
|
|
if m.ObjectChannelRouter != nil {
|
|
return m.ObjectChannelRouter.Route(obj)
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
// AddNotifier adds the notifier that implements the Notifier interface.
|
|
func (m *Notifiability) AddNotifier(notifier Notifier) {
|
|
m.notifiers = append(m.notifiers, notifier)
|
|
}
|
|
|
|
func (m *Notifiability) Notify(obj interface{}, args ...interface{}) {
|
|
if str, ok := obj.(string); ok {
|
|
simpleArgs := util.FilterSimpleArgs(args)
|
|
logrus.Infof(str, simpleArgs...)
|
|
}
|
|
|
|
for _, n := range m.notifiers {
|
|
n.Notify(obj, args...)
|
|
}
|
|
}
|
|
|
|
func (m *Notifiability) NotifyTo(channel string, obj interface{}, args ...interface{}) {
|
|
for _, n := range m.notifiers {
|
|
n.NotifyTo(channel, obj, args...)
|
|
}
|
|
}
|
|
|
|
func (m *Notifiability) SendPhoto(buffer *bytes.Buffer) {
|
|
for _, n := range m.notifiers {
|
|
n.SendPhoto(buffer)
|
|
}
|
|
}
|
|
|
|
func (m *Notifiability) SendPhotoTo(channel string, buffer *bytes.Buffer) {
|
|
for _, n := range m.notifiers {
|
|
n.SendPhotoTo(channel, buffer)
|
|
}
|
|
}
|
|
|
|
func (m *Notifiability) RegisterCommand(application, command string, handler func(string)) {
|
|
for _, n := range m.notifiers {
|
|
if application == n.ID() {
|
|
n.RegisterCommand(command, handler)
|
|
}
|
|
}
|
|
}
|