mirror of
https://github.com/aaronpo97/the-biergarten-app.git
synced 2026-04-05 18:09:04 +00:00
98083ab40c
Introduce a pluggable web client interface and concrete CURL implementation: adds IWebClient, CURLWebClient, and CurlGlobalState (headers + curl_web_client.cpp). DataDownloader now accepts an IWebClient and delegates downloads. Add WikipediaService for cached Wikipedia summary lookups. Refactor SqliteDatabase to return full City records and update consumers accordingly. Improve JsonLoader to use batched transactions during streaming parses. Enhance LlamaGenerator with sampling options, increased token limits, JSON extraction/validation, and other parsing helpers. Modernize CMake: set policy/version, add project_options, simplify FetchContent usage (spdlog), require Boost components (program_options/json), list pipeline sources explicitly, and tweak post-build/memcheck targets. Update README to match implementation changes and new CLI/config conventions.
81 lines
2.2 KiB
C++
81 lines
2.2 KiB
C++
#pragma once
|
|
|
|
#include <mutex>
|
|
#include <sqlite3.h>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
struct Country {
|
|
/// @brief Country identifier from the source dataset.
|
|
int id;
|
|
/// @brief Country display name.
|
|
std::string name;
|
|
/// @brief ISO 3166-1 alpha-2 code.
|
|
std::string iso2;
|
|
/// @brief ISO 3166-1 alpha-3 code.
|
|
std::string iso3;
|
|
};
|
|
|
|
struct State {
|
|
/// @brief State or province identifier from the source dataset.
|
|
int id;
|
|
/// @brief State or province display name.
|
|
std::string name;
|
|
/// @brief State or province short code.
|
|
std::string iso2;
|
|
/// @brief Parent country identifier.
|
|
int countryId;
|
|
};
|
|
|
|
struct City {
|
|
/// @brief City identifier from the source dataset.
|
|
int id;
|
|
/// @brief City display name.
|
|
std::string name;
|
|
/// @brief Parent country identifier.
|
|
int countryId;
|
|
};
|
|
|
|
/// @brief Thread-safe SQLite wrapper for pipeline writes and readbacks.
|
|
class SqliteDatabase {
|
|
private:
|
|
sqlite3 *db = nullptr;
|
|
std::mutex dbMutex;
|
|
|
|
void InitializeSchema();
|
|
|
|
public:
|
|
/// @brief Closes the SQLite connection if initialized.
|
|
~SqliteDatabase();
|
|
|
|
/// @brief Opens the SQLite database at dbPath and creates schema objects.
|
|
void Initialize(const std::string &dbPath = ":memory:");
|
|
|
|
/// @brief Starts a database transaction for batched writes.
|
|
void BeginTransaction();
|
|
|
|
/// @brief Commits the active database transaction.
|
|
void CommitTransaction();
|
|
|
|
/// @brief Inserts a country row.
|
|
void InsertCountry(int id, const std::string &name, const std::string &iso2,
|
|
const std::string &iso3);
|
|
|
|
/// @brief Inserts a state row linked to a country.
|
|
void InsertState(int id, int countryId, const std::string &name,
|
|
const std::string &iso2);
|
|
|
|
/// @brief Inserts a city row linked to state and country.
|
|
void InsertCity(int id, int stateId, int countryId, const std::string &name,
|
|
double latitude, double longitude);
|
|
|
|
/// @brief Returns city records including parent country id.
|
|
std::vector<City> QueryCities();
|
|
|
|
/// @brief Returns countries with optional row limit.
|
|
std::vector<Country> QueryCountries(int limit = 0);
|
|
|
|
/// @brief Returns states with optional row limit.
|
|
std::vector<State> QueryStates(int limit = 0);
|
|
};
|