init/periodic_scheduler.h

// Copyright 2020 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef INIT_PERIODIC_SCHEDULER_H_
#define INIT_PERIODIC_SCHEDULER_H_

#include <string>
#include <vector>

#include <base/files/file_path.h>
#include <base/time/time.h>

// PeriodicScheduler is responsible for executing task |task_name| by running
// |task_exec| every |period| seconds.
class PeriodicScheduler {
 public:
  PeriodicScheduler(const base::TimeDelta& period,
                    const base::TimeDelta& timeout,
                    const std::string& task_name,
                    const std::vector<std::string>& task_exec);
  ~PeriodicScheduler() = default;
  bool Run(bool start_immediately = false);

  void set_spool_dir_for_test(const base::FilePath& spool_dir) {
    spool_dir_ = spool_dir;
  }

  void set_check_freq_for_test(const base::TimeDelta& check_freq) {
    check_frequency_seconds_ = check_freq;
  }

 private:
  base::TimeDelta period_seconds_;
  base::TimeDelta timeout_seconds_;
  base::TimeDelta check_frequency_seconds_;
  std::string task_name_;
  base::FilePath spool_dir_;
  std::vector<std::string> process_args_;
};

#endif  // INIT_PERIODIC_SCHEDULER_H_