server: Write CSV to log file

[tattlekey] / server / src / main.rs

diff --git a/server/src/main.rs b/server/src/main.rs
index 37de6bd44fae83acde13602b49a1a90876938441..a186abd58ec1b0e8045b0e29881fbcd7ecd76229 100644 (file)
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -14,13 +14,14 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+use serde::Serialize;
 use std::collections::HashMap;
 use std::net::UdpSocket;
 use std::time::{Duration, SystemTime};
 
 const MESSAGE_SIZE: usize = 12;
 
-#[derive(Eq, Debug, Hash, PartialEq)]
+#[derive(Eq, Debug, Hash, PartialEq, Serialize)]
 struct MessageKey {
     epoch: u32,
     device: u16,
@@ -79,6 +80,12 @@ impl Range {
 }
 
 fn main() {
+    let logfile = std::fs::OpenOptions::new()
+        .create(true)
+        .append(true)
+        .open("log.csv")
+        .expect("Coudln't open log file");
+    let mut log = csv::Writer::from_writer(logfile);
     let mut presses = HashMap::<MessageKey, Range>::new();
     let socket = UdpSocket::bind("0.0.0.0:29803").expect("couldn't bind to address");
     loop {
@@ -92,13 +99,14 @@ fn main() {
                     continue;
                 }
                 let message = Message::try_from(filled_buf).expect("I can't count");
+                log.serialize((&message.key, message.t))
+                    .expect("Couldn't write log");
+                log.flush().expect("Couldn't flush log");
                 if let Some(r) = presses.get_mut(&message.key) {
                     if !r.contains(&message.t) {
                         r.extend(&message.t);
-                        println!("Updated press: {:?}: {r:?}", message.key);
                     }
                 } else {
-                    println!("Got new press: {:?}: {:?}", message.key, message.t);
                     presses.insert(message.key, Range::new(&message.t));
                 }
             }