1 files changed, 7 insertions, 5 deletions
diff --git a/src/modules/router/module.rs b/src/modules/router/module.rs
index dc9dae8..34d20db 100644
--- a/src/modules/router/module.rs
+++ b/src/modules/router/module.rs
@@ -16,8 +16,10 @@
 // Copyright (C) 2022-2022 Fuwn <[email protected]>
 // SPDX-License-Identifier: GPL-3.0-only
 
+use windmark::context::HookContext;
+
 pub fn module(router: &mut windmark::Router) {
-  router.set_pre_route_callback(Box::new(|context| {
+  router.set_pre_route_callback(move |context: HookContext<'_>| {
     info!(
       "accepted connection from {} to {}",
       context.tcp.peer_addr().unwrap().ip(),
@@ -25,9 +27,9 @@ pub fn module(router: &mut windmark::Router) {
     );
 
     super::ticker::module(&context);
-  }));
-  router.set_post_route_callback(Box::new({
-    |context, response| {
+  });
+  router.set_post_route_callback({
+    |context: HookContext<'_>, response: &mut windmark::Response| {
       info!(
         "closed connection from {} to {}",
         context.tcp.peer_addr().unwrap().ip(),
@@ -50,5 +52,5 @@ pub fn module(router: &mut windmark::Router) {
         }
       }
     }
-  }));
+  });
 }