重构代码之用多态替代条件逻辑

在代码重构中,用多态替代条件逻辑 是一种将冗长的条件逻辑替换为面向对象多态性的技巧。这种技术的核心思想是将不同条件分支中的行为封装到各自的子类中,从而避免在代码中出现复杂的if-elseswitch条件判断。通过使用多态性,可以让代码更具可读性、易于扩展,并且符合"开闭原则"(Open-Closed Principle)。

一、何时使用

用多态替代条件逻辑适用于以下场景:

  1. 重复的条件逻辑:代码中有多个条件分支,根据特定的状态或类型执行不同的逻辑。
  2. 复杂的条件分支if-elseswitch语句变得非常复杂,难以维护和理解。
  3. 频繁扩展的业务需求:代码逻辑可能经常需要扩展(添加新的分支),这样子类的设计会更适合长期维护。

二、代码示例

假设我们有一个订单系统,根据不同的订单类型来计算折扣。

2.1 重构前

在重构前,代码中存在一个复杂的switch语句:

csharp 复制代码
public class Order
{
    public enum OrderType { Regular, Premium, VIP }

    public OrderType Type { get; set; }
    public decimal Amount { get; set; }

    public decimal GetDiscount()
    {
        switch (Type)
        {
            case OrderType.Regular:
                return Amount * 0.05m; // 5% discount
            case OrderType.Premium:
                return Amount * 0.10m; // 10% discount
            case OrderType.VIP:
                return Amount * 0.20m; // 20% discount
            default:
                throw new InvalidOperationException("Unknown order type");
        }
    }
}

这个switch逻辑可能会随着业务需求增加而不断扩展,每次添加新类型的订单都需要修改此方法,不符合开闭原则。

2.2 重构后

可以将每种订单类型的折扣逻辑封装到各自的子类中,使用多态来替代条件判断:

csharp 复制代码
public abstract class Order
{
    public decimal Amount { get; set; }

    public abstract decimal GetDiscount();
}

public class RegularOrder : Order
{
    public override decimal GetDiscount() => Amount * 0.05m;
}

public class PremiumOrder : Order
{
    public override decimal GetDiscount() => Amount * 0.10m;
}

public class VIPOrder : Order
{
    public override decimal GetDiscount() => Amount * 0.20m;
}

然后,我们在创建订单时,根据订单类型选择相应的子类:

csharp 复制代码
public class OrderFactory
{
    public static Order CreateOrder(string type, decimal amount)
    {
        Order order = type switch
        {
            "Regular" => new RegularOrder { Amount = amount },
            "Premium" => new PremiumOrder { Amount = amount },
            "VIP" => new VIPOrder { Amount = amount },
            _ => throw new ArgumentException("Invalid order type")
        };
        return order;
    }
}

使用时:

csharp 复制代码
Order order = OrderFactory.CreateOrder("VIP", 1000);
decimal discount = order.GetDiscount();
Console.WriteLine($"Discount: {discount}");

三、优点

  1. 代码简洁:消除了条件语句,使代码更加清晰。
  2. 易于扩展:添加新的订单类型时,只需创建一个新的子类,而不需要修改现有代码,符合开闭原则。
  3. 提高可维护性:每个类的职责更加明确,便于理解和维护。

四、总结

通过"用多态替代条件逻辑"这一重构手法,我们可以将条件判断的复杂逻辑转移到类的层次结构上,从而提高代码的灵活性和可维护性。这一技术尤其适合频繁扩展的业务场景,使得新增功能的引入变得简单。

相关推荐
真正的醒悟26 分钟前
2025【1460天】网络工程师经验之道
运维·服务器
float_六七3 小时前
IntelliJ IDEA双击Ctrl的妙用
java·ide·intellij-idea
Tim风声(网络工程师)3 小时前
不同射频对应不同mac地址(查找无线用户连接AP信息)
服务器·网络·tcp/ip·智能路由器·无线ap
能摆一天是一天5 小时前
JAVA stream().flatMap()
java·windows
颜如玉5 小时前
🤲🏻🤲🏻🤲🏻临时重定向一定要能重定向🤲🏻🤲🏻🤲🏻
java·http·源码
程序员的世界你不懂7 小时前
【Flask】测试平台开发,新增说明书编写和展示功能 第二十三篇
java·前端·数据库
星空寻流年7 小时前
设计模式第一章(建造者模式)
java·设计模式·建造者模式
鳄鱼杆7 小时前
服务器 | Docker应用开发与部署的实践以及阿里云镜像加速配置
服务器·阿里云·docker
羚羊角uou7 小时前
【Linux】命名管道
linux·运维·服务器
gb42152877 小时前
java中将租户ID包装为JSQLParser的StringValue表达式对象,JSQLParser指的是?
java·开发语言·python