Add: pool module

X-FRI · Sep 21, 2023 · 30bd742 · 30bd742
1 parent 33ed5fd
commit 30bd742
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/kernel/lib/pool/connection_pool.ml b/kernel/lib/pool/connection_pool.ml
@@ -0,0 +1,10 @@
+module Connection_pool = Hashtbl.Make (String)
+
+type t = Io.Client.t Connection_pool.t
+
+let add pool (connection_id, connection_client) =
+    match Connection_pool.find_opt pool connection_id with
+    | Some _ -> Io.Client.close connection_client
+    | None -> Connection_pool.add pool connection_id connection_client
+
+let get pool connection_id = Connection_pool.find connection_id pool
diff --git a/kernel/lib/pool/dune b/kernel/lib/pool/dune
@@ -1,3 +1,3 @@
 (library
  (name pool)
- (libraries core dream log))
+ (libraries core dream io log))