随着互联网的迅速发展，日志已经成为了一个不可或缺的组成部分。它可以帮助我们记录系统运行情况，以便我们更好地了解系统运行的状态，及时发现问题并进行处理。在这篇文章中，我将向您介绍如何使用Go语言编写高效的日志接口并实现存储。
首先，让我们来看一下日志的基本结构。一个典型的日志记录包括时间戳、日志级别、日志内容等信息。在Go语言中，可以使用log包来实现日志的基本功能。但是，在实际应用中，我们往往需要更加灵活和高效的日志接口，这时候就需要自己编写日志接口了。
下面是一个简单的日志接口实现：
package logger
import (
"fmt"
"os"
"sync"
"time"
)
type LogLevel int
const (
DEBUG LogLevel = iota
INFO
WARN
ERROR
)
type Logger struct {
level LogLevel
mu sync.Mutex
}
func NewLogger(level LogLevel) *Logger {
return &Logger{level: level}
}
func (l *Logger) Debug(format string, v ...interface{}) {
l.log(DEBUG, format, v...)
}
func (l *Logger) Info(format string, v ...interface{}) {
l.log(INFO, format, v...)
}
func (l *Logger) Warn(format string, v ...interface{}) {
l.log(WARN, format, v...)
}
func (l *Logger) Error(format string, v ...interface{}) {
l.log(ERROR, format, v...)
}
func (l *Logger) log(level LogLevel, format string, v ...interface{}) {
if l.level > level {
return
}
l.mu.Lock()
defer l.mu.Unlock()
now := time.Now().Format("2006-01-02 15:04:05")
msg := fmt.Sprintf(format, v...)
file := fmt.Sprintf("%s.log", now[:10])
f, err := os.OpenFile(file, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
if err != nil {
fmt.Fprintf(os.Stderr, "open log file error: %v
", err)
return
}
defer f.Close()
logMsg := fmt.Sprintf("[%s] [%s] %s
", now, levelString(level), msg)
_, err = f.WriteString(logMsg)
if err != nil {
fmt.Fprintf(os.Stderr, "write log error: %v
", err)
}
}
func levelString(level LogLevel) string {
switch level {
case DEBUG:
return "DEBUG"
case INFO:
return "INFO"
case WARN:
return "WARN"
case ERROR:
return "ERROR"
default:
return "UNKNOWN"
}
}在上面的代码中，我们定义了一个Logger结构体，它包含日志级别和互斥锁。日志级别用来控制日志输出的级别，互斥锁用来保证日志的线程安全。
我们还定义了NewLogger()函数来创建Logger对象，并且定义了Debug()、Info()、Warn()和Error()等方法来输出不同级别的日志。在这些方法中，我们调用了log()方法来实现具体的日志输出功能。
在log()方法中，我们首先判断当前日志级别是否需要输出，如果不需要，直接返回。然后，我们加锁，以保证多线程下的安全性。接下来，我们获取当前时间，并将日志信息写入到文件中。
在实际应用中，我们还可以将日志信息写入到数据库中，或者通过网络发送到远程服务器。下面是一个将日志信息写入到数据库的示例代码：
package logger
import (
"fmt"
"sync"
"time"
"github.com/go-sql-driver/mysql"
)
type LogLevel int
const (
DEBUG LogLevel = iota
INFO
WARN
ERROR
)
type Logger struct {
level LogLevel
mu sync.Mutex
db *mysql.Conn
}
func NewLogger(level LogLevel, dsn string) (*Logger, error) {
db, err := mysql.Open("mysql", dsn)
if err != nil {
return nil, err
}
return &Logger{level: level, db: db}, nil
}
func (l *Logger) Debug(format string, v ...interface{}) {
l.log(DEBUG, format, v...)
}
func (l *Logger) Info(format string, v ...interface{}) {
l.log(INFO, format, v...)
}
func (l *Logger) Warn(format string, v ...interface{}) {
l.log(WARN, format, v...)
}
func (l *Logger) Error(format string, v ...interface{}) {
l.log(ERROR, format, v...)
}
func (l *Logger) log(level LogLevel, format string, v ...interface{}) {
if l.level > level {
return
}
l.mu.Lock()
defer l.mu.Unlock()
now := time.Now().Format("2006-01-02 15:04:05")
msg := fmt.Sprintf(format, v...)
_, err := l.db.Exec("INSERT INTO log (time, level, msg) VALUES (?, ?, ?)", now, levelString(level), msg)
if err != nil {
fmt.Fprintf(os.Stderr, "write log error: %v
", err)
}
}
func levelString(level LogLevel) string {
switch level {
case DEBUG:
return "DEBUG"
case INFO:
return "INFO"
case WARN:
return "WARN"
case ERROR:
return "ERROR"
default:
return "UNKNOWN"
}
}在上面的代码中，我们使用了go-sql-driver/mysql包来连接MySQL数据库。在NewLogger()函数中，我们创建了一个MySQL连接，并将其保存到Logger对象中。在log()方法中，我们使用Exec()方法将日志信息插入到log表中。
在实际应用中，我们还可以使用第三方日志库来实现更加丰富的日志功能。比如，可以使用logrus或zap等日志库。
总之，使用Go语言编写高效的日志接口并实现存储是非常有必要的。通过合理的日志记录和分析，我们可以更好地了解系统运行情况，及时发现问题并进行处理。
