return a number for current_visitors

vinceanalytics · Feb 17, 2024 · 75afca7 · 75afca7
1 parent 848095f
commit 75afca7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/docs/3-stats.md b/docs/3-stats.md
@@ -99,9 +99,7 @@ in the last 5 minutes.
 
 ```bash
 + curl -X GET 'http://localhost:8080/api/v1/stats/realtime/visitors?site_id=vinceanalytics.com'
-{
-  "visitors": "5"
-}
+6
 ```
 
 

diff --git a/request/request.go b/request/request.go
@@ -97,8 +97,14 @@ func Validate(ctx context.Context, w http.ResponseWriter, o proto.Message) bool
 	return true
 }
 
-func Write(ctx context.Context, w http.ResponseWriter, o proto.Message) {
-	data, err := m.Marshal(o)
+func Write(ctx context.Context, w http.ResponseWriter, o any) {
+	var data []byte
+	var err error
+	if a, ok := o.(proto.Message); ok {
+		data, err = m.Marshal(a)
+	} else {
+		data, err = json.Marshal(o)
+	}
 	if err != nil {
 		logger.Fail("Can't marshall proto messages", "err", err)
 	}

diff --git a/stats/current_visitors.go b/stats/current_visitors.go
@@ -46,5 +46,5 @@ func Realtime(w http.ResponseWriter, r *http.Request) {
 		request.Internal(ctx, w)
 		return
 	}
-	request.Write(ctx, w, &v1.Realtime_Response{Visitors: uint64(visitors)})
+	request.Write(ctx, w, uint64(visitors))
 }