init

src/DatabaseManager.cpp

@@ -0,0 +1,152 @@
+#include "DatabaseManager.h"
+#include "Logger.h"
+#include <boost/asio/bind_executor.hpp>
+#include <boost/asio/use_awaitable.hpp>
+#include <boost/mysql.hpp>
+
+DatabaseManager &DatabaseManager::GetInstance() {
+  static DatabaseManager instance;
+  return instance;
+}
+
+bool DatabaseManager::Init(boost::asio::io_context &io_context,
+                           const std::string &host, uint16_t port,
+                           const std::string &user, const std::string &password,
+                           const std::string &db) {
+  try {
+    strand_ = std::make_unique<
+        boost::asio::strand<boost::asio::io_context::executor_type>>(
+        io_context.get_executor());
+
+    boost::asio::ip::tcp::resolver resolver(io_context.get_executor());
+    auto endpoints = resolver.resolve(host, std::to_string(port));
+
+    boost::mysql::handshake_params params(user, password, db);
+    static boost::asio::ssl::context ssl_ctx(
+        boost::asio::ssl::context::tlsv12_client);
+
+    conn_ = std::make_unique<boost::mysql::tcp_ssl_connection>(
+        io_context.get_executor(), ssl_ctx);
+    conn_->connect(*endpoints.begin(), params);
+
+    Logger::Log("Database 연결 성공 (AsyncLock 적용): ", host, ":", port);
+    return true;
+  } catch (const std::exception &e) {
+    Logger::Log("Database 연결 실패: ", e.what());
+    return false;
+  }
+}
+
+// 비동기 락 획득 (이미 락이 걸려있으면 대기열에 들어감)
+boost::asio::awaitable<void> DatabaseManager::Lock() {
+  auto executor = co_await boost::asio::this_coro::executor;
+
+  // 이 체크 로직 자체가 strand 위에서 안전하게 돌아야 함
+  co_await boost::asio::post(
+      boost::asio::bind_executor(*strand_, boost::asio::use_awaitable));
+
+  if (!is_locked_) {
+    is_locked_ = true;
+    // 즉시 획득
+    co_return;
+  }
+
+  // 이미 잠겨있으면 타이머를 만들어서 대기열에 추가 (비동기 대기)
+  auto timer = std::make_shared<boost::asio::steady_timer>(
+      executor, std::chrono::steady_clock::time_point::max());
+  waiting_queue_.push(timer);
+
+  try {
+    co_await timer->async_wait(boost::asio::use_awaitable);
+  } catch (...) {
+    // 타이머 취소(Unlock에서 호출함)되면 성공적으로 락을 획득한 것으로 간주
+  }
+  co_return;
+}
+
+// 비동기 락 해제 (대기열에 다음 사람이 있으면 깨워줌)
+void DatabaseManager::Unlock() {
+  boost::asio::dispatch(*strand_, [this]() {
+    if (waiting_queue_.empty()) {
+      is_locked_ = false;
+    } else {
+      auto next = waiting_queue_.front();
+      waiting_queue_.pop();
+      // 다음 대기자를 깨움
+      next->cancel();
+    }
+  });
+}
+
+boost::asio::awaitable<DatabaseManager::UserData>
+DatabaseManager::LoadUser(std::string nickname) {
+  // 줄 서기
+  co_await Lock();
+
+  try {
+    if (!conn_) {
+      Unlock();
+      co_return UserData{nickname, 10000, 0};
+    }
+
+    boost::mysql::results result;
+    boost::mysql::statement stmt = co_await conn_->async_prepare_statement(
+        "SELECT username, gold, sword_level FROM users WHERE username = ?",
+        boost::asio::use_awaitable);
+    co_await conn_->async_execute(stmt.bind(nickname), result,
+                                  boost::asio::use_awaitable);
+
+    if (result.rows().empty()) {
+      boost::mysql::statement ins_stmt =
+          co_await conn_->async_prepare_statement(
+              "INSERT INTO users (username, gold, sword_level) VALUES (?, "
+              "10000, 0)",
+              boost::asio::use_awaitable);
+      co_await conn_->async_execute(ins_stmt.bind(nickname), result,
+                                    boost::asio::use_awaitable);
+      Logger::Log("새로운 유저 탄생 (DB): ", nickname);
+      Unlock();
+      co_return UserData{nickname, 10000, 0};
+    } else {
+      auto row = result.rows().at(0);
+      UserData data;
+      data.nickname = row.at(0).as_string();
+      data.gold = row.at(1).as_uint64();
+      data.swordLevel = (uint32_t)row.at(2).as_int64();
+      Unlock();
+      co_return data;
+    }
+  } catch (const std::exception &e) {
+    Logger::Log("LoadUser 에러: ", e.what());
+    Unlock();
+    co_return UserData{nickname, 10000, 0};
+  }
+}
+
+boost::asio::awaitable<void> DatabaseManager::SaveUser(std::string nickname,
+                                                       uint64_t gold,
+                                                       uint32_t swordLevel) {
+  // 줄 서기
+  co_await Lock();
+
+  try {
+    if (!conn_) {
+      Unlock();
+      co_return;
+    }
+
+    boost::mysql::results result;
+    boost::mysql::statement stmt = co_await conn_->async_prepare_statement(
+        "UPDATE users SET gold = ?, sword_level = ? WHERE username = ?",
+        boost::asio::use_awaitable);
+    co_await conn_->async_execute(
+        stmt.bind(gold, (int64_t)swordLevel, nickname), result,
+        boost::asio::use_awaitable);
+  } catch (const std::exception &e) {
+    Logger::Log("SaveUser 에러: ", e.what());
+  }
+
+  // 다음 사람 들어오세요
+  Unlock();
+  co_return;
+}