add node ID getting and filtering

Author: br0kenpixel

queries/get_node_ids.sql

@@ -0,0 +1,4 @@
+SELECT
+    id
+FROM
+    devices;

src/generator.rs

@@ -1,8 +1,33 @@
-use sqlx::PgConnection;
+use log::{debug, info};
+use sqlx::{query_file, PgConnection};
 use std::io::Write;
 
-pub async fn run(db: PgConnection, out: impl Write) -> anyhow::Result<()> {
-    // ...
+pub async fn run(
+    stats: bool,
+    verify: bool,
+    skip: &[u16],
+    mut db: PgConnection,
+    out: impl Write,
+) -> anyhow::Result<()> {
+    let mut nodes = get_node_ids(&mut db).await?;
+    debug!("Got node IDs: {nodes:?}");
+
+    debug!("Filtering nodes: {skip:?}");
+    nodes.retain(|id| skip.contains(id));
+    info!("Working with nodes: {nodes:?}");
 
     Ok(())
 }
+
+async fn get_node_ids(db: &mut PgConnection) -> anyhow::Result<Vec<u16>> {
+    let results = query_file!("queries/get_node_ids.sql")
+        .fetch_all(db)
+        .await?;
+
+    let ids: Vec<u16> = results
+        .iter()
+        .flat_map(|record| record.id.try_into())
+        .collect();
+
+    Ok(ids)
+}

src/main.rs

@@ -29,13 +29,18 @@ async fn main() -> Result<(), Box<dyn Error>> {
     debug!("Connected to database");
 
     match args.command {
-        cli::Command::Generate { out } => {
+        cli::Command::Generate {
+            stats,
+            verify,
+            skip,
+            out,
+        } => {
             let out: &mut dyn Write = match out {
                 Some(file) => &mut OpenOptions::new().write(true).open(file)?,
                 None => &mut stdout(),
             };
 
-            generator::run(db, out).await?
+            generator::run(stats, verify, &skip, db, out).await?
         }
     }