分析了一段时间的struct结构体的redis代码,越到最后越发现其实很多代码都是大同小异的。在struct包中还有1,2个文件还没分析,是关于set集合的一些东西,就放在下次分析好了,在选择下个分析的对象时,我考虑了一下,最后决定先把简单的test包下的东西看看一下,毕竟结构体这块有点复杂,所以这次分析个简单点的。test包下的文件并不多,代码量也很少,总共5个文件:

1.memtest.c 内存检测
2.redis_benchmark.c 用于redis性能测试的实现。
3.redis_check_aof.c 用于更新日志检查的实现。
4.redis_check_dump.c 用于本地数据库检查的实现。
5.testhelp.c 一个C风格的小型测试框架。

今天讨论3和5,先看看5,testhelp.c被解释为一个C风格的小型测试框架,看见这标题,一定想到这是一个很大的文件吧,都被称为测试框架了,其实不然,我贴出整个代码:


/* 预处理 */
#ifndef __TESTHELP_H
#define __TESTHELP_H

/* 开始时失败总数为0 */
int __failed_tests = 0;
/* 开始时测试总的例子数为0 */
int __test_num = 0;

/* 宏定义测试方法,输入参数,输入描述语,判断的式子作为参数 */
/* 有完全体现了函数式编程的思想 */
#define test_cond(descr,_c) do { \
    __test_num++; printf("%d - %s: ", __test_num, descr); \
    //判断式子在此处调用
    if(_c) printf("PASSED\n"); else {printf("FAILED\n"); __failed_tests++;} \
} while(0);

/* 测试报告在最结尾输出 */
#define test_report() do { \
    printf("%d tests, %d passed, %d failed\n", __test_num, \
                    __test_num-__failed_tests, __failed_tests); \
    if (__failed_tests) { \
        printf("=== WARNING === We have failed tests here...\n"); \
        exit(1); \
    } \
} while(0);

#endif



调用形式也是极其简单:


* test_cond("Check if 1 == 1", 1==1)
 * test_cond("Check if 5 > 10", 5 > 10)
 * test_report()



其实就是宏定义了一个判别表达式真假的方法,又用到了函数式编程的思想,把_c整个表达式的值传入了另一个函数中,就这么简单。好,下一个文件

redis_check_aof.c用于日志检测的,必然和文件的操作相关的,同样列出里面的API:

/* 方法API */
int consumeNewline(char *buf) /* 消除buf前面的换行符,即比较buf字符串中的前2个字符 */
int readLong(FILE *fp, char prefix, long *target) /* 从文件中读取long类型值 */
int readBytes(FILE *fp, char *target, long length) /* 从文件中读取字节 */
int readString(FILE *fp, char** target) /* 文件中读取字符串 */
int readArgc(FILE *fp, long *target) /* 文件中读取参数,首字符以“*”开头 */
off_t process(FILE *fp) /* 返回fp文件的偏移量 */



read的很多操作其实都是非常类似的,我就举出其中的一个read方法当做例子,就是纯粹文件的简单操作,熟悉C语言的同学一定很熟悉:

/* 从文件中读取long类型值 */
int readLong(FILE *fp, char prefix, long *target) {
    char buf[128], *eptr;
    //定位到文件的读取位置
    epos = ftello(fp);
    //将文件中的内容读取到buf中
    if (fgets(buf,sizeof(buf),fp) == NULL) {
    	//如果为空直接返回
        return 0;
    }
    
    //如果读取到的首字符不等于预期值,则提示报错
    if (buf[0] != prefix) {
        ERROR("Expected prefix '%c', got: '%c'",buf[0],prefix);
        return 0;
    }
    
    //将字符串值转成long类型值
    *target = strtol(buf+1,&eptr,10);
    
    return consumeNewline(eptr);
}



里面有一个比较特别的方法,consumeNewline()消除换行符的方法:

/* 消除buf前面的换行符,即比较buf字符串中的前2个字符 */
int consumeNewline(char *buf) {
    if (strncmp(buf,"\r\n",2) != 0) {
    	//如果不是等于"\r\n",则提示出错
        ERROR("Expected \\r\\n, got: %02x%02x",buf[0],buf[1]);
        return 0;
    }
    return 1;
}



里面比较复杂的方法是off_t process(FILE *fp)获取文件偏移量的操作,这个方法是用来后面截断文件的操作,截断后面的空文件的部分:

off_t pos = process(fp);

//截断文件的操作,从问价头部到后面的偏移量,没有用的空间截去
            if (ftruncate(fileno(fp), pos) == -1) {
                printf("Failed to truncate AOF\n");
                exit(1);
            } else {
                printf("Successfully truncated AOF\n");
            }



以上就是我所分析的内容了,非常简单,比起struct比,简单太多了,可以细细体会其中方法涉及的技巧,比如我们会考虑日志文件分析时会想到

用 consumeNewline()换行操作的设计吗