ServiceDiscovery.h

// Copyright CERN and copyright holders of ALICE O2. This software is
// distributed under the terms of the GNU General Public License v3 (GPL
// Version 3), copied verbatim in the file "COPYING".
//
// See http://alice-o2.web.cern.ch/license for full licensing information.
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.


#ifndef QC_SERVICEDISCOVERY_H
#define QC_SERVICEDISCOVERY_H

#include <atomic>
#include <curl/curl.h>
#include <memory>
#include <string>
#include <thread>
#include <boost/asio/ip/host_name.hpp>

namespace o2::quality_control::core
{

class ServiceDiscovery
{
 public:
  ServiceDiscovery(const std::string& url, const std::string& name, const std::string& id, const std::string& healthEndpoint = GetDefaultUrl());

  ~ServiceDiscovery();

  void _register(const std::string& objects);

  void deregister();

  static inline std::string GetDefaultUrl() 
  {
    return GetDefaultUrl(DefaultHealthPort);
  }

  static inline std::string GetDefaultUrl(size_t port) 
  {
    return boost::asio::ip::host_name() + ":" + std::to_string(port);
  }

  static constexpr size_t DefaultHealthPort = 7777; 

 private:
  static void deleteCurl(CURL* curl);

  std::unique_ptr<CURL, decltype(&ServiceDiscovery::deleteCurl)> curlHandle;

  const std::string mConsulUrl;     
  const std::string mName;          
  const std::string mId;            
  std::string mHealthEndpoint;      
  std::thread mHealthThread;        
  std::atomic<bool> mThreadRunning; 

  CURL* initCurl();

  void send(const std::string& path, std::string&& request);

  void runHealthServer(unsigned int port);
};

} // namespace o2::quality_control::core
#endif // QC_SERVICEDISCOVERY_H