rpc.h

Go to the documentation of this file.

// Copyright (C) 2013-2020 iFunFactory Inc. All Rights Reserved.
//
// This work is confidential and proprietary to iFunFactory Inc. and
// must not be used, disclosed, copied, or distributed without the prior
// consent of iFunFactory Inc.

#ifndef INCLUDE_FUNAPI_DISTRIBUTION_RPC_H_
#define INCLUDE_FUNAPI_DISTRIBUTION_RPC_H_

#include <boost/asio.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/function.hpp>
#include <boost/noncopyable.hpp>
#include <funapi/distribution/fun_rpc_message.pb.h>
#include <funapi/system/hardware_info.h>
#include <funapi/types.h>

#include <map>
#include <set>


namespace fun {

class FUNAPI_DLL_VISIBILITY Rpc : public boost::enable_shared_from_this<Rpc>,
                                  private boost::noncopyable {
 public:
  typedef Uuid Xid;
  typedef Uuid PeerId;
  typedef Uuid SerializationTag;
  typedef boost::asio::ip::tcp::endpoint PeerEndpoint;
  typedef std::map<PeerId, PeerEndpoint> PeerMap;
  typedef string Tag;
  typedef std::set<Tag> Tags;

  typedef boost::function<void(const Ptr<FunRpcMessage> & /*reply*/)> ReadyBack ASSERT_NO_ROLLBACK;

  typedef boost::function<void(const PeerId & /*new server*/)> ConnectHandler;

  typedef boost::function<void(const PeerId & /*server*/)> DisconnectHandler;

  typedef boost::function<
      void(const PeerId & /*sender*/, const Xid & /*xid*/,
           const Ptr<const FunRpcMessage> & /*request*/,
           const ReadyBack & /*finisher*/)> Handler;

  typedef boost::function<
      void(const PeerId & /*sender*/, const Xid & /*xid*/,
           const Ptr<const FunRpcMessage> & /*request*/)> VoidReplyHandler;

  typedef boost::function<
      void(const PeerId & /*sender*/, const Xid & /*xid*/,
           const Ptr<const FunRpcMessage> & /*reply*/)> Callback;

  static const Xid kNullXid;
  static const PeerId kNullPeerId;
  static const SerializationTag kDefaultSerializationTag;

  static void AddTag(const Tag &tag);

  static void AddTags(const Tags &tags);

  static void RemoveTag(const Tag &tag);

  static void RemoveTags(const Tags &tags);

  static bool HasTag(const Tag &tag);

  static Json GetStatus();

  static void SetStatus(const Json &status);

  static PeerId GetSelfId();

  static PeerEndpoint GetSelfEndpoint();

  static size_t GetPeers(PeerMap *ret, bool include_self = false);

  static size_t GetPeersWithTag(PeerMap *ret, const Tag &tag,
                                bool include_self = false);

  static size_t GetPeerTags(Tags *ret, const PeerId &peer);

  static std::string GetPeerExternalHostname(const PeerId &peer);

  static boost::asio::ip::address GetPeerExternalIp(const PeerId &peer);

  static HardwareInfo::ProtocolPortMap GetPeerExternalPorts(const PeerId &peer);

  static bool IsExternalPeer(const PeerId &peer);

  static Json GetPeerStatus(const PeerId &peer);

  static bool IsPeerConnected(const PeerId &peer);

  static void RegisterConnectHandler(const ConnectHandler &h);

  static void RegisterDisconnectHandler(const DisconnectHandler &h);

  static void RegisterHandler(const string &type, const Handler &h);

  static void Call(
      const PeerId &peer,
      const Ptr<FunRpcMessage> &msg,
      const Callback &callback,
      const SerializationTag &tag = kDefaultSerializationTag)
          ASSERT_NO_ROLLBACK;

  static void RegisterVoidReplyHandler(const string &type,
                                       const VoidReplyHandler &h);

  static void Call(
      const PeerId &peer,
      const Ptr<FunRpcMessage> &msg,
      const SerializationTag &tag = kDefaultSerializationTag)
          ASSERT_NO_ROLLBACK;
};

}  // namespace fun

#endif  // INCLUDE_FUNAPI_DISTRIBUTION_RPC_H_