52 lines
773 B
Go
52 lines
773 B
Go
package tunnel
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
INFO LogType = iota
|
|
WARNING
|
|
ERROR
|
|
DEBUG
|
|
)
|
|
|
|
type LogType int
|
|
|
|
type Log struct {
|
|
LogType LogType
|
|
Payload string
|
|
}
|
|
|
|
func print(data Log) {
|
|
switch data.LogType {
|
|
case INFO:
|
|
log.Infoln(data.Payload)
|
|
case WARNING:
|
|
log.Warnln(data.Payload)
|
|
case ERROR:
|
|
log.Errorln(data.Payload)
|
|
case DEBUG:
|
|
log.Debugln(data.Payload)
|
|
}
|
|
}
|
|
|
|
func (t *Tunnel) subscribeLogs() {
|
|
sub, err := t.observable.Subscribe()
|
|
if err != nil {
|
|
log.Fatalf("Can't subscribe tunnel log: %s", err.Error())
|
|
}
|
|
for elm := range sub {
|
|
data := elm.(Log)
|
|
print(data)
|
|
}
|
|
}
|
|
|
|
func newLog(logType LogType, format string, v ...interface{}) Log {
|
|
return Log{
|
|
LogType: logType,
|
|
Payload: fmt.Sprintf(format, v...),
|
|
}
|
|
}
|