Расширяем функционал key-value хранилища Redis

После некоторого изучения хранилища Redis (версия 1.01) для использования в высоконагруженном проекте впечатления остаются хорошие. Но лично мне не хватило одной простой команды — подсчет количества ключей по паттерну. Т.е. есть KEYS, но она возвращает массив со всеми ключами. Что, согласитесь, с размерами порядка сотни миллионов записей заставит задуматься сервер на долго. Если у него хватит ресурсов на это.

Немного покопавшись в исходниках, была введена новая команда COUNT , возвращающая количество записей по паттерну.

diff redis.c
352a353
> static void countCommand(redisClient *c);
441a443
> {"count",countCommand,2,REDIS_CMD_INLINE},
755c757
< if (!(loops % 5)) {
---
> if (!(loops % 30)) {
2502a2505,2528
> static void countCommand(redisClient *c) {
> dictIterator *di;
> dictEntry *de;
> sds pattern = c->argv[1]->ptr;
> int plen = sdslen(pattern);
> int numkeys = 0;
>
> di = dictGetIterator(c->db->dict);
> if (!di) oom("dictGetIterator");
> while((de = dictNext(di)) != NULL) {
> robj *keyobj = dictGetEntryKey(de);
>
> sds key = keyobj->ptr;
> if ((pattern[0] == '*' && pattern[1] == '\0') ||
> stringmatchlen(pattern,plen,key,sdslen(key),0)) {
> if (expireIfNeeded(c->db,keyobj) == 0) {
> numkeys++;
> }
> }
> }
> dictReleaseIterator(di);
> addReplySds(c, sdscatprintf(sdsempty(),":%lu\r\n", numkeys));
> }
>


diff redis-cli.c

98a99
> {"count",2,REDIS_CMD_INLINE},


diff redis.py

301a302,306
> def count(self, pattern):
> self.connect()
> self._write('COUNT %s\r\n' % pattern)
> return self.get_response()
>

Enjoy! ;-)