47 lines
813 B
Go
47 lines
813 B
Go
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/gorilla/websocket"
|
||
|
|
|
||
|
|
"github.com/genofire/logmania/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Logger struct {
|
||
|
|
AboveLevel log.LogLevel
|
||
|
|
conn *websocket.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Logger) hook(e *log.Entry) {
|
||
|
|
if e.Level < l.AboveLevel {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
err := l.conn.WriteJSON(e)
|
||
|
|
if err != nil {
|
||
|
|
log.Panic("[logmania] could not send token")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func (l *Logger) Close() {
|
||
|
|
l.conn.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func Init(url, token string) *Logger {
|
||
|
|
logger := &Logger{
|
||
|
|
AboveLevel: log.InfoLevel,
|
||
|
|
}
|
||
|
|
c, _, err := websocket.DefaultDialer.Dial(fmt.Sprint(url, "/logger"), nil)
|
||
|
|
if err != nil {
|
||
|
|
log.Panic("[logmania] error on connect")
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
err = c.WriteJSON(token)
|
||
|
|
if err != nil {
|
||
|
|
log.Panic("[logmania] could not send token")
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
logger.conn = c
|
||
|
|
log.AddHook(logger.hook)
|
||
|
|
return logger
|
||
|
|
}
|