logmania/log/hook/client/main.go

// logger to bind at github.com/genofire/logmania/log.AddLogger to send log entries to logmania server
package client

import (
	"fmt"

	"github.com/gorilla/websocket"

	"github.com/genofire/logmania/log"
)

// client logger
type Logger struct {
	log.Logger
	AboveLevel log.LogLevel
	conn       *websocket.Conn
}

const LOGGER_NAME = "client"

// CurrentLogger (for override settings e.g. AboveLevel)
var CurrentLogger *Logger

// create a new logmania client logger
func NewLogger(url, token string, AboveLevel log.LogLevel) *Logger {
	c, _, err := websocket.DefaultDialer.Dial(fmt.Sprint(url, "/logger"), nil)
	if err != nil {
		log.Error("[logmania] error on connect: ", err)
		return nil
	}
	err = c.WriteMessage(websocket.TextMessage, []byte(token))
	if err != nil {
		log.Error("[logmania] could not send token:", err)
		return nil
	}
	return &Logger{
		AboveLevel: AboveLevel,
		conn:       c,
	}
}

// handle a log entry (send to logmania server)
func (l *Logger) Hook(e *log.Entry) {
	if e.Level < l.AboveLevel {
		return
	}
	err := l.conn.WriteJSON(e)
	if err != nil {
		l.Close()
		log.Error("[logmania] could not send log entry:", err)
	}
}

// Listen if logmania server want to close the connection
func (l *Logger) Listen() {
	for {
		msgType, _, err := l.conn.ReadMessage()
		if msgType == -1 {
			l.conn.Close()
			l.Close()
			return
		}
		if err != nil {
			l.Close()
			log.Warn("[logmania] close listener:", err)
		}
	}
}

// close connection to logger
func (l *Logger) Close() {
	l.conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
	log.RemoveLogger(LOGGER_NAME)
}

// init logmania's client logger and bind
func Init(url, token string, AboveLevel log.LogLevel) *Logger {
	CurrentLogger = NewLogger(url, token, AboveLevel)
	go CurrentLogger.Listen()
	log.AddLogger(LOGGER_NAME, CurrentLogger)
	return CurrentLogger
}