SimpleStorageService.h
1 
10 #ifndef WRENCH_SIMPLESTORAGESERVICE_H
11 #define WRENCH_SIMPLESTORAGESERVICE_H
12 
13 #include "wrench/services/storage/storage_helpers/FileTransferThread.h"
14 #include "wrench/services/storage/StorageService.h"
15 #include "SimpleStorageServiceProperty.h"
16 #include "SimpleStorageServiceMessagePayload.h"
17 #include "wrench/services/memory/MemoryManager.h"
18 #include "wrench/simgrid_S4U_util/S4U_PendingCommunication.h"
19 
20 namespace wrench {
21 
22  class SimulationMessage;
23 
24  class SimulationTimestampFileCopyStart;
25 
26  class S4U_PendingCommunication;
27 
45 
46  private:
47  std::map <std::string, std::string> default_property_values = {
49  {SimpleStorageServiceProperty::BUFFER_SIZE, "10485760"}, // 10 MEGA BYTE
50  };
51 
52  std::map<std::string, double> default_messagepayload_values = {
67  };
68 
69  public:
70  // Public Constructor
71  SimpleStorageService(std::string hostname,
72  std::set <std::string> mount_points,
73  std::map <std::string, std::string> property_list = {},
74  std::map<std::string, double> messagepayload_list = {});
75 
76  /***********************/
78  /***********************/
79 
80  ~SimpleStorageService() override;
81 
82  void cleanup(bool has_returned_from_main, int return_value) override;
83 
84  /***********************/
86  /***********************/
87 
88  private:
89  friend class Simulation;
90 
91  // Low-level Constructor
92  SimpleStorageService(std::string hostname,
93  std::set <std::string> mount_points,
94  std::map <std::string, std::string> property_list,
95  std::map<std::string, double> messagepayload_list,
96  std::string suffix);
97 
98  int main() override;
99 
100  bool processNextMessage();
101 
102  unsigned long getNewUniqueNumber();
103 
104  bool processFileDeleteRequest(WorkflowFile *file, std::shared_ptr <FileLocation> location,
105  std::string answer_mailbox);
106 
107  bool processFileWriteRequest(WorkflowFile *file, std::shared_ptr <FileLocation>, std::string answer_mailbox,
108  unsigned long buffer_size);
109 
110  bool
111  processFileReadRequest(WorkflowFile *file, std::shared_ptr <FileLocation> location, std::string answer_mailbox,
112  std::string mailbox_to_receive_the_file_content, unsigned long buffer_size);
113 
114  bool processFileCopyRequest(WorkflowFile *file,
115  std::shared_ptr <FileLocation> src,
116  std::shared_ptr <FileLocation> dst,
117  std::string answer_mailbox);
118 
119  bool processFileTransferThreadNotification(
120  std::shared_ptr <FileTransferThread> ftt,
121  WorkflowFile *file,
122  std::string src_mailbox,
123  std::shared_ptr <FileLocation> src_location,
124  std::string dst_mailbox,
125  std::shared_ptr <FileLocation> dst_location,
126  bool success,
127  std::shared_ptr <FailureCause> failure_cause,
128  std::string answer_mailbox_if_read,
129  std::string answer_mailbox_if_write,
130  std::string answer_mailbox_if_copy);
131 
132  unsigned long num_concurrent_connections;
133 
134  void startPendingFileTransferThread();
135 
136  std::deque <std::shared_ptr<FileTransferThread>> pending_file_transfer_threads;
137  std::set <std::shared_ptr<FileTransferThread>> running_file_transfer_threads;
138 
139  void validateProperties();
140 
141  std::shared_ptr<MemoryManager> memory_manager;
142 
143  };
144 
145 };
146 
147 #endif //WRENCH_SIMPLESTORAGESERVICE_H
SimpleStorageService(std::string hostname, std::set< std::string > mount_points, std::map< std::string, std::string > property_list={}, std::map< std::string, double > messagepayload_list={})
Public constructor.
Definition: SimpleStorageService.cpp:71
static const std::string BUFFER_SIZE
Buffer size used when copying/communicating data:
Definition: StorageServiceProperty.h:32
static const std::string DAEMON_STOPPED_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to confirm it has terminated.
Definition: ServiceMessagePayload.h:33
static const std::string FILE_READ_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to answer a file copy request.
Definition: StorageServiceMessagePayload.h:54
static const std::string FILE_COPY_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to answer a file copy request.
Definition: StorageServiceMessagePayload.h:44
static const std::string FREE_SPACE_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to answer free space information reques...
Definition: StorageServiceMessagePayload.h:29
static const std::string FILE_WRITE_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to answer a file copy request.
Definition: StorageServiceMessagePayload.h:49
The storage service base class.
Definition: StorageService.h:36
static const std::string FILE_LOOKUP_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to request a file lookup.
Definition: StorageServiceMessagePayload.h:32
Definition: Alarm.cpp:20
static const std::string FILE_WRITE_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to request a file copy.
Definition: StorageServiceMessagePayload.h:47
static const std::string FILE_READ_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to request a file copy.
Definition: StorageServiceMessagePayload.h:52
static const std::string FILE_DELETE_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to answer a file deletion request.
Definition: StorageServiceMessagePayload.h:39
static const std::string FILE_DELETE_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to request a file deletion.
Definition: StorageServiceMessagePayload.h:37
static const std::string FREE_SPACE_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to request its free space information.
Definition: StorageServiceMessagePayload.h:27
static const std::string FILE_LOOKUP_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to answer a file lookup request.
Definition: StorageServiceMessagePayload.h:34
static const std::string STOP_DAEMON_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to terminate it.
Definition: ServiceMessagePayload.h:31
A class that provides basic simulation methods. Once the simulation object has been explicitly or imp...
Definition: Simulation.h:46
static const std::string FILE_COPY_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to request a file copy.
Definition: StorageServiceMessagePayload.h:42
static const std::string MAX_NUM_CONCURRENT_DATA_CONNECTIONS
The maximum number of concurrent data connections supported by the service (default = "infinity")
Definition: SimpleStorageServiceProperty.h:26
A storage service that provides direct access to some storage resources (e.g., one or more disks)....
Definition: SimpleStorageService.h:44
A data file used/produced by a WorkflowTask in a Workflow.
Definition: WorkflowFile.h:26