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

news2024/11/9 3:51:47

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

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 语句。

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

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

// 定义策略接口
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