mitkHealthController.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkHealthController_h
#define mitkHealthController_h
 
#include <mitkDataStorageBridge.h>
#include <mitkRestServerConfig.h>
#include <httplib.h>
 
#include <MitkRESTAPIExports.h>
 
#include <functional>
#include <optional>
#include <cstdint>
#include <string>
#include <vector>
 
namespace mitk
{
  class MITKRESTAPI_EXPORT HealthController
  {
  public:
    using UptimeCallback = std::function<std::optional<int64_t>()>;
 
    explicit HealthController(DataStorageBridge& bridge);
 
    void SetUptimeCallback(UptimeCallback callback);
 
    void HandleGET_health(const httplib::Request& req, httplib::Response& res);
 
    void HandleGET_info(const httplib::Request& req, httplib::Response& res);
 
    void HandleGET_config_file_access(const httplib::Request& req, httplib::Response& res);
 
    void SetFileAccessConfig(FileAccessMode mode, const std::vector<std::string>& allowedDirs);
 
    void SetMaxActiveTempDirsPerIp(size_t max);
 
  private:
    DataStorageBridge& m_Bridge;
    mutable std::mutex m_Mutex;
    UptimeCallback m_UptimeCallback;
    FileAccessMode m_FileAccessMode = FileAccessMode::Unrestricted;
    std::vector<std::string> m_AllowedFileDirectories;
    size_t m_MaxActiveTempDirsPerIp = 5;
  };
}
 
#endif