use opentelemetry::{
    trace::{Span, SpanKind, Status, Tracer as _},
    KeyValue,
};
use opentelemetry_semantic_conventions as semcov;
use rand::{thread_rng, Rng};
use std::{error::Error, time::Duration};

#[tokio::main]
async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
    env_logger::init();

    let tracer = opentelemetry_application_insights::new_pipeline_from_env()?
        .with_client(reqwest::Client::new())
        .with_live_metrics(true)
        .install_batch(opentelemetry_sdk::runtime::Tokio);

    print!("Simulating requests. Press Ctrl+C to stop.");

    let mut rng = thread_rng();
    loop {
        let success = rng.gen_ratio(9, 10);
        let _request = tracer
            .span_builder("request")
            .with_kind(SpanKind::Server)
            .with_status(if success {
                Status::Ok
            } else {
                Status::error("")
            })
            .with_attributes(vec![
                KeyValue::new(semcov::trace::HTTP_REQUEST_METHOD, "GET"),
                KeyValue::new(semcov::trace::URL_SCHEME, "https"),
                KeyValue::new(semcov::trace::URL_PATH, "/hello/world"),
                KeyValue::new(semcov::trace::URL_QUERY, "name=marry"),
                KeyValue::new(semcov::trace::HTTP_RESPONSE_STATUS_CODE, 200),
            ])
            .start(&tracer);
        {
            let mut db = tracer
                .span_builder("db")
                .with_kind(SpanKind::Client)
                .with_status(if success {
                    Status::Ok
                } else {
                    Status::error("")
                })
                .start(&tracer);
            if !success {
                let err: Box<dyn std::error::Error> = "An error".into();
                db.record_error(err.as_ref());
            }

            tokio::time::sleep(Duration::from_millis(5)).await;
        }

        tokio::time::sleep(Duration::from_millis(100)).await;
    }
}