chrissewell/beelzebub
1
0
Fork 0
mirror of https://github.com/mariocandela/beelzebub.git synced 2025-07-01 18:47:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Refactoring protocolManager.go and tracer.go, improce dependency injection

Browse Source
This commit is contained in:
Mario
2022-05-11 22:58:03 +02:00
parent 3683659d35
commit 3c5ac84ef0
4 changed files with 24 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
main.go
View File

@ -3,6 +3,7 @@ package main
import (
"beelzebub/parser"
"beelzebub/protocols"
"beelzebub/tracer"
log "github.com/sirupsen/logrus"
"io"
"os"
@ -31,8 +32,7 @@ func main() {
hypertextTransferProtocolStrategy := &protocols.HypertextTransferProtocolStrategy{}
// Protocol manager
manager := protocols.ProtocolManager{}
serviceManager := manager.InitServiceManager()
serviceManager := protocols.InitProtocolManager(logStrategy, hypertextTransferProtocolStrategy)
for _, beelzebubServiceConfiguration := range beelzebubServicesConfiguration {
switch beelzebubServiceConfiguration.Protocol {
@ -54,15 +54,20 @@ func main() {
}
<-quit
}
func logStrategy(event tracer.Event) {
log.WithFields(log.Fields{
"status": event.Status.String(),
"event": event,
}).Info("New Event")
}
func configureLogging(configurations parser.Logging) *os.File {
file, err := os.OpenFile(configurations.LogsPath, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
if err != nil {
log.Fatalf("error opening file: %v", err)
}
mw := io.MultiWriter(os.Stdout, file)
log.SetOutput(mw)
log.SetOutput(io.MultiWriter(os.Stdout, file))
log.SetFormatter(&log.JSONFormatter{
DisableTimestamp: configurations.LogDisableTimestamp,

9
protocols/protocolManager.go
View File

@ -5,14 +5,19 @@ import (
"beelzebub/tracer"
)
type ServiceStrategy interface {
Init(beelzebubServiceConfiguration parser.BeelzebubServiceConfiguration, tracer tracer.Tracer) error
}
type ProtocolManager struct {
strategy ServiceStrategy
tracer *tracer.Tracer
}
func (pm *ProtocolManager) InitServiceManager() *ProtocolManager {
func InitProtocolManager(tracerStrategy tracer.Strategy, strategy ServiceStrategy) *ProtocolManager {
return &ProtocolManager{
tracer: tracer.Init(),
tracer: tracer.Init(tracerStrategy),
strategy: strategy,
}
}

10
protocols/protocolStrategy.go
View File

@ -1,10 +0,0 @@
package protocols
import (
"beelzebub/parser"
"beelzebub/tracer"
)
type ServiceStrategy interface {
Init(beelzebubServiceConfiguration parser.BeelzebubServiceConfiguration, tracer tracer.Tracer) error
}

15
tracer/tracer.go
View File

@ -1,22 +1,23 @@
package tracer
import (
log "github.com/sirupsen/logrus"
"net/http"
)
type Strategy func(event Event)
type Tracer struct {
strategy Strategy
}
func Init() *Tracer {
return &Tracer{}
func Init(strategy Strategy) *Tracer {
return &Tracer{
strategy: strategy,
}
}
func (tracer *Tracer) TraceEvent(event Event) {
log.WithFields(log.Fields{
"status": event.Status.String(),
"event": event,
}).Info("New Event")
tracer.strategy(event)
}
type Event struct {