RedisInternal.h

Go to the documentation of this file.

#ifndef REDIS_INTERNAL_H
#define REDIS_INTERNAL_H
 
#include "Arduino.h"
#include "Client.h"
#include <vector>
#include <memory>
#include <functional>
 
#define CRLF F("\r\n")
 
typedef std::vector<String> ArgList;
 
/* The lack of RTTI on Ardunio is unfortunate but completely understandable.
 * However, we're not going to let that stop us, so RedisObject implements a basic
 * but functional type system for Redis objects.
 */
class RedisObject
{
public:
    typedef enum
    {
        NoType = '\0',
        SimpleString = '+',
        Error = '-',
        Integer = ':',
        BulkString = '$',
        Array = '*',
        InternalError = '!'
    } Type;
 
    RedisObject() {}
    RedisObject(Type tc) : _type(tc) {}
    RedisObject(Type tc, Client &c) : _type(tc) { init(c); }
 
    virtual ~RedisObject() {}
 
    static std::shared_ptr<RedisObject> parseTypeNonBlocking(Client &);
    static std::shared_ptr<RedisObject> parseType(Client &);
 
    virtual void init(Client &client);
 
    virtual String RESP() = 0;
 
    virtual operator String() { if(data) return data; return String("(nil)");}
 
    Type type() const { return _type; }
 
protected:
    String data;
    Type _type = Type::NoType;
};
 
class RedisSimpleString : public RedisObject
{
public:
    RedisSimpleString(Client &c) : RedisObject(Type::SimpleString, c) {}
    ~RedisSimpleString() override {}
 
    virtual String RESP() override;
};
 
class RedisBulkString : public RedisObject
{
public:
    RedisBulkString(Client &c) : RedisObject(Type::BulkString, c) { init(c); }
    RedisBulkString(String &s) : RedisObject(Type::BulkString) { data = s; }
    ~RedisBulkString() override {}
 
    virtual void init(Client &client) override;
 
    virtual String RESP() override;
};
 
class RedisArray : public RedisObject
{
public:
    RedisArray() : RedisObject(Type::Array) {}
    RedisArray(Client &c) : RedisObject(Type::Array, c) { init(c); }
    ~RedisArray() override { vec.empty(); }
 
    void add(std::shared_ptr<RedisObject> param) { vec.push_back(param); }
 
    operator std::vector<String>() const;
 
    operator std::vector<std::shared_ptr<RedisObject>>() const;
 
    virtual void init(Client &client) override;
 
    virtual String RESP() override;
 
protected:
    std::vector<std::shared_ptr<RedisObject>> vec;
};
 
class RedisInteger : public RedisSimpleString
{
public:
    RedisInteger(Client &c) : RedisSimpleString(c) { _type = Type::Integer; }
    ~RedisInteger() override {}
 
    operator int() { return data.toInt(); }
    operator bool() { return (bool)operator int(); }
};
 
class RedisError : public RedisSimpleString
{
public:
    RedisError(Client &c) : RedisSimpleString(c) { _type = Type::Error; }
    ~RedisError() override {}
};
 
class RedisInternalError : public RedisObject
{
public:
    typedef enum
    {
        UnknownError = -254,
        UnknownType,
        Disconnected,
        NoError = 0
    } RedisInternalErrorCode;
 
    RedisInternalError(RedisInternalErrorCode c) : RedisObject(Type::InternalError), _code(c) {}
    RedisInternalError(RedisInternalErrorCode c, String es) : RedisInternalError(c) { data = es; }
    ~RedisInternalError() override {}
 
    RedisInternalErrorCode code() { return _code; }
    void setErrorString(String es) { data = es; }
 
    virtual String RESP() override { return "-INTERNAL ERROR " + String(_code) + (data ? ": " + data : ""); }
 
protected:
    RedisInternalErrorCode _code;
};
 
class RedisCommand : public RedisArray
{
public:
    RedisCommand(String command) : RedisArray()
    {
        add(std::shared_ptr<RedisObject>(new RedisBulkString(command)));
    }
 
    RedisCommand(String command, ArgList args)
        : RedisCommand(command)
    {
        for (auto arg : args)
        {
            add(std::shared_ptr<RedisObject>(new RedisBulkString(arg)));
        }
    }
 
    ~RedisCommand() override {}
 
    std::shared_ptr<RedisObject> issue(Client &cmdClient);
 
    template <typename T>
    T issue_typed(Client &cmdClient);
 
private:
    String _err;
};
 
#endif // REDIS_INTERNAL_H