具有可变数量参数的函数

Function with variable number of argments

本文关键字:参数 函数      更新时间:2023-10-16

我的程序中有一个函数,可以计算给定目录中的文件数。该函数将输入参数作为目录的名称和文件名的扩展名(必须在目录中计数的文件类型)。

我想让它通用化,这样函数的调用方就可以指定任意数量的文件扩展名,这些扩展名可以作为输入参数,然后计算具有这些扩展名的特定文件。

void getNames(string dirName,  string fileExtension1, string fileExtension2, string fileExtension3){
vector<string> fileNames = //do some operations
if(fileType == fileExtension1 || fileType==fileExtension2....){
//increase count
}
return fileNames;
}

如何更改此函数,使其接受任意数量的参数,然后在函数内部使用相同的参数来计算文件数量?

最简单的方法是通过扩展的向量

void getNames(string dirName, const vector<string>& fileExtensions)

现在您的if将在for循环中

for(int i = 0; i<fileExtensions.size();i++){
if(fileType == fileExtension[i]){
//increase count
break;// this might be needed to ensure no multiple positives? depending on what you want to do.
}
}

在您的代码中,您有基于类型的决策:

if(fileType == fileExtension1 || fileType==fileExtension2....){
//increase count
}
return fileNames;
}

如果你有很多fileType和函数在上面运行,每次你都需要一些fileType特定的操作,你最终会得到广泛的基于类型的决策逻辑(如果你是这种类型,就这样做,否则如果你是…)

由于该函数是基于"fileType"(扩展名)对"fileName"执行某些操作,因此您可能会考虑将两者封装到一个类File中,因为每个文件都有一个名称和扩展名。然后,您的函数将对一组File对象进行操作。如果文件对象集是你在很多地方都在处理的东西,你可以定义一个类,例如FileSet,然后可以用向量、列表或任何存储文件的容器来实现。这样,您将得到一个带有单个参数的函数,例如const FileSet&,其余代码将由File类处理。无论如何,这不是一个明确的答案,我只建议考虑一下代码的OOD设计。