prefetch_repository.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include "kcenon/pacs/client/prefetch_types.h"
 
#include <kcenon/common/patterns/result.h>
 
#include <memory>
#include <optional>
#include <string>
#include <string_view>
#include <vector>
 
#ifdef PACS_WITH_DATABASE_SYSTEM
 
#include "kcenon/pacs/storage/pacs_database_adapter.h"
 
namespace kcenon::pacs::storage {
 
template <typename T>
using Result = kcenon::common::Result<T>;
 
using VoidResult = kcenon::common::VoidResult;
 
struct prefetch_rule_query_options {
    std::optional<bool> enabled_only;              
    std::optional<client::prefetch_trigger> trigger;  
    size_t limit{100};                             
    size_t offset{0};                              
};
 
struct prefetch_history_query_options {
    std::optional<std::string> patient_id;         
    std::optional<std::string> rule_id;            
    std::optional<std::string> status;             
    size_t limit{100};                             
    size_t offset{0};                              
};
 
class prefetch_repository {
public:
    // =========================================================================
    // Construction / Destruction
    // =========================================================================
 
    explicit prefetch_repository(std::shared_ptr<pacs_database_adapter> db);
 
    ~prefetch_repository();
 
    // Non-copyable, movable
    prefetch_repository(const prefetch_repository&) = delete;
    auto operator=(const prefetch_repository&) -> prefetch_repository& = delete;
    prefetch_repository(prefetch_repository&&) noexcept;
    auto operator=(prefetch_repository&&) noexcept -> prefetch_repository&;
 
    // =========================================================================
    // Rule CRUD Operations
    // =========================================================================
 
    [[nodiscard]] auto save_rule(const client::prefetch_rule& rule) -> VoidResult;
 
    [[nodiscard]] auto find_rule_by_id(std::string_view rule_id) const
        -> std::optional<client::prefetch_rule>;
 
    [[nodiscard]] auto find_rule_by_pk(int64_t pk) const
        -> std::optional<client::prefetch_rule>;
 
    [[nodiscard]] auto find_rules(const prefetch_rule_query_options& options = {}) const
        -> std::vector<client::prefetch_rule>;
 
    [[nodiscard]] auto find_enabled_rules() const
        -> std::vector<client::prefetch_rule>;
 
    [[nodiscard]] auto remove_rule(std::string_view rule_id) -> VoidResult;
 
    [[nodiscard]] auto rule_exists(std::string_view rule_id) const -> bool;
 
    // =========================================================================
    // Rule Statistics
    // =========================================================================
 
    [[nodiscard]] auto increment_triggered(std::string_view rule_id) -> VoidResult;
 
    [[nodiscard]] auto increment_studies_prefetched(
        std::string_view rule_id, size_t count = 1) -> VoidResult;
 
    [[nodiscard]] auto enable_rule(std::string_view rule_id) -> VoidResult;
 
    [[nodiscard]] auto disable_rule(std::string_view rule_id) -> VoidResult;
 
    // =========================================================================
    // History Operations
    // =========================================================================
 
    [[nodiscard]] auto save_history(const client::prefetch_history& history) -> VoidResult;
 
    [[nodiscard]] auto find_history(const prefetch_history_query_options& options = {}) const
        -> std::vector<client::prefetch_history>;
 
    [[nodiscard]] auto is_study_prefetched(std::string_view study_uid) const -> bool;
 
    [[nodiscard]] auto count_completed_today() const -> size_t;
 
    [[nodiscard]] auto count_failed_today() const -> size_t;
 
    [[nodiscard]] auto update_history_status(
        std::string_view study_uid,
        std::string_view status) -> VoidResult;
 
    [[nodiscard]] auto cleanup_old_history(std::chrono::hours max_age)
        -> Result<size_t>;
 
    // =========================================================================
    // Statistics
    // =========================================================================
 
    [[nodiscard]] auto rule_count() const -> size_t;
 
    [[nodiscard]] auto enabled_rule_count() const -> size_t;
 
    [[nodiscard]] auto history_count() const -> size_t;
 
    // =========================================================================
    // Database Information
    // =========================================================================
 
    [[nodiscard]] auto is_valid() const noexcept -> bool;
 
    [[nodiscard]] auto initialize_tables() -> VoidResult;
 
private:
    // =========================================================================
    // Private Implementation
    // =========================================================================
 
    [[nodiscard]] auto map_row_to_rule(const database_row& row) const
        -> client::prefetch_rule;
    [[nodiscard]] auto map_row_to_history(const database_row& row) const
        -> client::prefetch_history;
    [[nodiscard]] static auto serialize_modalities(
        const std::vector<std::string>& modalities) -> std::string;
    [[nodiscard]] static auto deserialize_modalities(
        std::string_view json) -> std::vector<std::string>;
    [[nodiscard]] static auto serialize_node_ids(
        const std::vector<std::string>& node_ids) -> std::string;
    [[nodiscard]] static auto deserialize_node_ids(
        std::string_view json) -> std::vector<std::string>;
    [[nodiscard]] auto parse_timestamp(const std::string& str) const
        -> std::chrono::system_clock::time_point;
    [[nodiscard]] auto format_timestamp(
        std::chrono::system_clock::time_point tp) const -> std::string;
 
