From d5c3390d32b6126a1192a1858710bbfb960afdd0 Mon Sep 17 00:00:00 2001
From: zd-double <adikingustc@gmail.com>
Date: Fri, 20 Jan 2017 15:02:12 +0800
Subject: [PATCH] Add create_with_init option in RpcChannelOptions

---
 src/sofa/pbrpc/rpc_channel.cc | 25 +++++++++++++++++++++----
 src/sofa/pbrpc/rpc_channel.h  | 13 +++++++++++++
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/src/sofa/pbrpc/rpc_channel.cc b/src/sofa/pbrpc/rpc_channel.cc
index 1086ef8..95e35f1 100644
--- a/src/sofa/pbrpc/rpc_channel.cc
+++ b/src/sofa/pbrpc/rpc_channel.cc
@@ -16,7 +16,10 @@ RpcChannel::RpcChannel(RpcClient* rpc_client,
         const RpcChannelOptions& options)
     : _impl(new SimpleRpcChannelImpl(rpc_client->impl(), server_address, options))
 {
-    _impl->Init();
+    if (options.create_with_init)
+    {
+        _impl->Init();
+    }
 }
 
 RpcChannel::RpcChannel(RpcClient* rpc_client,
@@ -27,7 +30,10 @@ RpcChannel::RpcChannel(RpcClient* rpc_client,
     std::ostringstream os;
     os << server_ip << ":" << server_port;
     _impl.reset(new SimpleRpcChannelImpl(rpc_client->impl(), os.str(), options));
-    _impl->Init();
+    if (options.create_with_init)
+    {
+        _impl->Init();
+    }
 }
 
 RpcChannel::RpcChannel(RpcClient* rpc_client,
@@ -35,7 +41,10 @@ RpcChannel::RpcChannel(RpcClient* rpc_client,
         const RpcChannelOptions& options)
     : _impl(new DynamicRpcChannelImpl(rpc_client->impl(), address_list, options))
 {
-    _impl->Init();
+    if (options.create_with_init)
+    {
+        _impl->Init();
+    }
 }
 
 RpcChannel::RpcChannel(RpcClient* rpc_client,
@@ -43,7 +52,15 @@ RpcChannel::RpcChannel(RpcClient* rpc_client,
         const RpcChannelOptions& options)
     : _impl(new DynamicRpcChannelImpl(rpc_client->impl(), address_provider, options))
 {
-    _impl->Init();
+    if (options.create_with_init)
+    {
+        _impl->Init();
+    }
+}
+
+bool RpcChannel::Init()
+{
+    return _impl->Init();
 }
 
 RpcChannel::~RpcChannel()
diff --git a/src/sofa/pbrpc/rpc_channel.h b/src/sofa/pbrpc/rpc_channel.h
index b5efa34..dbe46c8 100644
--- a/src/sofa/pbrpc/rpc_channel.h
+++ b/src/sofa/pbrpc/rpc_channel.h
@@ -32,9 +32,14 @@ struct RpcChannelOptions {
     // Value 0 means no limit, default value is 0.
     uint32 server_load_capacity;
 
+    // If initialize the RpcChannel in construct function, default is true.
+    // If create_with_init is false, RpcChannel should be initialized by calling Init().
+    bool create_with_init;
+
     RpcChannelOptions()
         : connect_timeout(10)
         , server_load_capacity(0)
+        , create_with_init(true)
     {}
 };
 
@@ -93,6 +98,14 @@ class RpcChannel : public google::protobuf::RpcChannel
                AddressProvider* address_provider,
                const RpcChannelOptions& options = RpcChannelOptions());
 
+    // Initialize RpcChannel.
+    // For single server point, it will resolve server address in this function,
+    // and if resolve server address succeed return true, otherwise return false.
+    // For multiple server points, it will update internal server list and
+    // register detect task. After all of these are completed return true, and
+    // never return false.
+    bool Init();
+
     // Destructor.
     virtual ~RpcChannel();