segment_service

server/src/analytic_unit.rs

@@ -1,22 +1,18 @@
-use hastic::config::Config;
-use hastic::services::data_service::Segment;
-use hastic::services::metric_service::MetricService;
 use anyhow;
+use hastic::services::metric_service::MetricService;
+use hastic::services::segments_service::Segment;
+use hastic::{config::Config, services::segments_service::SegmentType};
 
 use subbeat::metric::{Metric, MetricResult};
 
-
 struct AnalyticUnit {
-    metric_service: MetricService
+    metric_service: MetricService,
 }
 
 impl AnalyticUnit {
     fn new(config: &Config) -> AnalyticUnit {
-        AnalyticUnit{
-            metric_service: MetricService::new(
-                &config.prom_url,
-                &config.query
-            )
+        AnalyticUnit {
+            metric_service: MetricService::new(&config.prom_url, &config.query),
         }
     }
 
@@ -38,7 +34,12 @@ impl AnalyticUnit {
                 }
             } else {
                 if from.is_some() {
-                    result.push(Segment{ id: None, from: from.unwrap(), to: *t });
+                    result.push(Segment {
+                        id: None,
+                        from: from.unwrap(),
+                        to: *t,
+                        segment_type: SegmentType::Detection,
+                    });
                     from = None;
                 }
             }
@@ -46,4 +47,4 @@ impl AnalyticUnit {
 
         Ok(result)
     }
-}
+}

server/src/api.rs

@@ -1,5 +1,5 @@
 use hastic::config::Config;
-use hastic::services::{data_service, metric_service, user_service};
+use hastic::services::{metric_service, segments_service, user_service};
 use warp::http::HeaderValue;
 use warp::hyper::{Body, StatusCode};
 use warp::reject::Reject;
@@ -29,7 +29,7 @@ pub struct API<'a> {
     config: &'a Config,
     user_service: Arc<RwLock<user_service::UserService>>,
     metric_service: Arc<RwLock<metric_service::MetricService>>,
-    data_service: Arc<RwLock<data_service::DataService>>,
+    data_service: Arc<RwLock<segments_service::SegmentsService>>,
 }
 
 impl API<'_> {
@@ -41,7 +41,7 @@ impl API<'_> {
                 &config.prom_url,
                 &config.query,
             ))),
-            data_service: Arc::new(RwLock::new(data_service::DataService::new()?)),
+            data_service: Arc::new(RwLock::new(segments_service::SegmentsService::new()?)),
         })
     }
 

server/src/api/segments.rs

