SimpleStorageService.h
1 
10 #ifndef WRENCH_SIMPLESTORAGESERVICE_H
11 #define WRENCH_SIMPLESTORAGESERVICE_H
12 
13 
14 #include <wrench/services/storage/storage_helpers/FileTransferThread.h>
15 #include "wrench/services/storage/StorageService.h"
16 #include "SimpleStorageServiceProperty.h"
17 #include "SimpleStorageServiceMessagePayload.h"
18 #include "wrench/simgrid_S4U_util/S4U_PendingCommunication.h"
19 
20 namespace wrench {
21 
22  class SimulationMessage;
23  class SimulationTimestampFileCopyStart;
24 
25  class S4U_PendingCommunication;
26 
44 
45  private:
46  std::map<std::string, std::string> default_property_values = {
48  {SimpleStorageServiceProperty::BUFFER_SIZE, "1048576"}, // 1 MEGA BYTE
49  };
50 
51  std::map<std::string, double> default_messagepayload_values = {
66  };
67 
68  public:
69 
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  void cleanup(bool has_returned_from_main, int return_value) override;
82 
83 
84  /***********************/
86  /***********************/
87 
88  private:
89 
90  friend class Simulation;
91 
92  // Low-level Constructor
93  SimpleStorageService(std::string hostname,
94  std::set<std::string> mount_points,
95  std::map<std::string, std::string> property_list,
96  std::map<std::string, double> messagepayload_list,
97  std::string suffix);
98 
99  int main() override;
100 
101  bool processNextMessage();
102 
103  unsigned long getNewUniqueNumber();
104 
105  bool processFileDeleteRequest(WorkflowFile *file, std::shared_ptr<FileLocation> location, std::string answer_mailbox);
106 
107  bool processFileWriteRequest(WorkflowFile *file, std::shared_ptr<FileLocation>, std::string answer_mailbox, unsigned long buffer_size);
108 
109  bool processFileReadRequest(WorkflowFile *file, std::shared_ptr<FileLocation> location, std::string answer_mailbox,
110  std::string mailbox_to_receive_the_file_content, unsigned long buffer_size);
111 
112  bool processFileCopyRequest(WorkflowFile *file,
113  std::shared_ptr<FileLocation> src,
114  std::shared_ptr<FileLocation> dst,
115  std::string answer_mailbox);
116 
117  bool processFileTransferThreadNotification(
118  std::shared_ptr<FileTransferThread> ftt,
119  WorkflowFile *file,
120  std::string src_mailbox,
121  std::shared_ptr<FileLocation> src_location,
122  std::string dst_mailbox,
123  std::shared_ptr<FileLocation> dst_location,
124  bool success,
125  std::shared_ptr<FailureCause> failure_cause,
126  std::string answer_mailbox_if_read,
127  std::string answer_mailbox_if_write,
128  std::string answer_mailbox_if_copy);
129 
130  unsigned long num_concurrent_connections;
131 
132  void startPendingFileTransferThread();
133 
134  std::deque<std::shared_ptr<FileTransferThread>> pending_file_transfer_threads;
135  std::set<std::shared_ptr<FileTransferThread>> running_file_transfer_threads;
136 
137  void validateProperties();
138 
139  };
140 
141 };
142 
143 
144 #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
void cleanup(bool has_returned_from_main, int return_value) override
Cleanup method.
Definition: SimpleStorageService.cpp:57
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
std::map< std::string, double > messagepayload_list
The service's messagepayload list.
Definition: Service.h:112
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
unsigned long buffer_size
The service's buffer size.
Definition: StorageService.h:98
std::string hostname
The name of the host on which the daemon is running.
Definition: S4U_Daemon.h:51
A class that provides basic simulation methods. Once the simulation object has been explicitly or imp...
Definition: Simulation.h:45
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:43
A data file used/produced by a WorkflowTask in a Workflow.
Definition: WorkflowFile.h:26
~SimpleStorageService() override
Destructor.
Definition: SimpleStorageService.cpp:47
std::map< std::string, std::string > property_list
The service's property list.
Definition: Service.h:109