This commit is contained in:
2024-10-19 13:15:09 -07:00
commit 17b9e50f1f
12 changed files with 1623 additions and 0 deletions

18
server/Cargo.toml Normal file
View File

@@ -0,0 +1,18 @@
[package]
name = "server"
edition = "2021"
version = "0.1.0"
authors = ["Sergey <me@sergeysav.com>"]
[dependencies]
fern = "0.6.2"
log = "0.4.22"
prost = "0.13.3"
rand = "0.8.5"
tonic = { version = "0.12.3" }
tokio = { version = "1.40.0", features = ["rt-multi-thread"] }
chrono = "0.4.38"
[build-dependencies]
tonic-build = "0.12.3"

5
server/build.rs Normal file
View File

@@ -0,0 +1,5 @@
fn main() -> Result<(), Box<dyn std::error::Error>> {
tonic_build::compile_protos("proto/core.proto")?;
Ok(())
}

48
server/proto/core.proto Normal file
View File

@@ -0,0 +1,48 @@
syntax = "proto3";
package core;
enum TelemetryDataType {
FLOAT_32 = 0;
FLOAT_64 = 1;
}
message TelemetryValue {
oneof value {
float float_32 = 1;
float float_64 = 2;
}
}
message UUID {
bytes value = 1;
}
// UTC since UNIX
message Timestamp {
fixed64 secs = 1;
fixed32 nanos = 2;
}
message TelemetryDefinitionRequest {
string name = 1;
TelemetryDataType data_type = 2;
}
message TelemetryDefinitionResponse {
UUID uuid = 1;
}
message TelemetryItem {
UUID uuid = 1;
TelemetryValue value = 2;
Timestamp timestamp = 3;
}
message TelemetryInsertResponse {
}
service TelemetryService {
rpc NewTelemetry (TelemetryDefinitionRequest) returns (TelemetryDefinitionResponse);
rpc InsertTelemetry (stream TelemetryItem) returns (stream TelemetryInsertResponse);
}

74
server/src/lib.rs Normal file
View File

@@ -0,0 +1,74 @@
mod uuid;
pub mod core {
tonic::include_proto!("core");
}
use std::error::Error;
use std::pin::Pin;
use log::{trace};
use tokio::sync::mpsc;
use tonic::{Request, Response, Status, Streaming};
use tonic::codegen::tokio_stream::{Stream, StreamExt};
use tonic::codegen::tokio_stream::wrappers::ReceiverStream;
use tonic::transport::Server;
use core::telemetry_service_server::TelemetryService;
use crate::core::{TelemetryDefinitionRequest, TelemetryDefinitionResponse, TelemetryInsertResponse, TelemetryItem, Uuid};
use crate::core::telemetry_service_server::TelemetryServiceServer;
#[derive(Debug, Default)]
pub struct CoreTelemetryService {}
#[tonic::async_trait]
impl TelemetryService for CoreTelemetryService {
async fn new_telemetry(&self, request: Request<TelemetryDefinitionRequest>) -> Result<Response<TelemetryDefinitionResponse>, Status> {
trace!("TelemetryService::new_telemetry {:?}", request);
let reply = TelemetryDefinitionResponse {
uuid: Some(Uuid::random()),
};
Ok(Response::new(reply))
}
type InsertTelemetryStream = Pin<Box<dyn Stream<Item = Result<TelemetryInsertResponse, Status>> + Send>>;
async fn insert_telemetry(&self, request: Request<Streaming<TelemetryItem>>) -> Result<Response<Self::InsertTelemetryStream>, Status> {
trace!("TelemetryService::insert_telemetry {:?}", request);
let mut in_stream = request.into_inner();
let (tx, rx) = mpsc::channel(128);
tokio::spawn(async move {
while let Some(message) = in_stream.next().await {
match message {
Ok(tlm_item) => {
trace!("tlm_item {:?}", tlm_item);
tx
.send(Ok(TelemetryInsertResponse {}))
.await
.expect("working rx");
}
Err(err) => {
let _ = tx.send(Err(err)).await;
}
}
}
});
Ok(Response::new(Box::pin(ReceiverStream::new(rx))))
}
}
pub async fn setup() -> Result<(), Box<dyn Error>> {
let addr = "[::1]:50051".parse()?;
let tlm_service = CoreTelemetryService::default();
Server::builder()
.add_service(TelemetryServiceServer::new(tlm_service))
.serve(addr)
.await?;
Ok(())
}

30
server/src/main.rs Normal file
View File

@@ -0,0 +1,30 @@
use std::env;
use std::str::FromStr;
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let log_file = env::var("LOG_FILE");
let log_level = match env::var("LOG_LEVEL") {
Ok(log_level) => log::LevelFilter::from_str(&log_level)
.unwrap_or(log::LevelFilter::Info),
Err(_) => log::LevelFilter::Info
};
let mut log_config = fern::Dispatch::new()
.format(|out, message, record| {
out.finish(format_args!(
"[{}][{}][{}] {}",
chrono::Utc::now().format("%Y-%m-%d %H:%M:%S"),
record.target(),
record.level(),
message
))
})
.level(log_level)
.chain(std::io::stdout());
if let Ok(log_file) = log_file {
log_config = log_config.chain(fern::log_file(log_file)?)
}
log_config.apply()?;
server::setup().await
}

12
server/src/uuid.rs Normal file
View File

@@ -0,0 +1,12 @@
use rand::RngCore;
use crate::core::{Uuid};
impl Uuid {
pub fn random() -> Self {
let mut uuid = vec![0u8; 16];
rand::thread_rng().fill_bytes(&mut uuid);
Self {
value: uuid,
}
}
}