程序中的黑白名单控制

       在软件开发中, 经常需要对程序进行调试, 很多时候, 我们需要增加访问控制, 也就是所谓的黑白名单。 将普通用户定义为黑名单用户, 将VIP用户定义为白名单。 下面来看看程序:

#include 
#include 
#include 
#include 
using namespace std;

int main()
{
	string strUser;
	cin >> strUser;

	#define DEBUG_FLAG    // 仅在调试时打开
	#ifdef DEBUG_FLAG
	ifstream whiteNameFile("whiteNameFile.conf");  // 为避免为STL中的list混淆, 我把白名单称作white name, 而非white list
	set setWhiteNames;
	set::iterator itsetWhiteNames;
	string line;
	if(whiteNameFile)  // 有该文件
	{
		while (getline (whiteNameFile, line))  // line中不包括每行的换行符
		{ 
			cout << line << endl;
			setWhiteNames.insert(line);
		}
	}
	else // 没有该文件
	{
		cout <<"no such file" << endl;
		return -1;
	}

	itsetWhiteNames = setWhiteNames.find(strUser);
	if(setWhiteNames.end() == itsetWhiteNames)  // 不在白名单中
	{
		cout << "black name" << endl;
		return -2;
	}
	#endif   // end  #ifdef DEBUG_FLAG
	
	
	cout << "do business logic" << endl;
	return 0;
}

        经调试, 靠谱。  在调试阶段, 用黑白名单控制。 发布时, 需要注释掉如下行, 让黑白名单控制失效:

	#define DEBUG_FLAG    // 仅在调试时打开

        当然, 如果把debug release开关放在编译阶段, 那也可以, 此时不需要修改代码了。 这个很简单, 故不赘述。




你可能感兴趣的:(S1:,C/C++,S1:,STL,s2:,软件进阶)