src/surf/StorageImpl.cpp

   1 /* Copyright (c) 2013-2017. 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 "StorageImpl.hpp"
   8
   9 #include "surf_private.h"
  10 #include <boost/algorithm/string.hpp>
  11 #include <boost/algorithm/string/join.hpp>
  12 #include <boost/algorithm/string/split.hpp>
  13 #include <fstream>
  14
  15 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_storage, surf, "Logging specific to the SURF storage module");
  16
  17 simgrid::surf::StorageModel* surf_storage_model = nullptr;
  18
  19 namespace simgrid {
  20 namespace surf {
  21
  22 /*************
  23  * Callbacks *
  24  *************/
  25
  26 simgrid::xbt::signal<void(StorageImpl*)> storageCreatedCallbacks;
  27 simgrid::xbt::signal<void(StorageImpl*)> storageDestructedCallbacks;
  28 simgrid::xbt::signal<void(StorageImpl*, int, int)> storageStateChangedCallbacks; // signature: wasOn, isOn
  29 simgrid::xbt::signal<void(StorageAction*, Action::State, Action::State)> storageActionStateChangedCallbacks;
  30
  31 /* List of storages */
  32 std::unordered_map<std::string, StorageImpl*>* StorageImpl::storages =
  33     new std::unordered_map<std::string, StorageImpl*>();
  34
  35 StorageImpl* StorageImpl::byName(const char* name)
  36 {
  37   if (storages->find(name) == storages->end())
  38     return nullptr;
  39   return storages->at(name);
  40 }
  41
  42 /*********
  43  * Model *
  44  *********/
  45
  46 StorageModel::StorageModel() : Model()
  47 {
  48   maxminSystem_ = lmm_system_new(true /* lazy update */);
  49 }
  50
  51 StorageModel::~StorageModel()
  52 {
  53   lmm_system_free(maxminSystem_);
  54   surf_storage_model = nullptr;
  55 }
  56
  57 /************
  58  * Resource *
  59  ************/
  60
  61 StorageImpl::StorageImpl(Model* model, const char* name, lmm_system_t maxminSystem, double bread, double bwrite,
  62                          const char* type_id, const char* content_name, sg_size_t size, const char* attach)
  63     : Resource(model, name, lmm_constraint_new(maxminSystem, this, MAX(bread, bwrite)))
  64     , piface_(this)
  65     , size_(size)
  66     , usedSize_(0)
  67     , typeId_(xbt_strdup(type_id))
  68     , writeActions_(std::vector<StorageAction*>())
  69 {
  70   content_ = parseContent(content_name);
  71   attach_  = xbt_strdup(attach);
  72   turnOn();
  73   XBT_DEBUG("Create resource with Bread '%f' Bwrite '%f' and Size '%llu'", bread, bwrite, size);
  74   constraintRead_  = lmm_constraint_new(maxminSystem, this, bread);
  75   constraintWrite_ = lmm_constraint_new(maxminSystem, this, bwrite);
  76   storages->insert({name, this});
  77 }
  78
  79 StorageImpl::~StorageImpl()
  80 {
  81   storageDestructedCallbacks(this);
  82   if (content_ != nullptr) {
  83     for (auto entry : *content_)
  84       delete entry.second;
  85     delete content_;
  86   }
  87   free(typeId_);
  88   free(attach_);
  89 }
  90
  91 std::map<std::string, sg_size_t*>* StorageImpl::parseContent(const char* filename)
  92 {
  93   usedSize_ = 0;
  94   if ((not filename) || (strcmp(filename, "") == 0))
  95     return nullptr;
  96
  97   std::map<std::string, sg_size_t*>* parse_content = new std::map<std::string, sg_size_t*>();
  98
  99   std::ifstream* fs = surf_ifsopen(filename);
 100
 101   std::string line;
 102   std::vector<std::string> tokens;
 103   do {
 104     std::getline(*fs, line);
 105     boost::trim(line);
 106     if (line.length() > 0) {
 107       boost::split(tokens, line, boost::is_any_of(" \t"), boost::token_compress_on);
 108       xbt_assert(tokens.size() == 2, "Parse error in %s: %s", filename, line.c_str());
 109       sg_size_t size = std::stoull(tokens.at(1));
 110
 111       usedSize_ += size;
 112       sg_size_t* psize = new sg_size_t;
 113       *psize           = size;
 114       parse_content->insert({tokens.front(), psize});
 115     }
 116   } while (not fs->eof());
 117   delete fs;
 118   return parse_content;
 119 }
 120
 121 bool StorageImpl::isUsed()
 122 {
 123   THROW_UNIMPLEMENTED;
 124   return false;
 125 }
 126
 127 void StorageImpl::apply_event(tmgr_trace_event_t /*event*/, double /*value*/)
 128 {
 129   THROW_UNIMPLEMENTED;
 130 }
 131
 132 void StorageImpl::turnOn()
 133 {
 134   if (isOff()) {
 135     Resource::turnOn();
 136     storageStateChangedCallbacks(this, 0, 1);
 137   }
 138 }
 139 void StorageImpl::turnOff()
 140 {
 141   if (isOn()) {
 142     Resource::turnOff();
 143     storageStateChangedCallbacks(this, 1, 0);
 144   }
 145 }
 146
 147 std::map<std::string, sg_size_t*>* StorageImpl::getContent()
 148 {
 149   /* For the moment this action has no cost, but in the future we could take in account access latency of the disk */
 150   return content_;
 151 }
 152
 153 sg_size_t StorageImpl::getFreeSize()
 154 {
 155   return size_ - usedSize_;
 156 }
 157
 158 sg_size_t StorageImpl::getUsedSize()
 159 {
 160   return usedSize_;
 161 }
 162
 163 /**********
 164  * Action *
 165  **********/
 166 StorageAction::StorageAction(Model* model, double cost, bool failed, StorageImpl* storage,
 167                              e_surf_action_storage_type_t type)
 168     : Action(model, cost, failed), type_(type), storage_(storage), file_(nullptr)
 169 {
 170   progress_ = 0;
 171 };
 172
 173 StorageAction::StorageAction(Model* model, double cost, bool failed, lmm_variable_t var, StorageImpl* storage,
 174                              e_surf_action_storage_type_t type)
 175     : Action(model, cost, failed, var), type_(type), storage_(storage), file_(nullptr)
 176 {
 177   progress_ = 0;
 178 }
 179
 180 void StorageAction::setState(Action::State state)
 181 {
 182   Action::State old = getState();
 183   Action::setState(state);
 184   storageActionStateChangedCallbacks(this, old, state);
 185 }
 186 }
 187 }