fix download counts for ipv4 ips (#750)

src/lib.rs

@@ -4,8 +4,7 @@ use actix_web::web;
 use database::redis::RedisPool;
 use log::{info, warn};
 use queue::{
-    analytics::AnalyticsQueue, payouts::PayoutsQueue, session::AuthQueue,
+    analytics::AnalyticsQueue, payouts::PayoutsQueue, session::AuthQueue, socket::ActiveSockets,
-    socket::ActiveSockets,
 };
 use scheduler::Scheduler;
 use sqlx::Postgres;

src/queue/analytics.rs

@@ -1,10 +1,10 @@
 use crate::database::models::DatabaseError;
+use crate::database::redis::RedisPool;
 use crate::models::analytics::{Download, PageView, Playtime};
 use crate::routes::ApiError;
 use dashmap::{DashMap, DashSet};
 use redis::cmd;
 use sqlx::PgPool;
-use crate::database::redis::RedisPool;
 
 const DOWNLOADS_NAMESPACE: &str = "downloads";
 
@@ -35,10 +35,13 @@ impl AnalyticsQueue {
     }
 
     pub fn add_download(&self, download: Download) {
+        let ip_stripped = if let Some(ip) = download.ip.to_ipv4_mapped() {
+            let octets = ip.octets();
+            u64::from_be_bytes([0, 0, 0, 0, octets[0], octets[1], octets[2], octets[3]])
+        } else {
             let octets = download.ip.octets();
-        let ip_stripped = u64::from_be_bytes([
+            u64::from_be_bytes([0, 0, 0, 0, octets[0], octets[1], octets[2], octets[3]])
-            octets[0], octets[1], octets[2], octets[3], octets[4], octets[5], octets[6], octets[7],
+        };
-        ]);
         self.downloads_queue
             .insert(format!("{}-{}", ip_stripped, download.project_id), download);
     }