status_code_provider_impl.hpp

// status_code_provider_impl.hpp
// Created by greg on 2/19/25.
// Copyright (C) 2025 Presage Security, Inc.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 3 of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 
// === standard library includes (if any) ===
// === third-party includes (if any) ===
// === local includes (if any) ===
#include "status_code_provider.hpp"
 
#pragma once
 
namespace presage::smartspectra::grpc_bindings {
 
template<container::settings::IntegrationMode TIntegrationMode>
StatusCodeProvider<TIntegrationMode>::StatusCodeProvider(std::shared_ptr<BackgroundContainer> container) {
    SwapContainer(container);
}
 
template<container::settings::IntegrationMode TIntegrationMode>
void StatusCodeProvider<TIntegrationMode>::SwapContainer(std::shared_ptr<BackgroundContainer> new_container) {
    this->container.reset();
    this->container = std::move(new_container);
    if (this->container == nullptr) {
        this->Finish(grpc::Status(
            grpc::StatusCode::FAILED_PRECONDITION,
            "Physiology Preprocessing is not running, unable to get status codes."
        ));
        return;
    }
    auto status = this->container->SetOnStatusChange(
        [this](presage::physiology::StatusValue status_output) {
            this->EnqueueWrite(status_output);
            return absl::OkStatus();
        }
    );
 
    if (!status.ok()) {
        this->Finish(rpc::abslStatusToGrpc(status));
    } else {
        // Send the current status immediately if this is after initial construction
        // This handles the case when the container is swapped after initial setup
        auto status_code = this->container->GetStatusCode();
        auto current_time = std::chrono::system_clock::now().time_since_epoch();
        int64_t timestamp = std::chrono::duration_cast<std::chrono::microseconds>(current_time).count();
        this->gap_status = physiology::BuildStatusValue(status_code, timestamp);
        this->StartWrite(&this->gap_status);
    }
}
 
} // namespace presage::smartspectra::grpc_bindings