go语言中,利用策略模式可通过接口和结构体实现代码复用。通过定义 log() 方法的 logger 接口,并实现具体的 consolelogger 和 filelogger 结构体,可以根据不同的日志级别动态选择并使用不同的日志记录策略。这有助于提高日志记录的可扩展性,方便修改日志记录行为。
Go语言中利用策略模式实现代码复用
策略模式简介
策略模式是一种设计模式,它允许你根据不同的情况动态地选择不同的算法或行为。这有助于提高代码的可复用性,因为你可以将算法的实现与它们的调用方式分离。
在 Go 中实现策略模式
立即学习“go语言免费学习笔记(深入)”;
在 Go 中,策略模式可以通过接口和结构体来实现:
type Strategy interface {
DoSomething()
}
type ConcreteStrategyA struct{}
func (s ConcreteStrategyA) DoSomething() {
fmt.Println("ConcreteStrategyA: DoSomething")
}
type ConcreteStrategyB struct{}
func (s ConcreteStrategyB) DoSomething() {
fmt.Println("ConcreteStrategyB: DoSomething")
}
实战案例:日志记录
让我们看一个实战案例,使用策略模式实现日志记录:
package main
import (
"context"
"fmt"
"log"
"os"
)
type LogLevel int
const (
INFO LogLevel = iota
WARNING
ERROR
)
type Logger interface {
Log(ctx context.Context, level LogLevel, message string)
}
type ConsoleLogger struct {
Level LogLevel
}
func (l *ConsoleLogger) Log(ctx context.Context, level LogLevel, message string) {
if l.Level <= level {
fmt.Fprintln(os.Stdout, message)
}
}
type FileLogger struct {
Level LogLevel
File *os.File
}
func (l *FileLogger) Log(ctx context.Context, level LogLevel, message string) {
if l.Level <= level {
fmt.Fprintln(l.File, message)
}
}
func main() {
// 创建不同的日志记录器
consoleLogger := &ConsoleLogger{Level: INFO}
fileLogger, err := os.OpenFile("log.txt", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
log.Fatal(err)
}
defer fileLogger.Close()
fileLogger := &FileLogger{Level: ERROR, File: fileLogger}
// 使用策略模式写日志
consoleLogger.Log(context.Background(), INFO, "This is an info message.")
fileLogger.Log(context.Background(), ERROR, "This is an error message.")
}
在这个示例中,Logger 接口定义了 Log() 方法,该方法根据日志级别记录一条消息。ConsoleLogger 和 FileLogger 结构体实现了 Logger 接口,指定了如何记录消息以及使用哪个日志级别。
使用策略模式,我们可以在不同的上下文和要求中动态地选择不同的日志记录策略。这使我们可以轻松地扩展和修改日志记录行为,而无需修改调用代码。