loki-rs 0.1.0

A simple database client for Grafana Loki.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use serde::{Serialize, Deserialize};
use std::collections::HashMap;
use surf::post;

//type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;

/// push a log to Loki
pub async fn push_log(address: String, stream_name: String, stream_value: String, log_messages: Vec<[String; 2]> ) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {

    // Struct holds the messages for a given logstream
    #[derive(Serialize, Deserialize, Debug)]
    struct LogPushStream {
        stream: HashMap<String, String>,
        values: Vec<[String; 2]>
    }

    // Container struct
    #[derive(Serialize, Deserialize, Debug)]
    struct LogPushStreamContainer {
        streams: Vec<LogPushStream>
    }

    // Build the log stream
    let log_stream = LogPushStream {
        stream: HashMap::from([(stream_name, stream_value)]),
        values: log_messages
    };

    let stream_container = LogPushStreamContainer {
        streams: vec![log_stream]
    };

    // Send the log stream to Loki as JSON
    let uri = format!("{}/loki/api/v1/push", address);
    let res = post(uri).body_json(&stream_container)?.await?;

    let success = res.status() as u8 /100 == 2;

    if !success {
        return Err(format!("Failed to push log to Loki: {}", res.status()).into());
    } else {
        Ok(())
    }

}