centralized_initiator.h

// Copyright 2023 The SOLA authors
//
// This file is part of DAISI.
//
// DAISI is free software: you can redistribute it and/or modify it under the terms of the GNU
// General Public License as published by the Free Software Foundation; version 2.
//
// DAISI 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 DAISI. If not, see
// <https://www.gnu.org/licenses/>.
//
// SPDX-License-Identifier: GPL-2.0-only
 
#ifndef DAISI_CPPS_LOGICAL_ALGORITHMS_ASSIGNMENT_CENTRALIZED_INITIATOR_H_
#define DAISI_CPPS_LOGICAL_ALGORITHMS_ASSIGNMENT_CENTRALIZED_INITIATOR_H_
 
#include "assignment_initiator.h"
#include "cpps/common/cpps_communicator.h"
 
namespace daisi::cpps::logical {
 
class CentralizedInitiator : public AssignmentInitiator {
public:
  CentralizedInitiator(daisi::cpps::common::CppsCommunicatorPtr communicator,
                       std::shared_ptr<CppsLoggerNs3> logger);
  ~CentralizedInitiator() override = default;
 
  // TODO: check if possible to restict implementation by child classes without warnings
  //   REQUIRE_IMPLEMENTATION(AssignmentResponse);
  //   REQUIRE_IMPLEMENTATION(StatusUpdate);
 
  void addMaterialFlow(std::shared_ptr<material_flow::MFDLScheduler> scheduler) override;
 
protected:
  struct ParticipantInfo {
    std::string connection_string;
 
    amr::AmrStaticAbility ability;
 
    bool isValid() const { return (!connection_string.empty() && ability.isValid()); }
  };
 
  virtual void distributeMFTasks(uint32_t index, bool previously_allocated) = 0;
 
  virtual void storeParticipant(ParticipantInfo &info) = 0;
 
  std::vector<material_flow::MFDLScheduler> material_flows_;
 
  std::vector<AssignmentResponse> assignment_acceptions_;
 
  struct {
    util::Duration wait_to_receive_assignment_response = 0.3;
    util::Duration wait_to_receive_status_update = 0.3;
  } delays_;
 
private:
  void readAmrRequestFuture();
 
  std::future<minhton::FindResult> amr_find_result_;
 
  bool preparation_finished_ = false;
};
}  // namespace daisi::cpps::logical
 
#endif