From 9d7cac5da110250539d1ae139cb159425b0e67b1 Mon Sep 17 00:00:00 2001
From: Lyn Breedlove <tenseiken595@gmail.com>
Date: Fri, 26 Oct 2018 16:17:23 -0600
Subject: More random stuff with sharding

---
 lib/client/sharder/shard_manager.ml | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 lib/client/sharder/shard_manager.ml

(limited to 'lib/client/sharder/shard_manager.ml')

diff --git a/lib/client/sharder/shard_manager.ml b/lib/client/sharder/shard_manager.ml
new file mode 100644
index 0000000..68b3d13
--- /dev/null
+++ b/lib/client/sharder/shard_manager.ml
@@ -0,0 +1,31 @@
+open Lwt.Infix
+open Websocket
+
+module ShardSet = Set.Make(Shard)
+
+type t = {
+    shards: Shard.t ShardSet.t;
+    gateway_url: Uri.t;
+    token: string;
+}
+
+let create_shard ?(options=[]) manager =
+    let id = (ShardSet.cardinal manager.shards) + 1 in
+    Shard.connect ~options ~uri:manager.gateway_url ~id ~total:(ShardSet.cardinal manager.shards) ~token:manager.token ()
+    >|= fun shard ->
+    ShardSet.add shard manager.shards
+
+let update_shard shard manager =
+    match ShardSet.mem shard manager.shards with
+    | true -> ShardSet.add shard manager.shards
+    | false -> manager.shards
+
+let heartbeat shard manager =
+    Shard.heartbeat shard
+
+let identify shard manager =
+    let total = ShardSet.cardinal manager.shards in
+    Shard.identify total manager.token shard
+
+let resume shard manager =
+    Shard.resume manager.token shard
\ No newline at end of file
-- 
cgit v1.2.3