@@ -7,9 +7,10 @@ pub mod filters {
     pub fn filters(
         db: Db,
     ) -> impl Filter<Extract = impl warp::Reply, Error = warp::Rejection> + Clone {
-        list(db.clone()).or(create(db.clone()))
-        // .or(update(db.clone()))
-        .or(delete(db.clone()))
+        list(db.clone())
+            .or(create(db.clone()))
+            // .or(update(db.clone()))
+            .or(delete(db.clone()))
     }
 
     /// GET /segments?from=3&to=5
@@ -51,8 +52,8 @@ pub mod filters {
 }
 
 mod handlers {
-    use hastic::services::data_service;
-    use hastic::services::data_service::Segment;
+    use hastic::services::segments_service;
+    use hastic::services::segments_service::Segment;
 
     use super::models::{CreateResponse, Db, ListOptions};
     use crate::api;
@@ -67,7 +68,7 @@ mod handlers {
     }
 
     pub async fn create(
-        segment: data_service::Segment,
+        segment: segments_service::Segment,
         db: Db,
     ) -> Result<impl warp::Reply, warp::Rejection> {
         match db.write().insert_segment(&segment) {
@@ -81,20 +82,21 @@ mod handlers {
 
     pub async fn delete(opts: ListOptions, db: Db) -> Result<impl warp::Reply, warp::Rejection> {
         match db.read().delete_segments_in_range(opts.from, opts.to) {
-            Ok(count) => Ok(API::json(&api::Message{ message: count.to_string() })),
+            Ok(count) => Ok(API::json(&api::Message {
+                message: count.to_string(),
+            })),
             Err(e) => Err(warp::reject::custom(BadQuery)),
         }
     }
-
 }
 
 mod models {
-    use hastic::services::data_service::{self, SegmentId};
+    use hastic::services::segments_service::{self, SegmentId};
     use parking_lot::RwLock;
     use serde::{Deserialize, Serialize};
     use std::sync::Arc;
 
-    pub type Db = Arc<RwLock<data_service::DataService>>;
+    pub type Db = Arc<RwLock<segments_service::SegmentsService>>;
 
     // The query parameters for list_todos.
     #[derive(Debug, Deserialize)]

server/src/main.rs

@@ -1,7 +1,7 @@
 use anyhow;
 
-mod api;
 mod analytic_unit;
+mod api;
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {

server/src/services.rs

@@ -1,3 +1,3 @@
-pub mod data_service;
 pub mod metric_service;
+pub mod segments_service;
 pub mod user_service;

server/src/services/segments_service.rs

@@ -1,4 +1,4 @@
-use rusqlite::{params, Connection, ToSql};
+use rusqlite::{params, Connection, Row, ToSql};
 
 use serde::{Deserialize, Serialize};
 
@@ -9,33 +9,62 @@ use std::iter::repeat_with;
 const ID_LENGTH: usize = 20;
 pub type SegmentId = String;
 
+#[derive(Debug, Serialize, Deserialize, Clone, Copy)]
+pub enum SegmentType {
+    Label = 1,
+    Detection = 2,
+}
+
+impl SegmentType {
+    fn from(u: u64) -> SegmentType {
+        if u == 1 {
+            SegmentType::Label
+        } else {
+            SegmentType::Detection
+        }
+    }
+}
+
 #[derive(Debug, Serialize, Deserialize)]
 pub struct Segment {
     pub id: Option<SegmentId>,
     pub from: u64,
     pub to: u64,
+    pub segment_type: SegmentType,
+}
+
+impl Segment {
+    fn from(row: &Row) -> anyhow::Result<Segment, rusqlite::Error> {
+        Ok(Segment {
+            id: row.get(0)?,
+            from: row.get(1)?,
+            to: row.get(2)?,
+            segment_type: SegmentType::from(row.get(3)?),
+        })
+    }
 }
 
 // TODO: find a way to remove this unsafe
-unsafe impl Sync for DataService {}
+unsafe impl Sync for SegmentsService {}
 
-pub struct DataService {
+pub struct SegmentsService {
     connection: Arc<Mutex<Connection>>,
 }
 
-impl DataService {
-    pub fn new() -> anyhow::Result<DataService> {
-        let conn = Connection::open("./data.db")?;
+impl SegmentsService {
+    pub fn new() -> anyhow::Result<SegmentsService> {
+        let conn = Connection::open("./segments.db")?;
         conn.execute(
             "CREATE TABLE IF NOT EXISTS segment (
                       id        TEXT PRIMARY KEY,
                       start     INTEGER NOT NULL,
-                      end       INTEGER NOT NULL
+                      end       INTEGER NOT NULL,
+                      segment_type INTEGER NOT NULL
                  )",
             [],
         )?;
 
-        Ok(DataService {
+        Ok(SegmentsService {
             connection: Arc::new(Mutex::new(conn)),
         })
     }
@@ -61,29 +90,25 @@ impl DataService {
         self.delete_segments(&ids_to_delete)?;
 
         self.connection.lock().unwrap().execute(
-            "INSERT INTO segment (id, start, end) VALUES (?1, ?2, ?3)",
+            "INSERT INTO segment (id, start, end, segment_type) VALUES (?1, ?2, ?3, ?4)",
             params![id, min, max],
         )?;
         Ok(Segment {
             id: Some(id),
             from: min,
             to: max,
+            segment_type: segment.segment_type,
         })
     }
 
     pub fn get_segments_inside(&self, from: u64, to: u64) -> anyhow::Result<Vec<Segment>> {
         let conn = self.connection.lock().unwrap();
-        let mut stmt =
-            conn.prepare("SELECT id, start, end FROM segment WHERE ?1 < start AND end < ?2")?;
+        let mut stmt = conn.prepare(
+            "SELECT id, start, end, segment_type FROM segment WHERE ?1 < start AND end < ?2",
+        )?;
 
         let res = stmt
-            .query_map(params![from, to], |row| {
-                Ok(Segment {
-                    id: row.get(0)?,
-                    from: row.get(1)?,
-                    to: row.get(2)?,
-                })
-            })?
+            .query_map(params![from, to], Segment::from)?
             .map(|e| e.unwrap())
             .collect();
         Ok(res)
@@ -92,21 +117,16 @@ impl DataService {
     pub fn get_segments_intersected(&self, from: u64, to: u64) -> anyhow::Result<Vec<Segment>> {
         let conn = self.connection.lock().unwrap();
         let mut stmt = conn.prepare(
-            "SELECT id, start, end FROM segment 
-                                                  WHERE (start <= ?1 and ?1 <= end) OR 
-                                                        (start <= ?2 AND ?2 <= end) OR
-                                                        (?1 <= start AND start <= ?2) OR 
-                                                        (?1 <= end AND end <= ?2) ",
+            "SELECT id, start, end, segment_type
+                    FROM segment
+                    WHERE (start <= ?1 and ?1 <= end) OR 
+                          (start <= ?2 AND ?2 <= end) OR
+                          (?1 <= start AND start <= ?2) OR 
+                          (?1 <= end AND end <= ?2) ",
         )?;
 
         let res = stmt
-            .query_map(params![from, to], |row| {
-                Ok(Segment {
-                    id: row.get(0)?,
-                    from: row.get(1)?,
-                    to: row.get(2)?,
-                })
-            })?
+            .query_map(params![from, to], Segment::from)?
             .map(|e| e.unwrap())
             .collect();
         Ok(res)