Antennas basic endpoints (#4)

Co-authored-by: Felipe Diniello <felipediniello@pm.me>
Reviewed-on: #4

kairo-common/Cargo.toml

@@ -14,6 +14,8 @@ dotenv = { workspace = true }
 chrono = { workspace = true }
 serde = { workspace = true }
 serde_json = { workspace = true }
+diesel = { workspace = true }
+
 
 influxdb2 = "0.4.2"
 influxdb2-structmap = "0.2"

kairo-common/diesel.toml

@@ -0,0 +1,9 @@
+# For documentation on how to configure this file,
+# see https://diesel.rs/guides/configuring-diesel-cli
+
+[print_schema]
+file = "src/schema.rs"
+custom_type_derives = ["diesel::query_builder::QueryId"]
+
+[migrations_directory]
+dir = "migrations"

kairo-common/migrations/00000000000000_diesel_initial_setup/down.sql

@@ -0,0 +1,6 @@
+-- This file was automatically created by Diesel to setup helper functions
+-- and other internal bookkeeping. This file is safe to edit, any future
+-- changes will be added to existing projects as new migrations.
+
+DROP FUNCTION IF EXISTS diesel_manage_updated_at(_tbl regclass);
+DROP FUNCTION IF EXISTS diesel_set_updated_at();

kairo-common/migrations/00000000000000_diesel_initial_setup/up.sql

@@ -0,0 +1,36 @@
+-- This file was automatically created by Diesel to setup helper functions
+-- and other internal bookkeeping. This file is safe to edit, any future
+-- changes will be added to existing projects as new migrations.
+
+
+
+
+-- Sets up a trigger for the given table to automatically set a column called
+-- `updated_at` whenever the row is modified (unless `updated_at` was included
+-- in the modified columns)
+--
+-- # Example
+--
+-- ```sql
+-- CREATE TABLE users (id SERIAL PRIMARY KEY, updated_at TIMESTAMP NOT NULL DEFAULT NOW());
+--
+-- SELECT diesel_manage_updated_at('users');
+-- ```
+CREATE OR REPLACE FUNCTION diesel_manage_updated_at(_tbl regclass) RETURNS VOID AS $$
+BEGIN
+    EXECUTE format('CREATE TRIGGER set_updated_at BEFORE UPDATE ON %s
+                    FOR EACH ROW EXECUTE PROCEDURE diesel_set_updated_at()', _tbl);
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE OR REPLACE FUNCTION diesel_set_updated_at() RETURNS trigger AS $$
+BEGIN
+    IF (
+        NEW IS DISTINCT FROM OLD AND
+        NEW.updated_at IS NOT DISTINCT FROM OLD.updated_at
+    ) THEN
+        NEW.updated_at := current_timestamp;
+    END IF;
+    RETURN NEW;
+END;
+$$ LANGUAGE plpgsql;

kairo-common/migrations/2023-06-18-164844_create_antennas/down.sql

@@ -0,0 +1,3 @@
+-- This file should undo anything in `up.sql`
+
+DROP TABLE antennas;

kairo-common/migrations/2023-06-18-164844_create_antennas/up.sql

@@ -0,0 +1,10 @@
+-- Your SQL goes here
+
+CREATE TABLE antennas (
+    id VARCHAR(17) PRIMARY KEY,
+    tssi DOUBLE PRECISION NOT NULL,
+    pos_x DOUBLE PRECISION NOT NULL,
+    pos_y DOUBLE PRECISION NOT NULL,
+    pos_z DOUBLE PRECISION NOT NULL,
+    comment TEXT
+);

kairo-common/src/lib.rs

@@ -4,12 +4,12 @@
 #![allow(confusable_idents)]
 
 pub mod influx;
+pub mod postgres;
 
-// random functions for mqtt 
+// random functions for mqtt
 pub mod mqtt;
 pub mod unit_conversion;
 
-
 // Commonly used types across the services
 mod types {
     pub mod mac; // deprecated for the time being.
@@ -18,7 +18,8 @@ mod types {
 pub type Point = types::point::Point;
 pub type MAC = types::mac::MAC;
 
-
+// DB models: for SQL with Diesel and InfluxDB and influxdb-derive
+pub mod schema;
 mod models {
     pub mod antenna;
     pub mod beacon_measure;
@@ -30,8 +31,10 @@ mod models {
         pub data: Vec<crate::models::beacon_measure::BeaconMeasure>,
     }
 }
-pub type Antenna = models::antenna::Antenna;
 pub type DeviceReport = models::DeviceReport;
+
+pub type Antenna = models::antenna::Antenna;
+
 pub type KnownPosition = models::known_position::KnownPosition;
 pub type DynamicDeviceStatus = models::dynamic_device_status::DynamicDeviceStatus;
 pub type BeaconMeasure = models::beacon_measure::BeaconMeasure;

kairo-common/src/models/antenna.rs

@@ -1,15 +1,27 @@
+use diesel::prelude::*;
 use std::f64::consts::PI;
 
 use crate::{unit_conversion::UnitsConversion, Point};
 
-#[derive(Debug, Clone, Default)]
+#[derive(
+    Debug,
+    Clone,
+    Queryable,
+    Selectable,
+    Insertable,
+    AsChangeset,
+    serde::Serialize,
+    serde::Deserialize,
+)]
+#[diesel(check_for_backend(diesel::pg::Pg))]
+#[diesel(table_name = crate::schema::antennas)]
 pub struct Antenna {
     pub id: String,
     pub tssi: f64,
+    pub pos_x: f64,
+    pub pos_y: f64,
+    pub pos_z: f64,
     pub comment: Option<String>,
-    pos_x: f64,
-    pos_y: f64,
-    pos_z: f64,
 }
 
 impl Antenna {

kairo-common/src/postgres.rs

@@ -0,0 +1,25 @@
+use diesel::prelude::*;
+use diesel::r2d2::{ConnectionManager, Pool, PooledConnection};
+
+pub type DbConn = diesel::pg::PgConnection;
+pub type DbPool = Pool<ConnectionManager<PgConnection>>;
+pub struct DbPooledConn(pub PooledConnection<ConnectionManager<PgConnection>>);
+
+pub fn establish_connection() -> DbConn {
+    let database_url = dotenv::var("DATABASE_URL").expect("DATABASE_URL must be set");
+    PgConnection::establish(&database_url)
+        .unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
+}
+
+pub fn init_pool() -> DbPool {
+    let database_url = dotenv::var("DATABASE_URL").expect("DATABASE_URL must be set");
+    let manager = ConnectionManager::<PgConnection>::new(database_url);
+    DbPool::new(manager).expect("Error connecting to DB")
+}
+
+impl std::ops::Deref for DbPooledConn {
+    type Target = PgConnection;
+    fn deref(&self) -> &Self::Target {
+        &self.0
+    }
+}

kairo-common/src/schema.rs

@@ -0,0 +1,13 @@
+// @generated automatically by Diesel CLI.
+
+diesel::table! {
+    antennas (id) {
+        #[max_length = 17]
+        id -> Varchar,
+        tssi -> Float8,
+        pos_x -> Float8,
+        pos_y -> Float8,
+        pos_z -> Float8,
+        comment -> Nullable<Text>,
+    }
+}