Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Useless if().
[simgrid.git] / src / surf / HostImpl.cpp
1 /* Copyright (c) 2013-2019. The SimGrid Team. All rights reserved.          */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include "src/plugins/vm/VirtualMachineImpl.hpp"
7 #include "src/simix/smx_private.hpp"
8
9 #include <string>
10
11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_host, surf, "Logging specific to the SURF host module");
12
13 simgrid::surf::HostModel *surf_host_model = nullptr;
14
15 /*************
16  * Callbacks *t
17  *************/
18
19 namespace simgrid {
20 namespace surf {
21
22 /*********
23  * Model *
24  *********/
25 /* Helper function for executeParallelTask */
26 static inline double has_cost(const double* array, size_t pos)
27 {
28   if (array)
29     return array[pos];
30   return -1.0;
31 }
32
33 kernel::resource::Action* HostModel::execute_parallel(const std::vector<s4u::Host*>& host_list,
34                                                       const double* flops_amount, const double* bytes_amount,
35                                                       double rate)
36 {
37   kernel::resource::Action* action = nullptr;
38   if ((host_list.size() == 1) && (has_cost(bytes_amount, 0) <= 0) && (has_cost(flops_amount, 0) > 0)) {
39     action = host_list[0]->pimpl_cpu->execution_start(flops_amount[0]);
40   } else if ((host_list.size() == 1) && (has_cost(flops_amount, 0) <= 0)) {
41     action = surf_network_model->communicate(host_list[0], host_list[0], bytes_amount[0], rate);
42   } else if ((host_list.size() == 2) && (has_cost(flops_amount, 0) <= 0) && (has_cost(flops_amount, 1) <= 0)) {
43     int nb = 0;
44     double value = 0.0;
45
46     for (size_t i = 0; i < host_list.size() * host_list.size(); i++) {
47       if (has_cost(bytes_amount, i) > 0.0) {
48         nb++;
49         value = has_cost(bytes_amount, i);
50       }
51     }
52     if (nb == 1) {
53       action = surf_network_model->communicate(host_list[0], host_list[1], value, rate);
54     } else if (nb == 0) {
55       xbt_die("Cannot have a communication with no flop to exchange in this model. You should consider using the "
56           "ptask model");
57     } else {
58       xbt_die("Cannot have a communication that is not a simple point-to-point in this model. You should consider "
59           "using the ptask model");
60     }
61   } else {
62     xbt_die(
63         "This model only accepts one of the following. You should consider using the ptask model for the other cases.\n"
64         " - execution with one host only and no communication\n"
65         " - Self-comms with one host only\n"
66         " - Communications with two hosts and no computation");
67   }
68   return action;
69 }
70
71 /************
72  * Resource *
73  ************/
74 HostImpl::HostImpl(s4u::Host* host) : piface_(host)
75 {
76   /* The VM wants to reinstall a new HostImpl, but we don't want to leak the previously existing one */
77   delete piface_->pimpl_;
78   piface_->pimpl_ = this;
79 }
80
81 HostImpl::~HostImpl()
82 {
83   /* All processes should be gone when the host is turned off (by the end of the simulation). */
84   if (not actor_list_.empty()) {
85     std::string msg = std::string("Shutting down host, but it's not empty:");
86     for (auto const& process : actor_list_)
87       msg += "\n\t" + std::string(process.get_name());
88
89     SIMIX_display_process_status();
90     xbt_die("%s", msg.c_str());
91   }
92   for (auto const& arg : actors_at_boot_)
93     delete arg;
94   actors_at_boot_.clear();
95
96   for (auto const& d : disks_)
97     d->destroy();
98 }
99
100 /** Re-starts all the actors that are marked as restartable.
101  *
102  * Weird things will happen if you turn on a host that is already on. S4U is fool-proof, not this.
103  */
104 void HostImpl::turn_on()
105 {
106   for (auto const& arg : actors_at_boot_) {
107     XBT_DEBUG("Booting Actor %s(%s) right now", arg->name.c_str(), arg->host->get_cname());
108     simgrid::kernel::actor::ActorImplPtr actor = simgrid::kernel::actor::ActorImpl::create(
109         arg->name, arg->code, nullptr, arg->host, arg->properties.get(), nullptr);
110     if (arg->on_exit)
111       *actor->on_exit = *arg->on_exit;
112     if (arg->kill_time >= 0)
113       actor->set_kill_time(arg->kill_time);
114     if (arg->auto_restart)
115       actor->set_auto_restart(arg->auto_restart);
116     if (arg->daemon_)
117       actor->daemonize();
118   }
119 }
120
121 /** Kill all actors hosted here */
122 void HostImpl::turn_off()
123 {
124   for (auto& actor : actor_list_) {
125     XBT_DEBUG("Killing Actor %s@%s on behalf of %s which turned off that host.", actor.get_cname(),
126               actor.get_host()->get_cname(), SIMIX_process_self()->get_cname());
127     SIMIX_process_self()->kill(&actor);
128   }
129   // When a host is turned off, we want to keep only the actors that should restart for when it will boot again.
130   // Then get rid of the others.
131   auto elm = remove_if(begin(actors_at_boot_), end(actors_at_boot_), [](kernel::actor::ProcessArg* arg) {
132     if (arg->auto_restart)
133       return false;
134     delete arg;
135     return true;
136   });
137   actors_at_boot_.erase(elm, end(actors_at_boot_));
138 }
139
140 std::vector<s4u::ActorPtr> HostImpl::get_all_actors()
141 {
142   std::vector<s4u::ActorPtr> res;
143   for (auto& actor : actor_list_)
144     res.push_back(actor.ciface());
145   return res;
146 }
147 size_t HostImpl::get_actor_count()
148 {
149   return actor_list_.size();
150 }
151
152 std::vector<s4u::Disk*> HostImpl::get_disks()
153 {
154   std::vector<s4u::Disk*> disks;
155   for (auto const& d : disks_)
156     disks.push_back(&d->piface_);
157   return disks;
158 }
159
160 void HostImpl::add_disk(s4u::Disk* disk)
161 {
162   disks_.push_back(disk->get_impl());
163 }
164
165 void HostImpl::remove_disk(const std::string& disk_name)
166 {
167   auto position = disks_.begin();
168   for (auto const& d : disks_) {
169     if (d->get_name() == disk_name) {
170       disks_.erase(position);
171       break;
172     }
173     position++;
174   }
175 }
176
177 std::vector<const char*> HostImpl::get_attached_storages()
178 {
179   std::vector<const char*> storages;
180   for (auto const& s : storage_)
181     if (s.second->get_host() == piface_->get_cname())
182       storages.push_back(s.second->piface_.get_cname());
183   return storages;
184 }
185
186 } // namespace surf
187 } // namespace simgrid