#include "simgrid/s4u/Io.hpp"
#include "src/kernel/activity/IoImpl.hpp"
#include "src/kernel/actor/ActorImpl.hpp"
+#include "src/kernel/actor/SimcallObserver.hpp"
#include "xbt/log.h"
namespace simgrid {
xbt::signal<void(Io const&)> Io::on_start;
xbt::signal<void(Io const&)> Io::on_completion;
-Io::Io()
+Io::Io(kernel::activity::IoImplPtr pimpl)
{
- pimpl_ = kernel::activity::IoImplPtr(new kernel::activity::IoImpl());
+ pimpl_ = pimpl;
+}
+
+void Io::complete(Activity::State state)
+{
+ Activity::complete(state);
+ on_completion(*this);
}
IoPtr Io::init()
{
- return IoPtr(new Io());
+ auto pimpl = kernel::activity::IoImplPtr(new kernel::activity::IoImpl());
+ return IoPtr(pimpl->get_iface());
}
Io* Io::start()
{
- kernel::actor::simcall([this] {
- (*boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_))
- .set_name(get_name())
- .set_disk(disk_->get_impl())
- .set_size(size_)
- .set_type(type_)
- .start();
- });
+ kernel::actor::simcall(
+ [this] { (*boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)).set_name(get_name()).start(); });
if (suspended_)
pimpl_->suspend();
return this;
}
-Io* Io::cancel()
-{
- simgrid::kernel::actor::simcall([this] { boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)->cancel(); });
- state_ = State::CANCELED;
- on_completion(*this);
- return this;
-}
-
-Io* Io::wait()
-{
- return this->wait_for(-1);
-}
-
-Io* Io::wait_for(double timeout)
+ssize_t Io::wait_any_for(const std::vector<IoPtr>& ios, double timeout)
{
- if (state_ == State::INITED)
- vetoable_start();
+ std::vector<kernel::activity::IoImpl*> rios(ios.size());
+ std::transform(begin(ios), end(ios), begin(rios),
+ [](const IoPtr& io) { return static_cast<kernel::activity::IoImpl*>(io->pimpl_.get()); });
kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
- kernel::actor::simcall_blocking([this, issuer, timeout] { this->get_impl()->wait_for(issuer, timeout); });
- state_ = State::FINISHED;
- this->release_dependencies();
-
- on_completion(*this);
- return this;
+ kernel::actor::IoWaitanySimcall observer{issuer, rios, timeout};
+ ssize_t changed_pos = kernel::actor::simcall_blocking(
+ [&observer] {
+ kernel::activity::IoImpl::wait_any_for(observer.get_issuer(), observer.get_ios(), observer.get_timeout());
+ },
+ &observer);
+ if (changed_pos != -1)
+ ios.at(changed_pos)->complete(State::FINISHED);
+ return changed_pos;
}
-IoPtr Io::set_disk(sg_disk_t disk)
+IoPtr Io::set_disk(const_sg_disk_t disk)
{
- disk_ = disk;
+ xbt_assert(state_ == State::INITED || state_ == State::STARTING, "Cannot set disk once the Io is started");
+
+ kernel::actor::simcall(
+ [this, disk] { boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)->set_disk(disk->get_impl()); });
// Setting the disk may allow to start the activity, let's try
if (state_ == State::STARTING)
IoPtr Io::set_size(sg_size_t size)
{
- size_ = size;
- Activity::set_remaining(size_);
+ xbt_assert(state_ == State::INITED || state_ == State::STARTING, "Cannot set size once the Io is started");
+ kernel::actor::simcall(
+ [this, size] { boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)->set_size(size); });
+ Activity::set_remaining(size);
return this;
}
IoPtr Io::set_op_type(OpType type)
{
- type_ = type;
+ xbt_assert(state_ == State::INITED || state_ == State::STARTING, "Cannot set size once the Io is started");
+ kernel::actor::simcall(
+ [this, type] { boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)->set_type(type); });
return this;
}
return boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)->get_performed_ioops();
}
+bool Io::is_assigned() const
+{
+ return boost::static_pointer_cast<kernel::activity::IoImpl>(pimpl_)->get_disk() != nullptr;
+}
+
} // namespace s4u
} // namespace simgrid