redis_connection.hpp

// The MIT License (MIT)
//
// Copyright (c) 2015-2017 Simon Ninon <simon.ninon@gmail.com>
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

#pragma once

#include <functional>
#include <memory>
#include <mutex>
#include <string>
#include <vector>

#include <cpp_redis/builders/reply_builder.hpp>
#include <cpp_redis/network/tcp_client_iface.hpp>

#ifndef __CPP_REDIS_READ_SIZE
#define __CPP_REDIS_READ_SIZE 4096
#endif /* __CPP_REDIS_READ_SIZE */

namespace cpp_redis {

namespace network {

class redis_connection {
public:
#ifndef __CPP_REDIS_USE_CUSTOM_TCP_CLIENT
  redis_connection(void);
#endif /* __CPP_REDIS_USE_CUSTOM_TCP_CLIENT */

  explicit redis_connection(const std::shared_ptr<tcp_client_iface>& tcp_client);

  ~redis_connection(void);

  redis_connection(const redis_connection&) = delete;
  redis_connection& operator=(const redis_connection&) = delete;

public:
  typedef std::function<void(redis_connection&)> disconnection_handler_t;

  typedef std::function<void(redis_connection&, reply&)> reply_callback_t;

  void connect(
    const std::string& host                              = "127.0.0.1",
    std::size_t port                                     = 6379,
    const disconnection_handler_t& disconnection_handler = nullptr,
    const reply_callback_t& reply_callback               = nullptr,
    std::uint32_t timeout_msecs                          = 0);

  void disconnect(bool wait_for_removal = false);

  bool is_connected(void) const;

  redis_connection& send(const std::vector<std::string>& redis_cmd);

  redis_connection& commit(void);

private:
  void tcp_client_receive_handler(const tcp_client_iface::read_result& result);

  void tcp_client_disconnection_handler(void);

  std::string build_command(const std::vector<std::string>& redis_cmd);

private:
  void call_disconnection_handler(void);

private:
  std::shared_ptr<cpp_redis::network::tcp_client_iface> m_client;

  reply_callback_t m_reply_callback;

  disconnection_handler_t m_disconnection_handler;

  builders::reply_builder m_builder;

  std::string m_buffer;

  std::mutex m_buffer_mutex;
};

} // namespace network

} // namespace cpp_redis