56 lines
1020 B
Go
56 lines
1020 B
Go
package logrus
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"dev.sum7.eu/genofire/golang-lib/websocket"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"dev.sum7.eu/genofire/logmania/input"
|
|
)
|
|
|
|
const inputType = "logrus"
|
|
const WS_LOG_ENTRY = "log"
|
|
|
|
var logger = log.WithField("input", inputType)
|
|
|
|
type Input struct {
|
|
input.Input
|
|
input chan *websocket.Message
|
|
exportChannel chan *log.Entry
|
|
serverSocket *websocket.Server
|
|
}
|
|
|
|
func Init(config interface{}, exportChannel chan *log.Entry) input.Input {
|
|
inputMsg := make(chan *websocket.Message)
|
|
ws := websocket.NewServer(inputMsg, websocket.NewSessionManager())
|
|
|
|
http.HandleFunc("/input/"+inputType, ws.Handler)
|
|
|
|
input := &Input{
|
|
input: inputMsg,
|
|
serverSocket: ws,
|
|
exportChannel: exportChannel,
|
|
}
|
|
|
|
logger.Info("init")
|
|
|
|
return input
|
|
}
|
|
|
|
func (in *Input) Listen() {
|
|
logger.Info("listen")
|
|
for msg := range in.input {
|
|
if event, ok := msg.Body.(log.Entry); ok {
|
|
in.exportChannel <- &event
|
|
}
|
|
}
|
|
}
|
|
|
|
func (in *Input) Close() {
|
|
}
|
|
|
|
func init() {
|
|
input.Add(inputType, Init)
|
|
}
|