C++中从unique_ptr向量获取指定命令对象的类型疑问
命令行应用:根据名称获取命令对象的类型与实现方案
问题场景
我正在开发一款命令行应用,定义了BaseCommand基类及HelpCommand、ProductCommand两个子类,代码如下:
BaseCommand基类
struct BaseCommand { public: virtual ~BaseCommand() {} int id = 0; std::string name = "Base Command"; std::string description = "Base Command"; BaseCommand(); virtual std::string getName(); virtual int getId(); virtual std::string getDescription(); private: virtual void init(); virtual void run(); };
HelpCommand子类
class HelpCommand: public BaseCommand { public: HelpCommand(); std::string name = "help"; std::string description = "output help"; int id = 1; virtual std::string getName(); int getId() {return this->id;}; virtual std::string getDescription(); private: virtual void init(); virtual void run(); };
ProductCommand子类
class ProductCommand: public BaseCommand { public: ProductCommand(); std::string name = "prod"; std::string description = "product"; int id = 2; virtual std::string getName(); int getId() {return this->id;}; virtual std::string getDescription(); private: virtual void init(); virtual void run(); };
在主函数中,我将子类实例存入std::vector<std::unique_ptr<BaseCommand>>,希望根据命令名称获取对应的命令对象,但无法确定selectedCommand的正确类型,相关代码片段:
std::vector<std::unique_ptr<BaseCommand>> commands; // 存入命令实例... commands.emplace_back(new HelpCommand); commands.emplace_back(new ProductCommand); std::string command = "prod"; // selectedCommand应是什么类型? ?? selectedCommand; for (int i = 0; i < commands.size(); ++i) { if (commands[i]->getName() == command) { selectedCommand = commands[i]; } }
解决方案
1. selectedCommand的正确类型
因为vector中存储的是std::unique_ptr<BaseCommand>,我们不需要转移对象所有权,只是要引用容器中已有的命令对象,所以最适合的类型是裸指针BaseCommand*,声明方式:
BaseCommand* selectedCommand = nullptr;
如果后续需要接管对象所有权,可以用std::unique_ptr<BaseCommand>,但此时需要用std::move转移所有权(注意:转移后原vector中的对应元素会变为空):
std::unique_ptr<BaseCommand> selectedCommand; // 赋值时 selectedCommand = std::move(commands[i]);
但通常命令对象需要长期存在于容器中供多次使用,所以裸指针是更合理的选择。
2. 修正后的查找逻辑
结合裸指针的使用,修正后的代码如下:
std::vector<std::unique_ptr<BaseCommand>> commands; commands.emplace_back(std::make_unique<HelpCommand>()); // 推荐用make_unique代替new commands.emplace_back(std::make_unique<ProductCommand>()); std::string command = "prod"; BaseCommand* selectedCommand = nullptr; for (const auto& cmd_ptr : commands) { // 用范围for循环更简洁 if (cmd_ptr->getName() == command) { selectedCommand = cmd_ptr.get(); // 获取裸指针 break; // 找到后直接退出循环,避免多余遍历 } } // 后续使用selectedCommand if (selectedCommand != nullptr) { // 调用虚函数,比如执行命令 // selectedCommand->run(); std::cout << "Found command: " << selectedCommand->getName() << std::endl; } else { std::cout << "Command not found: " << command << std::endl; }
3. 优化子类设计(重要)
当前子类代码存在成员变量隐藏的问题:子类中重新定义了name、id、description,这些变量会隐藏基类的同名成员。比如如果基类的getName()实现是返回this->name,那么通过基类指针调用getName()时,会返回基类的name值,而非子类的。
优化方案:子类不要重新定义这些成员,而是在构造函数中初始化基类的成员:
class HelpCommand: public BaseCommand { public: HelpCommand() { id = 1; name = "help"; description = "output help"; } // 不需要重新定义name/id/description // 虚函数可以直接继承基类的实现,除非需要特殊逻辑 int getId() override { return id; }; // 用override关键字明确重写,更安全 private: void init() override; void run() override; };
这样既避免了成员隐藏问题,又能保证通过基类指针调用getName()、getId()时返回正确的子类值。
内容的提问来源于stack exchange,提问作者Zymawy
相关产品推荐
相关产品推荐

