#include "usdt_sample_lib1/lib1.h"

// std
#include <atomic>
#include <chrono>
#include <iostream>
#include <thread>

// usdt_sample_lib1
#include "folly/tracing/StaticTracepoint.h"

// When using systemtap-sdt-devel, the following file should be included:
#include "lib1_sdt.h"

OperationRequest::OperationRequest(const std::string& input_)
    : _input(input_)
{
}

OperationResponse::OperationResponse(const std::string& output_)
    : _output(output_)
{
}

OperationProvider::OperationProvider(std::uint32_t minLatencyMs_, std::uint32_t maxLatencyMs_)
    : _gen(std::random_device()())
    , _dis(minLatencyMs_, maxLatencyMs_)
{
}

std::shared_future<OperationResponse> OperationProvider::executeAsync(const OperationRequest& request)
{
    static std::atomic<std::uint64_t> operationIdCounter(0);
    std::uint64_t operationId = operationIdCounter++;

    // We create two probes at the same location. One created using the FOLLY_SDT macro, one created using dtrace.
    FOLLY_SDT(usdt_sample_lib1, operation_start, operationId, request.input().c_str());
    if (USDT_SAMPLE_LIB1_SDT_OPERATION_START_SDT_ENABLED()) {
        //std::cout << "operation_start probe enabled." << std::endl;
        USDT_SAMPLE_LIB1_SDT_OPERATION_START_SDT(operationId, request.input().c_str());
    }

    auto latencyMs = _dis(_gen);

    return std::async(std::launch::async, [latencyMs, operationId, request]() {
        std::this_thread::sleep_for(std::chrono::milliseconds(latencyMs));

        auto output = std::string("resp_") + request.input();
        OperationResponse response(output);

        // We create two probes at the same location. One created using the FOLLY_SDT macro, one created using dtrace.
        FOLLY_SDT(usdt_sample_lib1, operation_end, operationId, response.output().c_str());
        if (USDT_SAMPLE_LIB1_SDT_OPERATION_END_SDT_ENABLED()) {
            //std::cout << "operation_end probe enabled." << std::endl;
            USDT_SAMPLE_LIB1_SDT_OPERATION_END_SDT(operationId, response.output().c_str());
        }

        return response;
    });
}