Verified Commit 57c2a1ec authored by Alberto Miranda's avatar Alberto Miranda ♨️
Browse files

Update ADM_get_statistics()

parent 525893ef
Loading
Loading
Loading
Loading
+15 −30
Original line number Diff line number Diff line
#include <fmt/format.h>
#include <engine.hpp>
#include <admire.hpp>

int
main(int argc, char* argv[]) {
@@ -12,42 +12,27 @@ main(int argc, char* argv[]) {
        exit(EXIT_FAILURE);
    }

    scord::network::rpc_client rpc_client{"tcp"};
    rpc_client.register_rpcs();
    admire::server server{"tcp", argv[1]};

    auto endp = rpc_client.lookup(argv[1]);
    ADM_job_handle_t job{};
    ADM_job_stats_t* stats = nullptr;
    ADM_return_t ret = ADM_SUCCESS;

    fmt::print(
            stdout,
            "Calling ADM_get_statistics remote procedure on {} with job id {} and job step {} ...\n",
            argv[1], argv[2], argv[3]);
    ADM_get_statistics_in_t in;
    try {
        in.job_id = std::stoi(argv[2]);
    } catch(const std::exception& e) {
        fmt::print(stderr, "ERROR: Incorrect input type. Please try again.\n");
        exit(EXIT_FAILURE);
    }
    try {
        in.job_step = std::stoi(argv[3]);
        ret = admire::get_statistics(server, job, &stats);
    } catch(const std::exception& e) {
        fmt::print(stderr, "ERROR: Incorrect input type. Please try again.\n");
        fmt::print(stderr, "FATAL: ADM_cancel_transfer() failed: {}\n",
                   e.what());
        exit(EXIT_FAILURE);
    }

    ADM_get_statistics_out_t out;

    endp.call("ADM_get_statistics", &in, &out);


    if(out.ret < 0) {
        fmt::print(
                stdout,
                "ADM_get_statistics remote procedure not completed successfully\n");
    if(ret != ADM_SUCCESS) {
        fmt::print(stdout,
                   "ADM_cancel_transfer() remote procedure not completed "
                   "successfully\n");
        exit(EXIT_FAILURE);
    } else {
        fmt::print(
                stdout,
                "ADM_get_statistics remote procedure completed successfully\n");
    }

    fmt::print(stdout, "ADM_cancel_transfer() remote procedure completed "
                       "successfully\n");
}
+21 −1
Original line number Diff line number Diff line
@@ -534,7 +534,27 @@ get_statistics(const server& srv, ADM_job_handle_t job,
    (void) job;
    (void) stats;

    return ADM_OTHER_ERROR;
    scord::network::rpc_client rpc_client{srv.m_protocol};

    rpc_client.register_rpcs();

    auto endp = rpc_client.lookup(srv.m_address);

    LOGGER_INFO("ADM_get_statistics(...)");

    // FIXME: change RPC fields to ADM_transfer_handle_t
    ADM_get_statistics_in_t in{};
    ADM_get_statistics_out_t out;

    endp.call("ADM_get_statistics", &in, &out);

    if(out.ret < 0) {
        LOGGER_ERROR("ADM_get_statistics() = {}", out.ret);
        return static_cast<ADM_return_t>(out.ret);
    }

    LOGGER_INFO("ADM_get_statistics() = {}", ADM_SUCCESS);
    return ADM_SUCCESS;
}

} // namespace admire