策略模式通过统一接口解耦算法实现与使用逻辑,使Go程序能灵活切换压缩、加密等算法;定义CompressionStrategy接口规范压缩解压行为,Gzip、Zstd等具体策略分别实现该接口;运行时通过Compressor结构体动态设置策略,避免条件判断,结合工厂函数NewCompressionStrategy按名称创建策略,简化配置;接口隔离变化,新增算法无需修改调用代码,提升可维护性与扩展性。

在Go语言中,策略模式能有效解耦算法实现与使用逻辑,让程序更灵活地切换不同算法。面对多种算法可选的场景,比如排序、压缩或加密,策略模式通过统一接口隔离变化,使新增或替换算法变得简单安全。
定义统一的策略接口
所有具体算法需实现同一接口,这样调用方无需关心具体实现。以数据压缩为例:
type CompressionStrategy interface {
Compress(data []byte) ([]byte, error)
Decompress(data []byte) ([]byte, error)
}
该接口规定了压缩与解压行为,任何符合此契约的算法都能无缝接入。
实现多种具体策略
不同算法各自实现接口。例如:
立即学习“go语言免费学习笔记(深入)”;
type GzipStrategy struct{}
func (g *GzipStrategy) Compress(data []byte) ([]byte, error) {
var buf bytes.Buffer
writer := gzip.NewWriter(&buf)
_, err := writer.Write(data)
if err != nil {
return nil, err
}
writer.Close()
return buf.Bytes(), nil
}
func (g *GzipStrategy) Decompress(data []byte) ([]byte, error) {
reader, err := gzip.NewReader(bytes.NewReader(data))
if err != nil {
return nil, err
}
defer reader.Close()
return io.ReadAll(reader)
}
同样可实现Zstd、LZ4等策略。每个策略独立封装,互不影响。
运行时动态选择算法
通过配置或输入决定使用哪种策略:
type Compressor struct {
strategy CompressionStrategy
}
func (c *Compressor) SetStrategy(s CompressionStrategy) {
c.strategy = s
}
func (c *Compressor) Process(data []byte) ([]byte, error) {
if c.strategy == nil {
return nil, fmt.Errorf("no strategy set")
}
return c.strategy.Compress(data)
}
使用时根据条件切换:
compressor := &Compressor{}
if useGzip {
compressor.SetStrategy(&GzipStrategy{})
} else {
compressor.SetStrategy(&ZstdStrategy{})
}
result, _ := compressor.Process(inputData)
这种设计避免了大量条件判断,扩展新算法只需新增结构体并实现接口。
结合工厂模式简化创建
若策略选择逻辑复杂,可用工厂函数封装创建过程:
func NewCompressionStrategy(name string) (CompressionStrategy, error) {
switch name {
case "gzip":
return &GzipStrategy{}, nil
case "zstd":
return &ZstdStrategy{}, nil
default:
return nil, fmt.Errorf("unsupported strategy: %s", name)
}
}
调用方只需传入名称即可获取对应策略,进一步降低耦合。
基本上就这些。策略模式在Go中借助接口和组合,能清晰分离算法与上下文,提升代码可维护性和测试便利性。不复杂但容易忽略的是保持接口简洁,避免因过度设计引入冗余抽象。










