YugabyteDB (2.13.1.0-b60, 21121d69985fbf76aa6958d8f04a9bfa936293b5)

// 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.

//

// The following only applies to changes made to this file as part of YugaByte development.

//

// Portions Copyright (c) YugaByte, 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.

//

#ifndef YB_TSERVER_HEARTBEATER_H

#define YB_TSERVER_HEARTBEATER_H

#include <memory>

#include "yb/server/server_base_options.h"

#include "yb/gutil/macros.h"

#include "yb/master/master_fwd.h"

#include "yb/master/master_heartbeat.fwd.h"

#include "yb/tserver/tserver_fwd.h"

#include "yb/util/status_fwd.h"

#include "yb/util/net/net_util.h"

namespace yb {

namespace tserver {

// Interface data providers to be used for filling data into heartbeat request.

// Data provider could fill in data into TSHeartbeatRequestPB that will be send by Heartbeater

// after that.

class HeartbeatDataProvider {

 public:

  explicit HeartbeatDataProvider(TabletServer* server) : server_(*CHECK_NOTNULL(server)) {}

  virtual ~HeartbeatDataProvider() {}

  // Add data to heartbeat, provider could skip and do nothing if is it too early for example for

  // periodical provider.

  // Called on every heartbeat from Heartbeater::Thread::TryHeartbeat.

  virtual void AddData(

      const master::TSHeartbeatResponsePB& last_resp, master::TSHeartbeatRequestPB* req) = 0;

  const std::string& LogPrefix() const;

  TabletServer& server() { return server_; }

 private:

  TabletServer& server_;

};

// Component of the Tablet Server which is responsible for heartbeating to the

// leader master.

//

// TODO: send heartbeats to non-leader masters.

class Heartbeater {

 public:

  Heartbeater(

      const TabletServerOptions& options, TabletServer* server,

      std::vector<std::unique_ptr<HeartbeatDataProvider>>&& data_providers);

  Heartbeater(const Heartbeater& other) = delete;

  void operator=(const Heartbeater& other) = delete;

  CHECKED_STATUS Start();

  CHECKED_STATUS Stop();

  // Trigger a heartbeat as soon as possible, even if the normal

  // heartbeat interval has not expired.

  void TriggerASAP();

  void set_master_addresses(server::MasterAddressesPtr master_addresses);

  ~Heartbeater();

 private:

  class Thread;

  std::unique_ptr<Thread> thread_;

};

class PeriodicalHeartbeatDataProvider : public HeartbeatDataProvider {

 public:

  PeriodicalHeartbeatDataProvider(TabletServer* server, const MonoDelta& period) :

    HeartbeatDataProvider(server), period_(period) {}

  void AddData(

      const master::TSHeartbeatResponsePB& last_resp, master::TSHeartbeatRequestPB* req) override;

  CoarseTimePoint prev_run_time() const { return prev_run_time_; }

 private:

  virtual void DoAddData(

      const master::TSHeartbeatResponsePB& last_resp, master::TSHeartbeatRequestPB* req) = 0;

  MonoDelta period_;

  CoarseTimePoint prev_run_time_;

};

} // namespace tserver

} // namespace yb

#endif /* YB_TSERVER_HEARTBEATER_H */