策略模式
策略模式主要解决枚举变化的情况,转化为一系列算法类。
enum TaxBase
{
CN_Tax,
UR_Tax,
UK_Tax,
FR_Tax
};
class SalesOrder {
private:
TaxBase tax;
public:
double CalculateTax() {
if (tax == CN_Tax) {
//...
}
else if(tax==UR_Tax){
//...
}
else if (tax == UK_Tax) {
}
}
};
上述代码以汇率计算为例,可以看到,当要扩展另一个国家的汇率计算时,需要修改枚举和SalesOrder类,加上对应的else if,这样不利于扩展,可以将枚举定义成基类,如下代码所示:
#pragma once
#ifndef STRATEGY_H
#define STRATEGY_H
#include<iostream>
using namespace std;
class TaxStrategy {
public:
virtual double CalculateTax(const double& money) = 0;
virtual ~TaxStrategy() {}
};
class CNTax :public TaxStrategy {
public:
double CalculateTax(const double& money) override{
cout << "Chinese money" << endl;
return money;
}
};
class USTax :public TaxStrategy {
public:
double CalculateTax(const double& money) override {
cout << "US money" << endl;
return money;
}
};
class UKTax :public TaxStrategy {
public:
double CalculateTax(const double& money) override {
cout << "UK money" << endl;
return money;
}
};
#endif // !STRATEGY_H
strategy.h中,改变了枚举的类型,变成继承自抽象基类的子类。测试代码如下:
#include<iostream>
using namespace std;
#include"strategy.h"
class SalesOrder {
private:
TaxStrategy* strategy;
public:
SalesOrder(TaxStrategy* strategy) {//若有多个策略,此处可以结合工厂模式
this->strategy = strategy;
}
~SalesOrder() {
delete this->strategy;
}
double Calculatetax()
{
double val = strategy->CalculateTax(10);//多态调用
cout << "success calculate" << endl;
return val;
}
};
int main()
{
SalesOrder* sale = new SalesOrder(new USTax());
sale->Calculatetax();
return 0;
}
通过这种方式,解决了上述扩展时,需要更改各种文件代码的问题,只需要添加扩展类即可,也无需修改SalesOrder 类的方法。
运行结果如下:

本文介绍了如何通过策略模式解决枚举类型的扩展问题。原始代码中,`SalesOrder`类根据`CN_Tax`等枚举计算税额,导致扩展新国家税额时需修改多处代码。为了解决这一问题,文章引入了策略模式,将枚举转换为一系列继承自`TaxStrategy`抽象基类的子类,如`CNTax`、`USTax`等。这样,在添加新国家税额计算时,只需创建新的策略类,而无需修改`SalesOrder`。测试代码展示了如何使用多态调用不同国家的税额计算,实现了更灵活的扩展性。
1817

被折叠的 条评论
为什么被折叠?