    std::shared_ptr<pacs_database_adapter> db_;
};
 
}  // namespace kcenon::pacs::storage
 
#else  // !PACS_WITH_DATABASE_SYSTEM
 
// =============================================================================
// Legacy SQLite Interface
// =============================================================================
 
struct sqlite3;
 
namespace kcenon::pacs::storage {
 
template <typename T>
using Result = kcenon::common::Result<T>;
 
using VoidResult = kcenon::common::VoidResult;
 
struct prefetch_rule_query_options {
    std::optional<bool> enabled_only;
    std::optional<client::prefetch_trigger> trigger;
    size_t limit{100};
    size_t offset{0};
};
 
struct prefetch_history_query_options {
    std::optional<std::string> patient_id;
    std::optional<std::string> rule_id;
    std::optional<std::string> status;
    size_t limit{100};
    size_t offset{0};
};
 
class prefetch_repository {
public:
    explicit prefetch_repository(sqlite3* db);
    ~prefetch_repository();
 
    prefetch_repository(const prefetch_repository&) = delete;
    auto operator=(const prefetch_repository&) -> prefetch_repository& = delete;
    prefetch_repository(prefetch_repository&&) noexcept;
    auto operator=(prefetch_repository&&) noexcept -> prefetch_repository&;
 
    // Rule Operations
    [[nodiscard]] auto save_rule(const client::prefetch_rule& rule) -> VoidResult;
    [[nodiscard]] auto find_rule_by_id(std::string_view rule_id) const
        -> std::optional<client::prefetch_rule>;
    [[nodiscard]] auto find_rule_by_pk(int64_t pk) const
        -> std::optional<client::prefetch_rule>;
    [[nodiscard]] auto find_rules(const prefetch_rule_query_options& options = {}) const
        -> std::vector<client::prefetch_rule>;
    [[nodiscard]] auto find_enabled_rules() const
        -> std::vector<client::prefetch_rule>;
    [[nodiscard]] auto remove_rule(std::string_view rule_id) -> VoidResult;
    [[nodiscard]] auto rule_exists(std::string_view rule_id) const -> bool;
 
    // Rule Statistics
    [[nodiscard]] auto increment_triggered(std::string_view rule_id) -> VoidResult;
    [[nodiscard]] auto increment_studies_prefetched(
        std::string_view rule_id, size_t count = 1) -> VoidResult;
    [[nodiscard]] auto enable_rule(std::string_view rule_id) -> VoidResult;
    [[nodiscard]] auto disable_rule(std::string_view rule_id) -> VoidResult;
 
    // History Operations
    [[nodiscard]] auto save_history(const client::prefetch_history& history) -> VoidResult;
    [[nodiscard]] auto find_history(const prefetch_history_query_options& options = {}) const
        -> std::vector<client::prefetch_history>;
    [[nodiscard]] auto is_study_prefetched(std::string_view study_uid) const -> bool;
    [[nodiscard]] auto count_completed_today() const -> size_t;
    [[nodiscard]] auto count_failed_today() const -> size_t;
    [[nodiscard]] auto update_history_status(
        std::string_view study_uid, std::string_view status) -> VoidResult;
    [[nodiscard]] auto cleanup_old_history(std::chrono::hours max_age)
        -> Result<size_t>;
 
    // Statistics
    [[nodiscard]] auto rule_count() const -> size_t;
    [[nodiscard]] auto enabled_rule_count() const -> size_t;
    [[nodiscard]] auto history_count() const -> size_t;
 
    // Database Information
    [[nodiscard]] auto is_valid() const noexcept -> bool;
    [[nodiscard]] auto initialize_tables() -> VoidResult;
 
private:
    [[nodiscard]] auto parse_rule_row(void* stmt) const -> client::prefetch_rule;
    [[nodiscard]] auto parse_history_row(void* stmt) const -> client::prefetch_history;
    [[nodiscard]] static auto serialize_modalities(
        const std::vector<std::string>& modalities) -> std::string;
    [[nodiscard]] static auto deserialize_modalities(
        std::string_view json) -> std::vector<std::string>;
    [[nodiscard]] static auto serialize_node_ids(
        const std::vector<std::string>& node_ids) -> std::string;
    [[nodiscard]] static auto deserialize_node_ids(
        std::string_view json) -> std::vector<std::string>;
 
    sqlite3* db_{nullptr};
};
 
}  // namespace kcenon::pacs::storage
 
#endif  // PACS_WITH_DATABASE_SYSTEM