重构与优化之设计模式重构

设计模式是软件设计中经过验证的解决方案，使用设计模式可以提高代码的可维护性、可扩展性和可读性。AI编程工具可以帮助开发者识别代码中的设计模式应用机会，并进行设计模式重构。

1. 设计模式重构概述

设计模式重构是指在现有代码中识别和应用设计模式的过程。通过设计模式重构，可以解决以下问题：

• 代码重复：使用模板方法模式、策略模式等消除重复代码
• 耦合度高：使用依赖注入、观察者模式等降低模块间耦合
• 扩展性差：使用工厂模式、装饰器模式等提高代码扩展性
• 职责不清：使用单一职责原则和设计模式分离关注点

2. 常见设计模式重构场景

2.1 策略模式重构

当代码中存在大量的if-else或switch语句处理不同业务逻辑时，可以使用策略模式进行重构：

// 重构前：大量if-else语句
public double CalculatePrice(double price, string discountType)
{
    if (discountType == "Normal")
        return price;
    else if (discountType == "VIP")
        return price * 0.8;
    else if (discountType == "SuperVIP")
        return price * 0.6;
    else if (discountType == "Holiday")
        return price * 0.7;
    return price;
}

// 重构后：策略模式
public interface IDiscountStrategy
{
    double ApplyDiscount(double price);
}

public class NormalDiscount : IDiscountStrategy
{
    public double ApplyDiscount(double price) => price;
}

public class VIPDiscount : IDiscountStrategy
{
    public double ApplyDiscount(double price) => price * 0.8;
}

public class DiscountContext
{
    private readonly IDiscountStrategy _strategy;
    public DiscountContext(IDiscountStrategy strategy) => _strategy = strategy;
    public double CalculatePrice(double price) => _strategy.ApplyDiscount(price);
}

2.2 工厂模式重构

当对象创建逻辑复杂且分散时，可以使用工厂模式进行重构：

// 重构前：分散的对象创建
public void ProcessOrder(string orderType)
{
    if (orderType == "Online")
    {
        var order = new OnlineOrder();
        order.Process();
    }
    else if (orderType == "Offline")
    {
        var order = new OfflineOrder();
        order.Process();
    }
    else if (orderType == "VIP")
    {
        var order = new VIPOrder();
        order.Process();
    }
}

// 重构后：工厂模式
public interface IOrder
{
    void Process();
}

public class OrderFactory
{
    public static IOrder CreateOrder(string orderType)
    {
        return orderType switch
        {
            "Online" => new OnlineOrder(),
            "Offline" => new OfflineOrder(),
            "VIP" => new VIPOrder(),
            _ => throw new ArgumentException("Invalid order type")
        };
    }
}

public void ProcessOrder(string orderType)
{
    var order = OrderFactory.CreateOrder(orderType);
    order.Process();
}

2.3 装饰器模式重构

当需要动态添加功能且不希望修改现有类时，可以使用装饰器模式：

// 重构前：继承层级过深
public class BasicReport { }
public class ReportWithHeader : BasicReport { }
public class ReportWithFooter : BasicReport { }
public class ReportWithHeaderAndFooter : ReportWithHeader { }

// 重构后：装饰器模式
public interface IReport
{
    string Generate();
}

public class BasicReport : IReport
{
    public string Generate() => "Report content";
}

public abstract class ReportDecorator : IReport
{
    protected IReport _report;
    public ReportDecorator(IReport report) => _report = report;
    public virtual string Generate() => _report.Generate();
}

public class HeaderDecorator : ReportDecorator
{
    public HeaderDecorator(IReport report) : base(report) { }
    public override string Generate() => $"Header\\n{base.Generate()}";
}

public class FooterDecorator : ReportDecorator
{
    public FooterDecorator(IReport report) : base(report) { }
    public override string Generate() => $"{base.Generate()}\\nFooter";
}

3. AI辅助设计模式重构流程

flowchart TD
    A[分析现有代码] --> B[识别设计模式机会]
    B --> C[评估重构收益]
    C --> D{收益大于成本?}
    D -->|是| E[应用设计模式]
    D -->|否| F[保持现状]
    E --> G[编写单元测试]
    G --> H[验证重构结果]
    H --> I[优化调整]
    I --> J[完成重构]

4. AI提示词设计模式重构技巧

4.1 识别设计模式应用机会

使用AI识别代码中适合应用设计模式的地方：

请分析以下代码，识别其中适合应用设计模式的地方：

[粘贴代码]

请列出：
1. 可以应用的设计模式
2. 应用该模式的原因
3. 重构后的预期效果

4.2 生成设计模式实现代码

让AI根据现有代码生成设计模式重构方案：

请将以下代码使用策略模式重构：

[粘贴代码]

要求：
1. 定义策略接口
2. 创建具体策略类
3. 创建上下文类
4. 提供完整的重构代码

4.3 设计模式选择建议

让AI帮助选择最合适的设计模式：

我有一个场景：需要根据不同的支付方式处理支付逻辑，
支付方式可能会增加，且每种支付方式的处理逻辑不同。

请推荐适合的设计模式，并说明理由和实现思路。

5. 设计模式重构最佳实践

• 理解模式意图：在应用设计模式前，确保真正理解其意图和适用场景
• 不要过度设计：只有在确实需要时才应用设计模式，避免为了模式而模式
• 保持简单：优先选择简单的解决方案，设计模式是手段而非目的
• 测试驱动：在重构前编写测试，确保重构不会破坏现有功能
• 逐步重构：一次只应用一个设计模式，确保每次重构都经过验证

6. 设计模式重构示例：观察者模式

观察者模式用于实现对象间的一对多依赖关系，当一个对象状态改变时，所有依赖它的对象都会收到通知：

public interface IObserver
{
    void Update(string message);
}

public interface ISubject
{
    void Attach(IObserver observer);
    void Detach(IObserver observer);
    void Notify();
}

public class NewsPublisher : ISubject
{
    private List<IObserver> _observers = new List<IObserver>();
    private string _latestNews;

    public void Attach(IObserver observer) => _observers.Add(observer);
    public void Detach(IObserver observer) => _observers.Remove(observer);
    public void Notify()
    {
        foreach (var observer in _observers)
            observer.Update(_latestNews);
    }

    public void PublishNews(string news)
    {
        _latestNews = news;
        Notify();
    }
}

public class EmailSubscriber : IObserver
{
    public void Update(string message) => Console.WriteLine($"Email: {message}");
}

public class SMSSubscriber : IObserver
{
    public void Update(string message) => Console.WriteLine($"SMS: {message}");
}

7. 总结

设计模式重构是提高代码质量的重要手段。AI编程工具可以帮助开发者：

• 识别代码中的设计模式应用机会
• 生成设计模式重构代码
• 提供设计模式选择建议
• 验证重构后的代码质量

通过合理应用设计模式，可以使代码更加灵活、可维护和可扩展。