redisclientprotocol实现

2015-07-24 07:47:06 · 作者: · 浏览: 0

?

下面是解析RESP的所有函数,其中对外函数是RedisProtocol::Decode:

https://github.com/XJM2013/GameEngine/blob/master/lib/src/redis/redisprotocol.cpp

?

static unsigned int	EndLine(const char *buf, unsigned int len);
static int			ReadNumber(const char *buf, unsigned int len);
static char			_ReadString(const char *buf, unsigned int left_len, int &read_len, RedisData **data);
static int			ReadMessage(char type, const char *buf, unsigned int len, RedisBulkData **bulk_data);
static int			ReadInteger(const char *buf, unsigned int len, RedisBulkData **bulk_data);
static int			ReadString(const char *buf, unsigned int len, RedisBulkData **bulk_data);
static int			ReadArray(const char *buf, unsigned int left_len, RedisBulkData **bulk_data);

/*
	:1\r\n	前缀 + 结果 + 结束 最少也要有4个字节
*/
int RedisProtocol::Decode(const char *buf, unsigned int len, RedisBulkData **bulk_data)
{
	if (len < 4)
	{
		return OPR_MORE_DATA;
	}

	switch (buf[0])
	{
	case '+':
		return ReadMessage(REPLY_TYPE_OK, buf + 1, len - 1, bulk_data);
	case '-':
		return ReadMessage(REPLY_TYPE_ERROR, buf + 1, len - 1, bulk_data);
	case ':':
		return ReadInteger(buf + 1, len - 1, bulk_data);
	case '$':
		return ReadString(buf + 1, len - 1, bulk_data);
	case '*':
		return ReadArray(buf + 1, len - 1, bulk_data);
	default:
		return OPR_DATA_INVALID;
	}

	return OPR_DATA_INVALID;
}
从上面的代码来看,可以看出,代码的结构相当清晰、简洁;而且这代码的功能只是对协议的处理,不包含网络模块和其它更复杂的封装等处理,不需要做额外的辨别 阅读。对于学习协议者,个人觉得还是一个不错的选择。代码中我有做一些内存池的功能,不想使用的,可以用new/delete 或者 malloc/free代替。
下面我基于前一篇文章说的5个例子,分别作为测试例子:

https://github.com/XJM2013/GameEngine/blob/master/Test/testredis.h

?

#ifndef TEST_REDIS_H
#define TEST_REDIS_H

#include 
  
   
#include "lib/include/redis/redisprotocol.h"

namespace TestRedis
{
	void ShowBulkData(RedisBulkData *data)
	{
		
		std::list
   
    ::iterator itr = data->data_list.begin(); for (; itr != data->data_list.end(); ++itr) { printf("type = %d\n", (*itr)->type); switch ((*itr)->type) { case RedisProtocol::REPLY_TYPE_INTEGER: case RedisProtocol::REPLY_TYPE_STRING_ERROR: case RedisProtocol::REPLY_TYPE_ARRAY_ERROR: printf("data = %d\n", *(int *)(*itr)->data); break; default: printf("data = %.*s\n", (*itr)->len, (*itr)->data); break; } } } void Decode(char *reply) { RedisBulkData *data = NULL; if (RedisProtocol::Decode(reply, strlen(reply), &data) <= RedisProtocol::OPR_MORE_DATA) { return; } ShowBulkData(data); delete data; } // 短字符串 void Test1() { char *reply = "+OK\r\n"; Decode(reply); } // 错误 void Test2() { char *reply = "-ERR unknown command 'seet'\r\n"; Decode(reply); } // 整数 void Test3() { char *reply = ":11\r\n"; Decode(reply); } // 长字符串 void Test4() { char *reply1 = "$3\r\ncat\r\n"; printf("reply1:\n"); Decode(reply1); char *reply2 = "$-1\r\n"; printf("reply2:\n"); Decode(reply2); } // 数组 void Test5() { char *reply1 = "*2\r\n$3\r\ncat\r\n$2\r\n11\r\n"; printf("reply1:\n"); Decode(reply1); char *reply2 = "*2\r\n$4\r\nfish\r\n$-1\r\n"; printf("reply2:\n"); Decode(reply2); char *reply3 = "*-1\r\n"; printf("reply3:\n"); Decode(reply3); } } #endif
   
  
下面输出一下Test5()的结果:

?

\

协议解析总结:

1、从C/C++的角度来看,序列化保存数据,解析更快。

例如将1000条数据序列化成1条数据:

从分配空间的角度来看,1000条数据需要分配1000次空间,1条数据需要分配1次空间。

从解析的角度来看,RESP的结构是LV结构,也就是长度-值(length-value)结构,其不需要标志类型(T,type),因为它的类型都是字符串类型。而序列化实现可以是TV和TLV结构的结合。因此C/C++里面类型与长度是一一对应的,整型就是4个字节等等;对于字符串则使用TLV结构。这里大致只需要将RESP的L与序列化的T做一个对比。一个需要匹配"\r\n",并将字符串转化成数字,一个则只需要读取第一字节则可以判断数据长度。

2、set key val\r\n,当val很长,例如10k字节长,server 要匹配10k次才能匹配到\r\n,而且val中不允许出现\r\n否则会出错。