ae/ECAP_Server
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新

Browse Source
This commit is contained in:
种太阳的热带美短
2026-08-05 11:59:45 +08:00
parent 18e64dfe22
commit 587c1d6d01
22 changed files with 2517 additions and 3049 deletions
Download Patch File Download Diff File Expand all files Collapse all files
module/Local_Server/External_Database/global.h
+54 -41
View File
@@ -1,6 +1,4 @@
#pragma once
#include <string_view>
#include "Core/Base/JSON.h"
#include <SQLiteCpp/Database.h>
#include <asio/awaitable.hpp>
#include <cstddef>
@@ -14,7 +12,9 @@
#include <mutex>
#include <optional>
#include <string>
#include <string_view>
#include <vector>
#include "Core/Base/JSON.h"
class Global;
struct External_Database_Row {
std::map<std::string, std::optional<std::string>> columns;
@@ -26,8 +26,7 @@ struct External_Database_Row {
[[nodiscard]] std::optional<std::string> get(std::string_view column) const;
[[nodiscard]] Psc::JSON to_json() const;
};
using External_Database_Row_Map =
std::map<std::string, std::shared_ptr<External_Database_Row>>;
using External_Database_Row_Map = std::map<std::string, std::shared_ptr<External_Database_Row>>;
struct External_Resource_Status {
std::string name;
std::size_t row_count{};
@@ -46,18 +45,19 @@ public:
};
class External_Resources : public External_Resources_Data {
public:
External_Resources(std::string_view name, std::string_view cache_file_name,
std::string_view download_url, std::string_view description,
std::uint64_t update_interval_seconds);
External_Resources(std::string_view name, std::string_view cache_file_name, std::string_view download_url,
std::string_view description, std::uint64_t update_interval_seconds);
virtual ~External_Resources() = default;
[[nodiscard]] std::size_t row_count(SQLite::Database& db) const;
virtual void create_table(SQLite::Database& db) const = 0;
External_Resource_Status update(SQLite::Database& db, const std::filesystem::path& cache_pos, bool force_download, bool only_when_empty);
External_Resource_Status update(SQLite::Database& db, const std::filesystem::path& cache_pos, bool force_download,
bool only_when_empty);
External_Resource_Status import_offline(SQLite::Database& db, const std::filesystem::path& source_file);
External_Resource_Status clear_table(SQLite::Database& db, const std::filesystem::path& cache_pos);
[[nodiscard]] std::optional<External_Database_Row> query_one(SQLite::Database& db, const std::vector<std::string>& primary_key_values) const;
[[nodiscard]] std::optional<External_Database_Row>
query_one(SQLite::Database& db, const std::vector<std::string>& primary_key_values) const;
void init(const Psc::JSON* that_json) {
External_Resources_Data::from_base_json(that_json);
External_Resources_Data::from_base_json(that_json, false);
}
[[nodiscard]] Psc::JSON to_json() const {
return External_Resources_Data::to_base_json();
@@ -65,63 +65,76 @@ public:
std::string cache_file_name;
protected:
[[nodiscard]] virtual std::filesystem::path source_file(const std::filesystem::path& cache_pos) const;
virtual bool fetch(const std::filesystem::path& cache_pos,
bool force_download) const;
virtual std::size_t import_file(SQLite::Database& db,
const std::filesystem::path& source_file) const = 0;
virtual bool fetch(const std::filesystem::path& cache_pos, bool force_download) const;
virtual std::size_t import_file(SQLite::Database& db, const std::filesystem::path& source_file) const = 0;
[[nodiscard]] virtual const std::vector<std::string>& primary_key_columns() const = 0;
void mark_updated();
void clear_last_updated_at();
};
class External_Resources_Manager {
public:
using Status_Callback =
std::function<void(std::exception_ptr, External_Resource_Status)>;
using Status_List_Callback = std::function<void(
std::exception_ptr, std::vector<External_Resource_Status>)>;
using Row_Callback = std::function<void(
std::exception_ptr, std::optional<External_Database_Row>)>;
using Row_Map_Callback =
std::function<void(std::exception_ptr, External_Database_Row_Map)>;
using Status_Callback = std::function<void(std::exception_ptr, External_Resource_Status)>;
using Status_List_Callback = std::function<void(std::exception_ptr, std::vector<External_Resource_Status>)>;
using Row_Callback = std::function<void(std::exception_ptr, std::optional<External_Database_Row>)>;
using Row_Map_Callback = std::function<void(std::exception_ptr, External_Database_Row_Map)>;
External_Resources_Manager();
void init(const Psc::JSON* that_json);
void from_json(const Psc::JSON* that_json, bool not_exist_use_default_value);
[[nodiscard]] Psc::JSON to_base_json() const;
void server(Global* g);
void load_sqlite_db(std::string_view path);
std::vector<External_Resource_Status> refresh_external_databases(bool force_download = true);
External_Resource_Status refresh_external_database(std::string_view resource_name, bool force_download = true);
External_Resource_Status import_external_database(std::string_view resource_name, const std::filesystem::path& source_file);
External_Resource_Status import_external_database(std::string_view resource_name,
const std::filesystem::path& source_file);
External_Resource_Status clear_external_database_table(std::string_view resource_name);
[[nodiscard]] std::optional<External_Database_Row> query_external_database(std::string_view resource_name, const std::vector<std::string>& primary_key_values) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>> query_aircraft_external_databases(std::string_view icao24) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>> query_callsign_external_databases(const std::optional<std::string>& callsign) const;
[[nodiscard]] std::optional<External_Database_Row>
query_external_database(std::string_view resource_name, const std::vector<std::string>& primary_key_values) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>>
query_aircraft_external_databases(std::string_view icao24) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>>
query_callsign_external_databases(const std::optional<std::string>& callsign) const;
[[nodiscard]] std::vector<External_Resource_Status> external_database_status() const;
void async_refresh_external_databases(bool force_download, Status_List_Callback callback);
void async_refresh_external_database(std::string_view resource_name, bool force_download, Status_Callback callback);
void async_import_external_database(std::string_view resource_name, std::filesystem::path source_file, Status_Callback callback);
void async_import_external_database(std::string_view resource_name, std::filesystem::path source_file,
Status_Callback callback);
void async_clear_external_database_table(std::string_view resource_name, Status_Callback callback);
void async_query_external_database(std::string_view resource_name, std::vector<std::string> primary_key_values, Row_Callback callback) const;
void async_query_external_database(std::string_view resource_name, std::vector<std::string> primary_key_values,
Row_Callback callback) const;
void async_query_aircraft_external_databases(std::string_view icao24, Row_Map_Callback callback) const;
void async_query_callsign_external_databases(std::optional<std::string> callsign, Row_Map_Callback callback) const;
void async_external_database_status(Status_List_Callback callback) const;
[[nodiscard]] asio::awaitable<std::vector<External_Resource_Status>> refresh_external_databases_coro(bool force_download = true);
[[nodiscard]] asio::awaitable<External_Resource_Status> refresh_external_database_coro(std::string_view resource_name, bool force_download = true);
[[nodiscard]] asio::awaitable<External_Resource_Status> import_external_database_coro(std::string_view resource_name, std::filesystem::path source_file);
[[nodiscard]] asio::awaitable<External_Resource_Status> clear_external_database_table_coro(std::string_view resource_name);
[[nodiscard]] asio::awaitable<std::optional<External_Database_Row>> query_external_database_coro(std::string_view resource_name, std::vector<std::string> primary_key_values) const;
[[nodiscard]] asio::awaitable<std::shared_ptr<External_Database_Row_Map>> query_aircraft_external_databases_coro(std::string_view icao24) const;
[[nodiscard]] asio::awaitable<std::shared_ptr<External_Database_Row_Map>> query_callsign_external_databases_coro(std::optional<std::string> callsign) const;
[[nodiscard]] asio::awaitable<std::vector<External_Resource_Status>>
refresh_external_databases_coro(bool force_download = true);
[[nodiscard]] asio::awaitable<External_Resource_Status>
refresh_external_database_coro(std::string_view resource_name, bool force_download = true);
[[nodiscard]] asio::awaitable<External_Resource_Status>
import_external_database_coro(std::string_view resource_name, std::filesystem::path source_file);
[[nodiscard]] asio::awaitable<External_Resource_Status>
clear_external_database_table_coro(std::string_view resource_name);
[[nodiscard]] asio::awaitable<std::optional<External_Database_Row>>
query_external_database_coro(std::string_view resource_name, std::vector<std::string> primary_key_values) const;
[[nodiscard]] asio::awaitable<std::shared_ptr<External_Database_Row_Map>>
query_aircraft_external_databases_coro(std::string_view icao24) const;
[[nodiscard]] asio::awaitable<std::shared_ptr<External_Database_Row_Map>>
query_callsign_external_databases_coro(std::optional<std::string> callsign) const;
[[nodiscard]] asio::awaitable<std::vector<External_Resource_Status>> external_database_status_coro() const;
void register_external_resource(std::unique_ptr<External_Resources> external_res);
void initialize(SQLite::Database& db, std::filesystem::path cache_pos);
std::vector<External_Resource_Status> sync_missing(SQLite::Database& db);
std::vector<External_Resource_Status> refresh_all(SQLite::Database& db, bool force_download = true);
External_Resource_Status refresh_one(SQLite::Database& db, std::string_view resource_name, bool force_download = true);
External_Resource_Status import_offline(SQLite::Database& db, std::string_view resource_name, const std::filesystem::path& source_file);
External_Resource_Status refresh_one(SQLite::Database& db, std::string_view resource_name,
bool force_download = true);
External_Resource_Status import_offline(SQLite::Database& db, std::string_view resource_name,
const std::filesystem::path& source_file);
External_Resource_Status clear_table(SQLite::Database& db, std::string_view resource_name);
[[nodiscard]] std::optional<External_Database_Row> query_one(SQLite::Database& db, std::string_view resource_name, const std::vector<std::string>& primary_key_values) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>> query_aircraft(SQLite::Database& db, std::string_view icao24) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>> query_callsign(SQLite::Database& db, const std::optional<std::string>& callsign) const;
[[nodiscard]] std::optional<External_Database_Row>
query_one(SQLite::Database& db, std::string_view resource_name,
const std::vector<std::string>& primary_key_values) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>>
query_aircraft(SQLite::Database& db, std::string_view icao24) const;
[[nodiscard]] std::map<std::string, std::shared_ptr<External_Database_Row>>
query_callsign(SQLite::Database& db, const std::optional<std::string>& callsign) const;
[[nodiscard]] std::vector<External_Resource_Status> status(SQLite::Database& db) const;
std::string ecap_sqlite_path;
std::filesystem::path cache_pos_;