本篇记录项目中的一些小点,比如支持读取配置文件,命令行读取参数等。
读取配置文件
程序运行中用到的一些参数,首先会读取配置文件中的配置,若在程序运行时指定了参数,按指定的参数运行,首先来看读取配置文件:
配置文件是server.conf
文件,配置项按照key:value
的方式书写,比如:
PORT:8080
THREADNUM:8
端口号为8080,线程数为8。
来看一下读取配置文件的类Config
中的关键代码:
void Config::readConfigFile()
{
char *pLine = nullptr;
size_t len = 0;
ssize_t nread;
while ((nread = getline(&pLine, &len, m_pFp)) != -1) {
//移除多余的空格等符号
char *p = strchr(pLine, ':');
if (p == nullptr) return;
char *pValue = removeSpace(p, strlen(p));
char *pKey = removeSpace(pLine, strlen(pLine) - strlen(p) - 1);
printf("%s, %s", pKey, pValue);
m_mapKeyValue.insert(std::make_pair(pKey, pValue));
}
}
按行读取配置文件,以:
分割键值,去除两边多余的空格,保存到一个map成员变量m_mapKeyValue
中,这样获取配置的时候直接从内存中取即可,不用重复读取文件。
命令行参数
程序支持命令行读取参数,比如
./server -d -p 8080 //-d表示以守护进程方式运行,-p表示执行端口号
主要是通过getopt
调用:
int opt = 0;
while ((opt = getopt(argc, argv, "dp:")) != -1) {
switch (opt) {
case 'd':
//守护进程
{
pid_t pid;
if (fork() < 0) {
perror("fork()");
} else if (fork() > 0) {
exit(0);
}
//子进程
setsid();
close(1);
}
break;
case 'p':
//端口号
iPort = atoi(optarg);
break;
default:
fprintf(stderr, "Usage: %s [-d] [-p port]\n", argv[0]);
exit(-1);
}
}
调试宏
在调试程序时,除了可以查看日志的输出,还可以定义一个这样的宏:
#ifdef DEBUG
#define DBG(format, args...) {\
printf(format, ##args);\
}
#else
#define DBG(format, ...)
#endif
当我们需要输出Debug信息时,在gcc编译时通过-D
选项定义一个DEBUG宏即可。比如:
gcc test.cpp -DDEBUG
这样运行后便可以看到一些输出信息,方便调试。
构建脚本
项目采用cmake构建,执行cmake后往往还需要执行make构建,因此我们可以将这些操作放在一个脚本中做,比如build.sh
#!/bin/bash
mkdir build
cd build
cmake ../
make
这样执行一个build.sh即可。
日志文件输出
server运行产生的日志文件,以当前日期命名(比如2022_05_13_log
),并输出在当前目录:
char log_full_name[256] = {0};
getcwd(log_full_name, 256);
snprintf(log_full_name, 255, "%s%d_%02d_%02d_%s", log_full_name, my_tm.tm_year + 1900, my_tm.tm_mon +1, my_tm.tm_mday, FILE_TAIL);
还有一个小技巧,在运行server时实时查看日志文件。可以通过tail -f 2022_05_13_log
查看日志文件,有日志写入时都能实时看到。