Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
simplify storage internals
[simgrid.git] / src / surf / storage_n11.cpp
1 /* Copyright (c) 2013-2015. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "storage_n11.hpp"
8 #include "simgrid/s4u/engine.hpp"
9 #include "src/kernel/routing/NetPoint.hpp"
10 #include "surf_private.h"
11 #include <math.h> /*ceil*/
12
13 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(surf_storage);
14
15 /*************
16  * CallBacks *
17  *************/
18
19 static inline void routing_storage_type_free(void *r)
20 {
21   storage_type_t stype = (storage_type_t) r;
22   free(stype->model);
23   free(stype->type_id);
24   free(stype->content);
25   free(stype->content_type);
26   xbt_dict_free(&(stype->properties));
27   xbt_dict_free(&(stype->model_properties));
28   free(stype);
29 }
30
31 static void check_disk_attachment()
32 {
33   xbt_lib_cursor_t cursor;
34   char* key;
35   void** data;
36   xbt_lib_foreach(storage_lib, cursor, key, data) {
37     if (xbt_lib_get_level(xbt_lib_get_elm_or_null(storage_lib, key), SURF_STORAGE_LEVEL) != nullptr) {
38       simgrid::surf::Storage* storage =
39           static_cast<simgrid::surf::Storage*>(xbt_lib_get_or_null(storage_lib, key, SURF_STORAGE_LEVEL));
40       simgrid::kernel::routing::NetPoint* host_elm = sg_netpoint_by_name_or_null(storage->attach_);
41       if (!host_elm)
42         surf_parse_error("Unable to attach storage %s: host %s does not exist.", storage->cname(), storage->attach_);
43     }
44   }
45 }
46
47 void storage_register_callbacks()
48 {
49   simgrid::s4u::onPlatformCreated.connect(check_disk_attachment);
50   instr_routing_define_callbacks();
51
52   ROUTING_STORAGE_LEVEL = xbt_lib_add_level(storage_lib, xbt_free_f);
53   ROUTING_STORAGE_TYPE_LEVEL = xbt_lib_add_level(storage_type_lib, routing_storage_type_free);
54   SURF_STORAGE_LEVEL = xbt_lib_add_level(storage_lib, [](void *self) {
55     delete static_cast<simgrid::surf::Storage*>(self);
56   });
57 }
58
59 /*********
60  * Model *
61  *********/
62
63 void surf_storage_model_init_default()
64 {
65   surf_storage_model = new simgrid::surf::StorageN11Model();
66   all_existing_models->push_back(surf_storage_model);
67 }
68
69 namespace simgrid {
70 namespace surf {
71
72 Storage* StorageN11Model::createStorage(const char* id, const char* type_id, const char* content_name,
73                                         const char* content_type, const char* attach)
74 {
75
76   xbt_assert(!surf_storage_resource_priv(surf_storage_resource_by_name(id)),
77       "Storage '%s' declared several times in the platform file", id);
78
79   storage_type_t storage_type = (storage_type_t) xbt_lib_get_or_null(storage_type_lib, type_id,ROUTING_STORAGE_TYPE_LEVEL);
80
81   double Bread  = surf_parse_get_bandwidth((char*)xbt_dict_get(storage_type->model_properties, "Bread"),
82       "property Bread, storage",type_id);
83   double Bwrite = surf_parse_get_bandwidth((char*)xbt_dict_get(storage_type->model_properties, "Bwrite"),
84       "property Bwrite, storage",type_id);
85   double Bconnection   = surf_parse_get_bandwidth((char*)xbt_dict_get(storage_type->model_properties, "Bconnection"),
86       "property Bconnection, storage",type_id);
87
88   Storage* storage = new StorageN11(this, id, maxminSystem_, Bread, Bwrite, Bconnection, type_id, (char*)content_name,
89                                     content_type, storage_type->size, (char*)attach);
90   storageCreatedCallbacks(storage);
91   xbt_lib_set(storage_lib, id, SURF_STORAGE_LEVEL, storage);
92
93   XBT_DEBUG("SURF storage create resource\n\t\tid '%s'\n\t\ttype '%s'\n\t\tBread '%f'\n", id, type_id, Bread);
94
95   p_storageList.push_back(storage);
96
97   return storage;
98 }
99
100 double StorageN11Model::nextOccuringEvent(double now)
101 {
102   double min_completion = StorageModel::nextOccuringEventFull(now);
103
104   for(auto storage: p_storageList) {
105     double rate = 0;
106     // Foreach write action on that disk
107     for (auto write_action: storage->writeActions_) {
108       rate += lmm_variable_getvalue(write_action->getVariable());
109     }
110     if(rate > 0)
111       min_completion = MIN(min_completion, (storage->size_-storage->usedSize_)/rate);
112   }
113
114   return min_completion;
115 }
116
117 void StorageN11Model::updateActionsState(double /*now*/, double delta)
118 {
119
120   ActionList *actionSet = getRunningActionSet();
121   for(ActionList::iterator it(actionSet->begin()), itNext=it, itend(actionSet->end())
122       ; it != itend ; it=itNext) {
123     ++itNext;
124
125     StorageAction *action = static_cast<StorageAction*>(&*it);
126
127     if (action->type_ == WRITE) {
128       // Update the disk usage
129       // Update the file size
130       // For each action of type write
131       double current_progress = delta * lmm_variable_getvalue(action->getVariable());
132       long int incr = current_progress;
133
134       XBT_DEBUG("%s:\n\t progress =  %.2f, current_progress = %.2f, incr = %ld, lrint(1) = %ld, lrint(2) = %ld",
135                 action->file_->name, action->progress_, current_progress, incr,
136                 lrint(action->progress_ + current_progress), lrint(action->progress_) + incr);
137
138       /* take care of rounding error accumulation */
139       if (lrint(action->progress_ + current_progress) > lrint(action->progress_) + incr)
140         incr++;
141
142       action->progress_ += current_progress;
143
144       action->storage_->usedSize_ += incr;     // disk usage
145       action->file_->current_position += incr; // current_position
146       //  which becomes the new file size
147       action->file_->size = action->file_->current_position;
148
149       sg_size_t *psize = xbt_new(sg_size_t,1);
150       *psize                  = action->file_->size;
151       xbt_dict_t content_dict = action->storage_->content_;
152       xbt_dict_set(content_dict, action->file_->name, psize, nullptr);
153     }
154
155     action->updateRemains(lmm_variable_getvalue(action->getVariable()) * delta);
156
157     if (action->getMaxDuration() > NO_MAX_DURATION)
158       action->updateMaxDuration(delta);
159
160     if (action->getRemainsNoUpdate() > 0 && lmm_get_variable_weight(action->getVariable()) > 0 &&
161         action->storage_->usedSize_ == action->storage_->size_) {
162       action->finish();
163       action->setState(Action::State::failed);
164     } else if (((action->getRemainsNoUpdate() <= 0) && (lmm_get_variable_weight(action->getVariable()) > 0)) ||
165                ((action->getMaxDuration() > NO_MAX_DURATION) && (action->getMaxDuration() <= 0))) {
166       action->finish();
167       action->setState(Action::State::done);
168     }
169   }
170 }
171
172 /************
173  * Resource *
174  ************/
175
176 StorageN11::StorageN11(StorageModel* model, const char* name, lmm_system_t maxminSystem, double bread, double bwrite,
177                        double bconnection, const char* type_id, char* content_name, const char* content_type,
178                        sg_size_t size, char* attach)
179     : Storage(model, name, maxminSystem, bread, bwrite, bconnection, type_id, content_name, content_type, size, attach)
180 {
181   XBT_DEBUG("Create resource with Bconnection '%f' Bread '%f' Bwrite '%f' and Size '%llu'", bconnection, bread, bwrite, size);
182 }
183
184 StorageAction *StorageN11::open(const char* mount, const char* path)
185 {
186   XBT_DEBUG("\tOpen file '%s'",path);
187
188   sg_size_t size;
189   sg_size_t* psize = (sg_size_t*)xbt_dict_get_or_null(content_, path);
190   // if file does not exist create an empty file
191   if(psize)
192     size = *psize;
193   else {
194     psize = xbt_new(sg_size_t,1);
195     size = 0;
196     *psize = size;
197     xbt_dict_set(content_, path, psize, nullptr);
198     XBT_DEBUG("File '%s' was not found, file created.",path);
199   }
200   surf_file_t file = xbt_new0(s_surf_file_t,1);
201   file->name = xbt_strdup(path);
202   file->size = size;
203   file->mount = xbt_strdup(mount);
204   file->current_position = 0;
205
206   StorageAction* action = new StorageN11Action(model(), 0, isOff(), this, OPEN);
207   action->file_         = file;
208
209   return action;
210 }
211
212 StorageAction *StorageN11::close(surf_file_t fd)
213 {
214   XBT_DEBUG("\tClose file '%s' size '%llu'", fd->name, fd->size);
215   // unref write actions from storage
216   for (std::vector<StorageAction*>::iterator it = writeActions_.begin(); it != writeActions_.end();) {
217     StorageAction *write_action = *it;
218     if ((write_action->file_) == fd) {
219       write_action->unref();
220       it = writeActions_.erase(it);
221     } else {
222       ++it;
223     }
224   }
225   free(fd->name);
226   free(fd->mount);
227   xbt_free(fd);
228   StorageAction* action = new StorageN11Action(model(), 0, isOff(), this, CLOSE);
229   return action;
230 }
231
232 StorageAction *StorageN11::read(surf_file_t fd, sg_size_t size)
233 {
234   if(fd->current_position + size > fd->size){
235     if (fd->current_position > fd->size){
236       size = 0;
237     } else {
238       size = fd->size - fd->current_position;
239     }
240     fd->current_position = fd->size;
241   }
242   else
243     fd->current_position += size;
244
245   StorageAction* action = new StorageN11Action(model(), size, isOff(), this, READ);
246   return action;
247 }
248
249 StorageAction *StorageN11::write(surf_file_t fd, sg_size_t size)
250 {
251   char *filename = fd->name;
252   XBT_DEBUG("\tWrite file '%s' size '%llu/%llu'",filename,size,fd->size);
253
254   StorageAction* action = new StorageN11Action(model(), size, isOff(), this, WRITE);
255   action->file_         = fd;
256   /* Substract the part of the file that might disappear from the used sized on the storage element */
257   usedSize_ -= (fd->size - fd->current_position);
258   // If the storage is full before even starting to write
259   if(usedSize_==size_) {
260     action->setState(Action::State::failed);
261   }
262   return action;
263 }
264
265 /**********
266  * Action *
267  **********/
268
269 StorageN11Action::StorageN11Action(Model *model, double cost, bool failed, Storage *storage, e_surf_action_storage_type_t type)
270 : StorageAction(model, cost, failed,
271     lmm_variable_new(model->getMaxminSystem(), this, 1.0, -1.0 , 3),
272     storage, type) {
273   XBT_IN("(%s,%g", storage->cname(), cost);
274
275   // Must be less than the max bandwidth for all actions
276   lmm_expand(model->getMaxminSystem(), storage->constraint(), getVariable(), 1.0);
277   switch(type) {
278   case OPEN:
279   case CLOSE:
280   case STAT:
281     break;
282   case READ:
283     lmm_expand(model->getMaxminSystem(), storage->constraintRead_, getVariable(), 1.0);
284     break;
285   case WRITE:
286     lmm_expand(model->getMaxminSystem(), storage->constraintWrite_, getVariable(), 1.0);
287
288     //TODO there is something annoying with what's below. Have to sort it out...
289     //    Action *action = this;
290     //    storage->p_writeActions->push_back(action);
291     //    ref();
292     break;
293   default:
294     THROW_UNIMPLEMENTED;
295   }
296   XBT_OUT();
297 }
298
299 int StorageN11Action::unref()
300 {
301   refcount_--;
302   if (!refcount_) {
303     if (action_hook.is_linked())
304       stateSet_->erase(stateSet_->iterator_to(*this));
305     if (getVariable())
306       lmm_variable_free(getModel()->getMaxminSystem(), getVariable());
307     xbt_free(getCategory());
308     delete this;
309     return 1;
310   }
311   return 0;
312 }
313
314 void StorageN11Action::cancel()
315 {
316   setState(Action::State::failed);
317 }
318
319 void StorageN11Action::suspend()
320 {
321   XBT_IN("(%p)", this);
322   if (suspended_ != 2) {
323     lmm_update_variable_weight(getModel()->getMaxminSystem(), getVariable(), 0.0);
324     suspended_ = 1;
325   }
326   XBT_OUT();
327 }
328
329 void StorageN11Action::resume()
330 {
331   THROW_UNIMPLEMENTED;
332 }
333
334 bool StorageN11Action::isSuspended()
335 {
336   return suspended_ == 1;
337 }
338
339 void StorageN11Action::setMaxDuration(double /*duration*/)
340 {
341   THROW_UNIMPLEMENTED;
342 }
343
344 void StorageN11Action::setPriority(double /*priority*/)
345 {
346   THROW_UNIMPLEMENTED;
347 }
348
349 }
350 }