Initial commit

src/storage/service/StorageServer.cc

@@ -0,0 +1,60 @@
+#include "storage/service/StorageServer.h"
+
+#include <folly/experimental/coro/BlockingWait.h>
+#include <folly/experimental/coro/WithCancellation.h>
+#include <folly/logging/xlog.h>
+
+#include "common/kv/mem/MemKVEngine.h"
+#include "common/utils/Result.h"
+#include "core/service/CoreService.h"
+#include "storage/service/ReliableForwarding.h"
+#include "storage/service/StorageService.h"
+#include "stubs/common/RealStubFactory.h"
+#include "stubs/mgmtd/MgmtdServiceStub.h"
+
+namespace hf3fs::storage {
+
+StorageServer::StorageServer(const Components::Config &config)
+    : net::Server(config.base()),
+      components_(config) {}
+
+StorageServer::~StorageServer() {
+  stopAndJoin();
+  XLOGF(INFO, "Destructor StorageServer");
+}
+
+Result<Void> StorageServer::beforeStart() {
+  RETURN_AND_LOG_ON_ERROR(addSerdeService(std::make_unique<StorageService>(components_.storageOperator), true));
+  RETURN_AND_LOG_ON_ERROR(addSerdeService(std::make_unique<core::CoreService>()));
+  groups().front()->setCoroutinesPoolGetter([this](const serde::MessagePacket<> &packet) -> DynamicCoroutinesPool & {
+    switch (packet.serviceId) {
+      case StorageSerde<>::kServiceID:
+        return components_.getCoroutinesPool(packet.methodId);
+      default:
+        return components_.defaultPool;
+    }
+  });
+  RETURN_AND_LOG_ON_ERROR(components_.start(appInfo(), tpg()));
+  return Void{};
+}
+
+Result<Void> StorageServer::beforeStop() {
+  components_.reliableUpdate.beforeStop();
+  components_.reliableForwarding.beforeStop();
+  return Void{};
+}
+
+Result<Void> StorageServer::afterStop() {
+  RETURN_AND_LOG_ON_ERROR(components_.stopAndJoin(tpg().procThreadPool()));
+  return Void{};
+}
+
+hf3fs::Result<Void> StorageServer::start(const flat::AppInfo &info,
+                                         std::unique_ptr<::hf3fs::net::Client> client,
+                                         std::shared_ptr<::hf3fs::client::MgmtdClient> mgmtdClient) {
+  components_.netClient = std::move(client);
+  components_.mgmtdClient = std::make_unique<hf3fs::client::MgmtdClientForServer>(std::move(mgmtdClient));
+  return net::Server::start(info);
+}
+
+}  // namespace hf3fs::storage