pika_monitor_thread.h

// Copyright (c) 2015-present, Qihoo, Inc.  All rights reserved.
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree. An additional grant
// of patent rights can be found in the PATENTS file in the same directory.

#ifndef  PIKA_MONITOR_THREAD_H_
#define  PIKA_MONITOR_THREAD_H_
#include <map>
#include <set>
#include <atomic>
#include <list>
#include <deque>
#include <queue>

#include "pink_thread.h"
#include "pink_epoll.h"
#include "pika_client_conn.h"
#include "slash_mutex.h"
#include "pika_define.h"

class PikaMonitorThread : public pink::Thread {
public:
  PikaMonitorThread();
  virtual ~PikaMonitorThread();

  void AddMonitorClient(pink::RedisConn* client_ptr);
  void AddMonitorMessage(const std::string &monitor_message);
  int32_t ThreadClientList(std::vector<ClientInfo>* client = NULL);
  bool ThreadClientKill(const std::string& ip_port = "all");
  bool HasMonitorClients();

private:
  void AddCronTask(MonitorCronTask task);
  bool FindClient(const std::string& ip_port);
  pink::WriteStatus SendMessage(int32_t fd, std::string& message);
  void RemoveMonitorClient(const std::string& ip_port);

  slash::Mutex monitor_mutex_protector_;
  slash::CondVar monitor_cond_;

  std::list<ClientInfo> monitor_clients_;
  std::deque<std::string> monitor_messages_;
  std::atomic<bool> is_running_;
  std::atomic<bool> should_exit_;
  std::queue<MonitorCronTask> cron_tasks_;

  virtual void* ThreadMain();
  void RemoveMonitorClient(int32_t client_fd);
  void StartThread();
};
#endif