114 lines
4.3 KiB
C++
114 lines
4.3 KiB
C++
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
#include "util/thrift_rpc_helper.h"
|
|
|
|
#include <gen_cpp/Types_types.h>
|
|
#include <glog/logging.h>
|
|
#include <thrift/Thrift.h>
|
|
#include <thrift/protocol/TBinaryProtocol.h>
|
|
#include <thrift/transport/TTransportException.h>
|
|
// IWYU pragma: no_include <bits/chrono.h>
|
|
#include <chrono> // IWYU pragma: keep
|
|
#include <sstream>
|
|
#include <thread>
|
|
|
|
#include "common/status.h"
|
|
#include "runtime/client_cache.h"
|
|
#include "runtime/exec_env.h" // IWYU pragma: keep
|
|
#include "util/network_util.h"
|
|
|
|
namespace apache {
|
|
namespace thrift {
|
|
namespace protocol {
|
|
class TProtocol;
|
|
} // namespace protocol
|
|
namespace transport {
|
|
class TBufferedTransport;
|
|
class TSocket;
|
|
class TTransport;
|
|
} // namespace transport
|
|
} // namespace thrift
|
|
} // namespace apache
|
|
|
|
namespace doris {
|
|
|
|
using apache::thrift::protocol::TProtocol;
|
|
using apache::thrift::protocol::TBinaryProtocol;
|
|
using apache::thrift::transport::TSocket;
|
|
using apache::thrift::transport::TTransport;
|
|
using apache::thrift::transport::TBufferedTransport;
|
|
|
|
ExecEnv* ThriftRpcHelper::_s_exec_env;
|
|
|
|
void ThriftRpcHelper::setup(ExecEnv* exec_env) {
|
|
_s_exec_env = exec_env;
|
|
}
|
|
|
|
template <typename T>
|
|
Status ThriftRpcHelper::rpc(const std::string& ip, const int32_t port,
|
|
std::function<void(ClientConnection<T>&)> callback, int timeout_ms) {
|
|
TNetworkAddress address = make_network_address(ip, port);
|
|
Status status;
|
|
ClientConnection<T> client(_s_exec_env->get_client_cache<T>(), address, timeout_ms, &status);
|
|
if (!status.ok()) {
|
|
LOG(WARNING) << "Connect frontend failed, address=" << address << ", status=" << status;
|
|
return status;
|
|
}
|
|
try {
|
|
try {
|
|
callback(client);
|
|
} catch (apache::thrift::transport::TTransportException& e) {
|
|
LOG(WARNING) << "retrying call frontend service after "
|
|
<< config::thrift_client_retry_interval_ms << " ms, address=" << address
|
|
<< ", reason=" << e.what();
|
|
std::this_thread::sleep_for(
|
|
std::chrono::milliseconds(config::thrift_client_retry_interval_ms));
|
|
status = client.reopen(timeout_ms);
|
|
if (!status.ok()) {
|
|
LOG(WARNING) << "client reopen failed. address=" << address
|
|
<< ", status=" << status;
|
|
return status;
|
|
}
|
|
callback(client);
|
|
}
|
|
} catch (apache::thrift::TException& e) {
|
|
LOG(WARNING) << "call frontend service failed, address=" << address
|
|
<< ", reason=" << e.what();
|
|
std::this_thread::sleep_for(
|
|
std::chrono::milliseconds(config::thrift_client_retry_interval_ms * 2));
|
|
// just reopen to disable this connection
|
|
client.reopen(timeout_ms);
|
|
return Status::RpcError("failed to call frontend service, reason: {}", e.what());
|
|
}
|
|
return Status::OK();
|
|
}
|
|
|
|
template Status ThriftRpcHelper::rpc<FrontendServiceClient>(
|
|
const std::string& ip, const int32_t port,
|
|
std::function<void(ClientConnection<FrontendServiceClient>&)> callback, int timeout_ms);
|
|
|
|
template Status ThriftRpcHelper::rpc<BackendServiceClient>(
|
|
const std::string& ip, const int32_t port,
|
|
std::function<void(ClientConnection<BackendServiceClient>&)> callback, int timeout_ms);
|
|
|
|
template Status ThriftRpcHelper::rpc<TPaloBrokerServiceClient>(
|
|
const std::string& ip, const int32_t port,
|
|
std::function<void(ClientConnection<TPaloBrokerServiceClient>&)> callback, int timeout_ms);
|
|
|
|
} // namespace doris
|