lelgenio/made-you-look
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix ip matching

Browse source
This commit is contained in:
lelgenio 2024-10-05 14:06:56 -03:00
parent 55e9cf1f9c
commit acd8778864
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/lib.rs
View file

@ -1,4 +1,10 @@
use std::{collections::HashSet, future::Future, net::SocketAddr, pin::Pin, sync::Arc}; use std::{
collections::HashSet,
future::Future,
net::{IpAddr, SocketAddr},
pin::Pin,
sync::Arc,
};
use anyhow::{Context, Result}; use anyhow::{Context, Result};
use askama::Template; use askama::Template;
@ -18,7 +24,7 @@ async fn routes() -> Router {
#[derive(Clone)] #[derive(Clone)]
pub struct AppState { pub struct AppState {
view_count: Arc<RwLock<u32>>, view_count: Arc<RwLock<u32>>,
seen_ips: Arc<RwLock<HashSet<SocketAddr>>>, seen_ips: Arc<RwLock<HashSet<IpAddr>>>,
} }
impl AppState { impl AppState {
@ -40,7 +46,7 @@ impl AppState {
} }
} }
async fn increment(&self, ip: SocketAddr) -> u32 { async fn increment(&self, ip: IpAddr) -> u32 {
if self.seen_ips.read().await.contains(&ip) { if self.seen_ips.read().await.contains(&ip) {
let view_count = self.view_count.read().await; let view_count = self.view_count.read().await;
return *view_count; return *view_count;
@ -99,8 +105,9 @@ async fn write_count(new_count: u32) -> anyhow::Result<()> {
#[axum::debug_handler] #[axum::debug_handler]
pub async fn home( pub async fn home(
State(state): State<AppState>, State(state): State<AppState>,
ConnectInfo(ip): ConnectInfo<SocketAddr>, ConnectInfo(addr): ConnectInfo<SocketAddr>,
) -> HomeTemplate { ) -> HomeTemplate {
let ip = addr.ip();
tracing::info!(ip = ip.to_string()); tracing::info!(ip = ip.to_string());
let view_count = state.increment(ip).await; let view_count = state.increment(ip).await;