[minor](rpc) Check client before RPC (apache#43626)

Check client before RPC
py023 · Nov 13, 2024 · 97a5fda · 97a5fda
1 parent 4974a8e
commit 97a5fda
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/be/src/runtime/client_cache.h b/be/src/runtime/client_cache.h
@@ -35,6 +35,7 @@
 #include <unordered_map>
 
 #include "common/config.h"
+#include "common/exception.h"
 #include "common/status.h"
 #include "util/hash_util.hpp"
 #include "util/metrics.h"
@@ -189,7 +190,12 @@ class ClientConnection {
 
     inline bool is_alive() { return _client != nullptr; }
 
-    T* operator->() const { return _client; }
+    T* operator->() const {
+        if (_client == nullptr) {
+            throw doris::Exception(ErrorCode::INTERNAL_ERROR, "Invalid RPC client!");
+        }
+        return _client;
+    }
 
 private:
     ClientCache<T>* _client_cache = nullptr;