Refactor: Rename NanoKVM to BatchuKVM and update server URL

server/logger/formatter.go

@@ -0,0 +1,42 @@
+package logger
+
+import (
+	"bytes"
+	"fmt"
+	"path/filepath"
+
+	"github.com/sirupsen/logrus"
+)
+
+type formatter struct{}
+
+func (f *formatter) Format(entry *logrus.Entry) ([]byte, error) {
+	var (
+		text   string
+		buffer *bytes.Buffer
+	)
+
+	if entry.Buffer != nil {
+		buffer = entry.Buffer
+	} else {
+		buffer = &bytes.Buffer{}
+	}
+
+	now := entry.Time.Format("2006-01-02 15:04:05.000")
+
+	if entry.HasCaller() {
+		fileName := filepath.Base(entry.Caller.File)
+		text = fmt.Sprintf(
+			"[%s] [%s] [%s:%d] %s\n",
+			now, entry.Level, fileName, entry.Caller.Line, entry.Message,
+		)
+	} else {
+		text = fmt.Sprintf(
+			"[%s] [%s] %s \n",
+			now, entry.Level, entry.Message,
+		)
+	}
+
+	buffer.WriteString(text)
+	return buffer.Bytes(), nil
+}

server/logger/logger.go

@@ -0,0 +1,51 @@
+package logger
+
+import (
+	"os"
+	"path/filepath"
+
+	"NanoKVM-Server/config"
+
+	"github.com/sirupsen/logrus"
+)
+
+func openLogFile(filename string) (*os.File, error) {
+	absPath, err := filepath.Abs(filename)
+	if err != nil {
+		return nil, err
+	}
+
+	file, err := os.OpenFile(absPath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o644)
+	if err != nil {
+		return nil, err
+	}
+
+	return file, nil
+}
+
+func Init() {
+	conf := config.GetInstance()
+
+	level, err := logrus.ParseLevel(conf.Logger.Level)
+	if err != nil {
+		level = logrus.ErrorLevel
+	}
+
+	logrus.SetLevel(level)
+	if conf.Logger.File == "" || conf.Logger.File == "stdout" {
+		logrus.SetOutput(os.Stdout)
+	} else {
+		fh, err := openLogFile(conf.Logger.File)
+		if err != nil {
+			logrus.Error("open log file failed:", err)
+			logrus.SetOutput(os.Stdout)
+		} else {
+			logrus.SetOutput(fh)
+		}
+	}
+
+	logrus.SetReportCaller(true)
+	logrus.SetFormatter(&formatter{})
+
+	logrus.Info("logger set success")
+}