mirror of
https://github.com/kubenetworks/kubevpn.git
synced 2025-12-24 11:51:13 +08:00
a64eaf66da
* hotfix(portfoward): fix port-forward bug and ssh reconnect bug * hotfix: remove larger overlapping cidrs for adding routes * feat: retry port-forward if get pod err is not forbidden * hotfix: fix ssh and port-forward * feat: add more log * hotfix: set go default revolver perfergo options to true
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"k8s.io/utils/ptr"
|
|
)
|
|
|
|
func InitLoggerForClient(debug bool) {
|
|
if debug {
|
|
log.SetLevel(log.DebugLevel)
|
|
} else {
|
|
log.SetLevel(log.InfoLevel)
|
|
}
|
|
log.SetReportCaller(false)
|
|
log.SetFormatter(&format{})
|
|
}
|
|
|
|
func InitLoggerForServer(debug bool) {
|
|
if debug {
|
|
log.SetLevel(log.DebugLevel)
|
|
} else {
|
|
log.SetLevel(log.InfoLevel)
|
|
}
|
|
log.SetReportCaller(true)
|
|
log.SetFormatter(&serverFormat{})
|
|
}
|
|
|
|
type format struct {
|
|
log.Formatter
|
|
}
|
|
|
|
// Format
|
|
// message
|
|
func (*format) Format(e *log.Entry) ([]byte, error) {
|
|
return []byte(fmt.Sprintf("%s\n", e.Message)), nil
|
|
}
|
|
|
|
type serverFormat struct {
|
|
log.Formatter
|
|
}
|
|
|
|
// Format
|
|
// same like log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
// 2009/01/23 01:23:23 d.go:23: message
|
|
func (*serverFormat) Format(e *log.Entry) ([]byte, error) {
|
|
// e.Caller maybe is nil, because pkg/handler/connect.go:252
|
|
return []byte(
|
|
fmt.Sprintf("%s %s:%d %s: %s\n",
|
|
e.Time.Format("2006-01-02 15:04:05"),
|
|
filepath.Base(ptr.Deref(e.Caller, runtime.Frame{}).File),
|
|
ptr.Deref(e.Caller, runtime.Frame{}).Line,
|
|
e.Level.String(),
|
|
e.Message,
|
|
)), nil
|
|
}
|