代码中的设计模式-策略模式

假如我们有一段代码,有很多的if else

javascript 复制代码
function executeAction(type) {
    if (type === 'A') {
        console.log('Action A');
    } else if (type === 'B') {
        console.log('Action B');
    } else if (type === 'C') {
        console.log('Action C');
    } else {
        console.log('Unknown action');
    }
}

executeAction('A'); // 输出: Action A
executeAction('B'); // 输出: Action B
executeAction('C'); // 输出: Action C
executeAction('D'); // 输出: Unknown action

如果if else有很多很多, 那么代码就会变的非常不直观且难以维护

那对于这种情况有什么好的解决办法吗

这就引出了我们今天的主角: 策略模式

什么是策略模式

策略模式(Strategy Pattern)是一种行为设计模式,它允许你定义一系列算法或行为,并将它们封装在独立的策略类中。然后,你可以根据需要在运行时选择不同的策略。这种模式特别适合处理复杂的条件逻辑,比如多个 if-else if 语句。

下面来看使用策略模式的解决方案

我们可以将每个条件分支封装在一个策略类中,然后在运行时选择合适的策略。

javascript 复制代码
// 定义策略接口
class Strategy {
    execute() {
        throw new Error('You have to implement the method execute!');
    }
}

// 具体策略A
class StrategyA extends Strategy {
    execute() {
        console.log('Action A');
    }
}

// 具体策略B
class StrategyB extends Strategy {
    execute() {
        console.log('Action B');
    }
}

// 具体策略C
class StrategyC extends Strategy {
    execute() {
        console.log('Action C');
    }
}

// 默认策略
class DefaultStrategy extends Strategy {
    execute() {
        console.log('Unknown action');
    }
}

// 策略上下文
class Context {
    constructor() {
        this.strategies = {
            'A': new StrategyA(),
            'B': new StrategyB(),
            'C': new StrategyC()
        };
    }

    executeAction(type) {
        const strategy = this.strategies[type] || new DefaultStrategy();
        strategy.execute();
    }
}

// 使用策略模式
const context = new Context();
context.executeAction('A'); // 输出: Action A
context.executeAction('B'); // 输出: Action B
context.executeAction('C'); // 输出: Action C
context.executeAction('D'); // 输出: Unknown action

策略接口 (Strategy): 定义了一个 execute 方法,所有具体的策略类都必须实现这个方法。

具体策略类 (StrategyA, StrategyB, StrategyC, DefaultStrategy): 每个类都实现了 execute 方法,执行特定的操作。

策略上下文 (Context): 维护一个策略对象的映射,并根据传入的 type 选择合适的策略来执行。如果没有找到对应的策略,则使用默认策略。

使用策略模式: 通过 Context 类的 executeAction 方法来执行不同的策略。

这样实现有哪些好处

可扩展性: 如果需要添加新的策略,只需添加一个新的策略类,并在 Context 中注册即可,无需修改现有代码。

代码清晰: 将复杂的条件逻辑分解到不同的策略类中,使代码更易于理解和维护。

灵活性: 可以在运行时动态地选择不同的策略,而不需要修改代码。

相关推荐
Cosolar5 小时前
提示词工程面试题系列 - Zero-Shot Prompting 和 Few-Shot Prompting 的核心区别是什么?
人工智能·设计模式·架构
geovindu11 小时前
go:Template Method Pattern
开发语言·后端·设计模式·golang·模板方法模式
钝挫力PROGRAMER12 小时前
贫血模型的改进
java·开发语言·设计模式·架构
qcx2313 小时前
Warp源码深度解析(二):自研GPU UI框架——WarpUI的ECH模式与渲染管线
人工智能·ui·设计模式·rust
qcx2315 小时前
Warp源码深度解析(三):Block-Based终端引擎——Grid模型、PTY与Shell Integration
人工智能·设计模式·架构·wrap
mounter62515 小时前
Linux Kernel Design Patterns (Part 2):从经典链表到现代 XArray,拆解内核复杂数据结构的设计哲学
linux·数据结构·链表·设计模式·内存管理·kernel
rrr215 小时前
【PyQt5】| 多线程设计模式
开发语言·qt·设计模式
SteveDraw15 小时前
常见的设计模式及工业场景下应用(更新中)
设计模式·c#·编码规范·gof23
ximu_polaris15 小时前
设计模式(C++)-行为型模式-状态模式
c++·设计模式·状态模式
ximu_polaris15 小时前
设计模式(C++)-行为型模式-迭代器模式
c++·设计模式·迭代器模式