35 lines
1.1 KiB
Rust
35 lines
1.1 KiB
Rust
use anyhow::{anyhow, bail, Context, Result};
|
|
use chrono::{DateTime, Utc};
|
|
use reqwest::Client;
|
|
use serde_json::Value;
|
|
use std::collections::HashMap;
|
|
|
|
use crate::models::{
|
|
EnergyReading, HaReading, InfluxDbSettings, NetworkReading, Reading, ZoneReading,
|
|
};
|
|
|
|
const DEVICE_MEASUREMENT: &str = "gree_device";
|
|
const ZONE_MEASUREMENT: &str = "gree_zone";
|
|
const HA_MEASUREMENT: &str = "gree_ha";
|
|
const ENERGY_MEASUREMENT: &str = "gree_energy";
|
|
const NETWORK_MEASUREMENT: &str = "gree_network";
|
|
|
|
// Functional source split intentionally keeps items in the existing module namespace.
|
|
include!("influxdb/write.rs");
|
|
include!("influxdb/query.rs");
|
|
include!("influxdb/codec.rs");
|
|
|
|
pub async fn test_connection(client: &Client, settings: &InfluxDbSettings) -> Result<()> {
|
|
validate(settings)?;
|
|
if settings.version == "1" {
|
|
query_v1(client, settings, "SHOW MEASUREMENTS LIMIT 1").await?;
|
|
} else {
|
|
let query = format!(
|
|
"from(bucket: {}) |> range(start: -1m) |> limit(n: 1)",
|
|
flux_string(&settings.bucket)
|
|
);
|
|
query_v2(client, settings, &query).await?;
|
|
}
|
|
Ok(())
|
|
}
|