zyearn commented on a change in pull request #694: Health check by rpc call URL: https://github.com/apache/incubator-brpc/pull/694#discussion_r269896275
########## File path: src/brpc/details/health_check.cpp ########## @@ -0,0 +1,246 @@ +// Copyright (c) 2014 Baidu, Inc. +// +// Licensed 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. + +// Authors: Ge,Jun (ge...@baidu.com) +// Jiashun Zhu(zhujias...@baidu.com) + +#include "brpc/details/health_check.h" +#include "brpc/socket.h" +#include "brpc/channel.h" +#include "brpc/controller.h" +#include "brpc/details/controller_private_accessor.h" +#include "brpc/global.h" +#include "brpc/log.h" +#include "bthread/unstable.h" +#include "bthread/bthread.h" + +namespace brpc { + +// declared at socket.cpp +extern SocketVarsCollector* g_vars; + +DEFINE_string(health_check_path, "", "Http path of health check call." + "By default health check succeeds if the server is connectable." + "If this flag is set, health check is not completed until a http " + "call to the path succeeds within -health_check_timeout_ms(to make " + "sure the server functions well)."); +DEFINE_int32(health_check_timeout_ms, 500, "The timeout for both establishing " + "the connection and the http call to -health_check_path over the connection"); + +class HealthCheckChannel : public brpc::Channel { +public: + HealthCheckChannel() {} + ~HealthCheckChannel() {} + + int Init(SocketId id, const ChannelOptions* options); +}; + +int HealthCheckChannel::Init(SocketId id, const ChannelOptions* options) { + brpc::GlobalInitializeOrDie(); + if (InitChannelOptions(options) != 0) { + return -1; + } + _server_id = id; + return 0; +} + +class OnAppHealthCheckDone : public google::protobuf::Closure { +public: + virtual void Run(); + + HealthCheckChannel channel; + brpc::Controller cntl; + SocketId id; + int64_t interval_s; + int64_t last_check_time_ms; +}; + +class HealthCheckManager { +public: + static void StartCheck(SocketId id, int64_t check_interval_s) { + SocketUniquePtr ptr; + const int rc = Socket::AddressFailedAsWell(id, &ptr); + if (rc < 0) { + RPC_VLOG << "SocketId=" << id + << " was abandoned during health checking"; + return; + } + LOG(INFO) << "Checking path=" << ptr->remote_side() << FLAGS_health_check_path; + OnAppHealthCheckDone* done = new OnAppHealthCheckDone; + done->id = id; + done->interval_s = check_interval_s; + brpc::ChannelOptions options; + options.protocol = PROTOCOL_HTTP; + options.max_retry = 0; + options.timeout_ms = + std::min((int64_t)FLAGS_health_check_timeout_ms, check_interval_s * 1000); + if (done->channel.Init(id, &options) != 0) { + LOG(WARNING) << "Fail to init health check channel to SocketId=" << id; + ptr->_ninflight_app_health_check.fetch_sub( + 1, butil::memory_order_relaxed); + delete done; + return; + } + AppCheck(done); + } + + static void* AppCheck(void* arg) { + OnAppHealthCheckDone* done = static_cast<OnAppHealthCheckDone*>(arg); + done->cntl.Reset(); + done->cntl.http_request().uri() = FLAGS_health_check_path; + ControllerPrivateAccessor(&done->cntl).set_health_check_call(); + done->last_check_time_ms = butil::gettimeofday_ms(); + done->channel.CallMethod(NULL, &done->cntl, NULL, NULL, done); + return NULL; + } + + static void RunAppCheck(void* arg) { + bthread_t th = 0; + int rc = bthread_start_background( + &th, &BTHREAD_ATTR_NORMAL, AppCheck, arg); + if (rc != 0) { + LOG(ERROR) << "Fail to start AppCheck"; + AppCheck(arg); + return; + } + } +}; + +void OnAppHealthCheckDone::Run() { + std::unique_ptr<OnAppHealthCheckDone> self_guard(this); + SocketUniquePtr ptr; + const int rc = Socket::AddressFailedAsWell(id, &ptr); + if (rc < 0) { + RPC_VLOG << "SocketId=" << id + << " was abandoned during health checking"; + return; + } + if (!cntl.Failed() || ptr->Failed()) { Review comment: OK，已加 ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services --------------------------------------------------------------------- To unsubscribe, e-mail: dev-unsubscr...@brpc.apache.org For additional commands, e-mail: dev-h...@brpc.apache.org
