predixy/src/Subscribe.cpp

/*
 * predixy - A high performance and full features proxy for redis.
 * Copyright (C) 2017 Joyield, Inc. <joyield.com@gmail.com>
 * All rights reserved.
 */

#include "Subscribe.h"

Subscribe::Subscribe():
    mPendSub(0),
    mSub(0)
{
}



SubscribeParser::Status SubscribeParser::parse(const Segment& body, int& chs)
{
    SegmentStr<128> str(body);
    Status st = Unknown;
    chs = 0;
    if (str.hasPrefix("*3\r\n$7\r\nmessage\r\n")) {
        st = Message;
    } else if (str.hasPrefix("*4\r\n$8\r\npmessage\r\n")) {
        st = Pmessage;
    } else if (str.hasPrefix("*3\r\n$9\r\nsubscribe\r\n")) {
        st = Subscribe;
        chs = -1;
    } else if (str.hasPrefix("*3\r\n$10\r\npsubscribe\r\n")) {
        st = Psubscribe;
        chs = -1;
    } else if (str.hasPrefix("*3\r\n$11\r\nunsubscribe\r\n")) {
        st = Unsubscribe;
        chs = -1;
    } else if (str.hasPrefix("*3\r\n$12\r\npunsubscribe\r\n")) {
        st = Punsubscribe;
        chs = -1;
    } else if (str.hasPrefix("-")) {
        st = Error;
    } else if (str.hasPrefix("$")) {
        st = String;
    }
    if (chs < 0) {
        if (!str.complete()) {
            Segment tmp(body);
            tmp.rewind();
            tmp.cut(body.length() - 12);
            str.set(tmp);
        }
        const char* p = str.data() + str.length();
        for (int i = 0; i < str.length(); ++i) {
            if (*--p == ':') {
                chs = atoi(p + 1);
                break;
            }
        }
    }
    return st;
}