mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-27 17:50:53 +01:00
refactor: code cleanup
This commit is contained in:
@@ -1,6 +0,0 @@
|
||||
package event
|
||||
|
||||
func SendEvent(eventName string, data ...interface{}) {
|
||||
svc := NewEventService()
|
||||
svc.SendEvent(eventName, data...)
|
||||
}
|
||||
@@ -1,92 +0,0 @@
|
||||
package event
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/apex/log"
|
||||
"github.com/crawlab-team/crawlab/core/entity"
|
||||
"github.com/crawlab-team/crawlab/core/interfaces"
|
||||
"github.com/crawlab-team/crawlab/core/utils"
|
||||
"github.com/crawlab-team/crawlab/trace"
|
||||
"github.com/thoas/go-funk"
|
||||
"regexp"
|
||||
)
|
||||
|
||||
var S interfaces.EventService
|
||||
|
||||
type Service struct {
|
||||
keys []string
|
||||
includes []string
|
||||
excludes []string
|
||||
chs []*chan interfaces.EventData
|
||||
}
|
||||
|
||||
func (svc *Service) Register(key, include, exclude string, ch *chan interfaces.EventData) {
|
||||
svc.keys = append(svc.keys, key)
|
||||
svc.includes = append(svc.includes, include)
|
||||
svc.excludes = append(svc.excludes, exclude)
|
||||
svc.chs = append(svc.chs, ch)
|
||||
}
|
||||
|
||||
func (svc *Service) Unregister(key string) {
|
||||
idx := funk.IndexOfString(svc.keys, key)
|
||||
if idx != -1 {
|
||||
svc.keys = append(svc.keys[:idx], svc.keys[(idx+1):]...)
|
||||
svc.includes = append(svc.includes[:idx], svc.includes[(idx+1):]...)
|
||||
svc.excludes = append(svc.excludes[:idx], svc.excludes[(idx+1):]...)
|
||||
svc.chs = append(svc.chs[:idx], svc.chs[(idx+1):]...)
|
||||
log.Infof("[EventService] unregistered %s", key)
|
||||
}
|
||||
}
|
||||
|
||||
func (svc *Service) SendEvent(eventName string, data ...interface{}) {
|
||||
for i, key := range svc.keys {
|
||||
// include
|
||||
include := svc.includes[i]
|
||||
matchedInclude, err := regexp.MatchString(include, eventName)
|
||||
if err != nil {
|
||||
trace.PrintError(err)
|
||||
continue
|
||||
}
|
||||
if !matchedInclude {
|
||||
continue
|
||||
}
|
||||
|
||||
// exclude
|
||||
exclude := svc.excludes[i]
|
||||
matchedExclude, err := regexp.MatchString(exclude, eventName)
|
||||
if err != nil {
|
||||
trace.PrintError(err)
|
||||
continue
|
||||
}
|
||||
if matchedExclude {
|
||||
continue
|
||||
}
|
||||
|
||||
// send event
|
||||
utils.LogDebug(fmt.Sprintf("key %s matches event %s", key, eventName))
|
||||
ch := svc.chs[i]
|
||||
go func(ch *chan interfaces.EventData) {
|
||||
for _, d := range data {
|
||||
*ch <- &entity.EventData{
|
||||
Event: eventName,
|
||||
Data: d,
|
||||
}
|
||||
}
|
||||
}(ch)
|
||||
}
|
||||
}
|
||||
|
||||
func NewEventService() (svc interfaces.EventService) {
|
||||
if S != nil {
|
||||
return S
|
||||
}
|
||||
|
||||
svc = &Service{
|
||||
chs: []*chan interfaces.EventData{},
|
||||
keys: []string{},
|
||||
}
|
||||
|
||||
S = svc
|
||||
|
||||
return svc
|
||||
}
|
||||
Reference in New Issue
Block a user