/** collector

A full notice with attributions is provided along with this source code.

This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License version 2 as published by the Free Software Foundation.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

* In addition, as a special exception, the copyright holders give
* permission to link the code of portions of this program with the
* OpenSSL library under certain conditions as described in each
* individual source file, and distribute linked combinations
* including the two.
* You must obey the GNU General Public License in all respects
* for all of the code used other than OpenSSL.  If you modify
* file(s) with this exception, you may extend this exception to your
* version of the file(s), but you are not obligated to do so.  If you
* do not wish to do so, delete this exception statement from your
* version.
*/

#ifndef _COLLECTOR_SERVICE_H_
#define _COLLECTOR_SERVICE_H_

#include <vector>

#include "CollectorConfig.h"
#include "CollectorStats.h"

namespace collector {

class CollectorService {
 public:
  enum ControlValue {
    RUN = 0,           // Keep running
    INTERRUPT_SYSDIG,  // Stop running sysdig, but resume collector operation (e.g., for chisel update)
    STOP_COLLECTOR,    // Stop the collector (e.g., SIGINT or SIGTERM received).
  };

  CollectorService(const CollectorConfig& config, std::atomic<ControlValue>* control, const std::atomic<int>* signum);

  void RunForever();

 private:
  void OnChiselReceived(const std::string& chisel);
  bool WaitForGRPCServer();

  CollectorConfig config_;

  std::string chisel_;
  bool update_chisel_ = false;
  std::mutex chisel_mutex_;

  std::atomic<ControlValue>* control_;
  const std::atomic<int>& signum_;
};

}  // namespace collector

#endif  // _COLLECTOR_SERVICE_H_
