pika_dispatch_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_DISPATCH_THREAD_H_
#define PIKA_DISPATCH_THREAD_H_

#include "slash/include/env.h"
#include "pink/include/server_thread.h"
#include "include/pika_client_conn.h"

class PikaDispatchThread {
 public:
  PikaDispatchThread(std::set<std::string> &ips, int port, int work_num,
                     int cron_interval, int queue_limit);
  ~PikaDispatchThread();
  int StartThread();

  int64_t ThreadClientList(std::vector<ClientInfo> *clients);

  bool ClientKill(const std::string& ip_port);
  void ClientKillAll();

  void SetQueueLimit(int queue_limit) {
    thread_rep_->SetQueueLimit(queue_limit);
  }

 private:
  class ClientConnFactory : public pink::ConnFactory {
   public:
    virtual pink::PinkConn *NewPinkConn(
        int connfd,
        const std::string &ip_port,
        pink::ServerThread *server_thread,
        void* worker_specific_data) const {
      return new PikaClientConn(connfd, ip_port, server_thread, worker_specific_data);
    }
  };

  class Handles : public pink::ServerHandle {
   public:
    explicit Handles(PikaDispatchThread* pika_disptcher)
        : pika_disptcher_(pika_disptcher) {
    }
    using pink::ServerHandle::AccessHandle;
    bool AccessHandle(std::string& ip) const override;
    void CronHandle() const override;
    int CreateWorkerSpecificData(void** data) const override;
    int DeleteWorkerSpecificData(void* data) const override;

   private:
    PikaDispatchThread* pika_disptcher_;
  };

  ClientConnFactory conn_factory_;
  Handles handles_;
  pink::ServerThread* thread_rep_;
};
#endif