rawalcher/short-link-backend
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

sqlite database with methods added

Browse source
This commit is contained in:
rawalcher 2024-12-29 17:05:55 +01:00
parent 897310fe28
commit cbfc0b563c
3 changed files with 105 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

36
CMakeLists.txt
View file

@ -1,5 +1,4 @@
cmake_minimum_required(VERSION 3.16)
cmake_policy(SET CMP0167 OLD)
project(short-link VERSION 1.0 LANGUAGES CXX)
@ -9,17 +8,20 @@ set(CMAKE_CXX_STANDARD_REQUIRED True)
set(BUILD_SHARED_LIBS OFF)
include_directories(${PROJECT_SOURCE_DIR}/includes)
add_executable(Application src/main.cpp
src/http_connection.cpp
src/http_server.cpp
src/request_handler.cpp
add_executable(Application
src/main.cpp
src/http_connection.cpp
src/http_server.cpp
src/request_handler.cpp
src/database_connection.cpp
)
add_custom_target(copy_frontend ALL
COMMAND ${CMAKE_COMMAND} -E copy_directory ${PROJECT_SOURCE_DIR}/frontend ${CMAKE_BINARY_DIR}/frontend
COMMAND ${CMAKE_COMMAND} -E copy_directory ${PROJECT_SOURCE_DIR}/frontend ${CMAKE_BINARY_DIR}/frontend
)
find_package(Boost REQUIRED filesystem system)
find_package(Boost REQUIRED COMPONENTS filesystem system)
if(Boost_FOUND)
include_directories(${Boost_INCLUDE_DIRS})
@ -27,4 +29,22 @@ if(Boost_FOUND)
target_link_libraries(Application PRIVATE Boost::filesystem Boost::system)
else()
message(FATAL_ERROR "Boost not found!")
endif()
endif()
find_package(SOCI REQUIRED sqlite3)
if(SOCI_FOUND)
include_directories(${SOCI_INCLUDE_DIRS})
target_link_libraries(Application PRIVATE soci_core soci_sqlite3)
else()
message(FATAL_ERROR "SOCI not found!")
endif()
find_package(SQLite3 REQUIRED)
if(SQLite3_FOUND)
include_directories(${SQLite3_INCLUDE_DIRS})
target_link_libraries(Application PRIVATE SQLite::SQLite3)
else()
message(FATAL_ERROR "SQLite3 not found!")
endif()

31
includes/database_service.hpp
View file

@ -1,35 +1,36 @@
#ifndef DATABASE_SERVICE_HPP
#define DATABASE_SERVICE_HPP
#include <boost/asio/io_context.hpp>
#include <boost/system/detail/error_code.hpp>
#include <cstddef>
#include <boost/redis/connection.hpp>
#include <string>
#include <functional>
#include <memory>
#include <mutex>
#include <memory>
#include <vector>
#include <soci/soci.h>
#include <soci/sqlite3/soci-sqlite3.h>
class DatabaseService {
public:
static DatabaseService& getInstance(boost::asio::io_context& io_context, std::size_t pool_size);
void asyncSetKey(const std::string& key, const std::string& value, std::function<void(boost::system::error_code)> callback);
void asyncGetKey(const std::string& key, std::function<void(boost::system::error_code, std::string)> callback); //we could use a varient here for the result
static DatabaseService& getInstance(const std::string& db_path, std::size_t pool_size);
void shortenURL(const std::string& longURL, std::function<void(std::error_code, std::string)> callback);
void getLongURL(const std::string& shortCode, std::function<void(std::error_code, std::string)> callback);
private:
DatabaseService(boost::asio::io_context& io_context, std::size_t pool_size = 4);
private:
DatabaseService(const std::string& db_path, std::size_t pool_size = 4);
~DatabaseService();
DatabaseService(DatabaseService const&);
void operator=(DatabaseService const&);
std::shared_ptr<boost::redis::connection> getConnection();
std::shared_ptr<soci::session> getConnection();
std::string generateShortUUID();
boost::asio::io_context& io_context_;
std::string db_path_;
std::size_t pool_size_;
std::vector<std::shared_ptr<boost::redis::connection>> connection_pool_;
std::size_t current_index_ = 0; //index used for round robin selection
std::vector<std::shared_ptr<soci::session>> connection_pool_;
std::size_t current_index_ = 0; // Index used for round-robin selection
static DatabaseService* INSTANCE;
static std::mutex singleton_mutex;
};
#endif
#endif

71
src/database_connection.cpp
View file

@ -1,37 +1,46 @@
#include "../includes/database_service.hpp"
#include <boost/redis/connection.hpp>
#include <cstddef>
#include <memory>
#include <boost/uuid/uuid.hpp>
#include <boost/uuid/uuid_generators.hpp>
#include <boost/uuid/uuid_io.hpp>
#include <iostream>
#include <mutex>
#include <stdexcept>
std::mutex DatabaseService::singleton_mutex;
DatabaseService* DatabaseService::INSTANCE = nullptr;
DatabaseService& DatabaseService::getInstance(boost::asio::io_context &io_context, std::size_t pool_size) {
DatabaseService& DatabaseService::getInstance(const std::string& db_path, std::size_t pool_size) {
std::lock_guard<std::mutex> lock(singleton_mutex);
if (!INSTANCE) {
INSTANCE = new DatabaseService(io_context, pool_size);
INSTANCE = new DatabaseService(db_path, pool_size);
}
return *INSTANCE;
}
DatabaseService::DatabaseService(boost::asio::io_context& io_context, std::size_t pool_size) : io_context_(io_context), pool_size_(pool_size), current_index_(0) {
DatabaseService::DatabaseService(const std::string& db_path, std::size_t pool_size)
: db_path_(db_path), pool_size_(pool_size), current_index_(0) {
if (pool_size_ <= 0) {
throw std::invalid_argument("Connection pool size must be greater than zero.");
}
for (std::size_t i = 0; i < pool_size_; ++i) {
auto conn = std::make_shared<boost::redis::connection>(io_context_);
connection_pool_.emplace_back(conn);
auto session = std::make_shared<soci::session>(soci::sqlite3, db_path_);
connection_pool_.emplace_back(session);
}
// Ensure the table exists
auto conn = getConnection();
*conn << "CREATE TABLE IF NOT EXISTS urls ("
"short_code CHAR(8) PRIMARY KEY, "
"original_url TEXT NOT NULL UNIQUE, "
"created_at INTEGER DEFAULT CURRENT_TIMESTAMP);";
}
DatabaseService::~DatabaseService() {
connection_pool_.clear();
}
std::shared_ptr<boost::redis::connection> DatabaseService::getConnection() {
std::shared_ptr<soci::session> DatabaseService::getConnection() {
std::lock_guard<std::mutex> lock(singleton_mutex);
if (connection_pool_.empty()) {
throw std::runtime_error("Connection pool is empty.");
@ -39,5 +48,47 @@ std::shared_ptr<boost::redis::connection> DatabaseService::getConnection() {
auto conn = connection_pool_[current_index_];
current_index_ = (current_index_ + 1) % pool_size_;
return conn;
return conn;
}
std::string DatabaseService::generateShortUUID() {
boost::uuids::uuid uuid = boost::uuids::random_generator()();
std::string uuidStr = to_string(uuid);
return uuidStr.substr(0, 8);
}
void DatabaseService::shortenURL(const std::string& longURL, std::function<void(std::error_code, std::string)> callback) {
try {
auto conn = getConnection();
std::string shortCode;
*conn << "SELECT short_code FROM urls WHERE original_url = :long_url", soci::use(longURL), soci::into(shortCode);
if (!shortCode.empty()) {
callback({}, shortCode);
return;
}
shortCode = generateShortUUID();
*conn << "INSERT INTO urls (short_code, original_url) VALUES (:short_code, :long_url)",
soci::use(shortCode), soci::use(longURL);
callback({}, shortCode);
} catch (const std::exception& e) {
callback(std::make_error_code(std::errc::io_error), "");
}
}
void DatabaseService::getLongURL(const std::string& shortCode, std::function<void(std::error_code, std::string)> callback) {
try {
auto conn = getConnection();
std::string longURL;
*conn << "SELECT original_url FROM urls WHERE short_code = :short_code", soci::use(shortCode), soci::into(longURL);
if (!longURL.empty()) {
callback({}, longURL);
} else {
callback(std::make_error_code(std::errc::no_such_file_or_directory), "");
}
} catch (const std::exception& e) {
callback(std::make_error_code(std::errc::io_error), "");
}
}