connection.h

// Copyright (c) 2014 Luca Marturana. All rights reserved.
// Licensed under Apache 2.0, see LICENSE for details

#pragma once

#include <string>
#include <redis3m/utils/exception.h>
#include <redis3m/reply.h>
#include <vector>
#include <boost/shared_ptr.hpp>
#include <boost/noncopyable.hpp>
#include <boost/assign/list_of.hpp>

struct redisContext;

namespace redis3m {
    REDIS3M_EXCEPTION(unable_to_connect)
    REDIS3M_EXCEPTION(transport_failure)

    // Utility to build commands, an alias of boost::assign::list_of<std::string>
    // increases readability of code
    extern boost::assign_detail::generic_list<std::string>(&command)(const std::string&);

    class connection: boost::noncopyable
    {
    public:
        typedef boost::shared_ptr<connection> ptr_t;

        inline static ptr_t create(const std::string& host="localhost",
                                   const unsigned int port=6379)
        {
            return ptr_t(new connection(host, port));
        }

        ~connection();

        bool is_valid();
        
        void append(const std::vector<std::string>& args);

        reply get_reply();
        
        std::vector<reply> get_replies(unsigned int count);

        inline reply run(const std::vector<std::string>& args)
        {
            append(args);
            return get_reply();
        }

        inline redisContext* c_ptr() { return c; }
        
        enum role_t {
            ANY = 0,
            MASTER = 1,
            SLAVE = 2
        };
        
    private:
        friend class connection_pool;
        connection(const std::string& host, const unsigned int port);

        role_t _role;
        redisContext *c;
    };
}