show total visitors

d-led · Jun 16, 2024 · 2f546be · 2f546be
1 parent 4a7db27
commit 2f546be
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/counter_listener.go b/counter_listener.go
@@ -18,5 +18,5 @@ func NewCounterListener(events *pubsub.PubSub[string, Event]) *CounterListener {
 
 func (n *CounterListener) OnNewCount(count int64) {
 	log.Println("New visitor count:", count)
-	n.events.Pub(NewEventWithParam("TotalVisitors", count))
+	n.events.Pub(NewEventWithParam("TotalVisitors", count), Topic)
 }
diff --git a/fly_peers.go b/fly_peers.go
@@ -81,6 +81,7 @@ func (ps *PeerSource) listenToInternalEventsForever() {
 	for event := range subscription {
 		if event.Name == "VisitorJoined" {
 			ps.counter.Increment()
+			ps.events.Pub(NewEventWithParam("TotalVisitors", ps.counter.Value()), Topic)
 		}
 	}
 }

diff --git a/ui-src/index.html b/ui-src/index.html
@@ -76,6 +76,10 @@ <h4>Server-side updates & info</h4>
               <td>Replicas</td>
               <td><span id="replicas"></span></td>
             </tr>
+            <tr class="monospaced">
+              <td>Total visitors (local for now)</td>
+              <td><span id="total-visitors"></span></td>
+            </tr>
           </tbody>
         </table>
       </div>

diff --git a/ui-src/index.ts b/ui-src/index.ts
@@ -112,6 +112,13 @@ function showReplicasActive(count: number) {
   replaceText("#replicas", `${count}`);
 }
 
+function showTotalVisitors(count: number) {
+  if (count == null) {
+    return;
+  }
+  replaceText("#total-visitors", `${count}`);
+}
+
 function showServerRevision(text: string) {
   replaceText("#server-revision", text);
 }
@@ -172,6 +179,10 @@ async function processEvent(event) {
       showReplicasActive(event?.properties?.param);
       // do not show this event in the log
       return;
+    case "TotalVisitors":
+      showTotalVisitors(event?.properties?.param);
+      // do not show this event in the log
+      return;
     case "Revision":
       showServerRevision(event?.properties?.param);
       return;