Merge pull request #25 from status-im/feat/peer-count

feat: handle peer count metric

telemetry/peer_count.go

@@ -0,0 +1,35 @@
+package telemetry
+
+import (
+	"database/sql"
+	"time"
+)
+
+type PeerCount struct {
+	ID            int    `json:"id"`
+	CreatedAt     int64  `json:"createdAt"`
+	Timestamp     int64  `json:"timestamp"`
+	NodeName      string `json:"nodeName"`
+	NodeKeyUid    string `json:"nodeKeyUid"`
+	PeerCount     int    `json:"peerCount"`
+	StatusVersion string `json:"statusVersion"`
+}
+
+func (r *PeerCount) put(db *sql.DB) error {
+	stmt, err := db.Prepare("INSERT INTO peerCount (timestamp, nodeName, nodeKeyUid, peerCount, statusVersion, createdAt) VALUES ($1, $2, $3, $4, $5, $6) RETURNING id;")
+	if err != nil {
+		return err
+	}
+
+	defer stmt.Close()
+
+	r.CreatedAt = time.Now().Unix()
+	lastInsertId := 0
+	err = stmt.QueryRow(r.Timestamp, r.NodeName, r.NodeKeyUid, r.PeerCount, r.StatusVersion, r.CreatedAt).Scan(&lastInsertId)
+	if err != nil {
+		return err
+	}
+	r.ID = lastInsertId
+
+	return nil
+}

telemetry/server.go

@@ -66,6 +66,7 @@ const (
 	UpdateEnvelopeMetric       TelemetryType = "UpdateEnvelope"
 	ReceivedMessagesMetric     TelemetryType = "ReceivedMessages"
 	ErrorSendingEnvelopeMetric TelemetryType = "ErrorSendingEnvelope"
+	PeerCountMetric            TelemetryType = "PeerCount"
 )
 
 type TelemetryRequest struct {
@@ -128,6 +129,16 @@ func (s *Server) createTelemetryData(w http.ResponseWriter, r *http.Request) {
 				errorDetails = append(errorDetails, map[string]interface{}{"Id": data.Id, "Error": fmt.Sprintf("Error saving error sending envelope: %v", err)})
 				continue
 			}
+		case PeerCountMetric:
+			var peerCount PeerCount
+			if err := json.Unmarshal(*data.TelemetryData, &peerCount); err != nil {
+				errorDetails = append(errorDetails, map[string]interface{}{"Id": data.Id, "Error": fmt.Sprintf("Error decoding peer count: %v", err)})
+				continue
+			}
+			if err := peerCount.put(s.DB); err != nil {
+				errorDetails = append(errorDetails, map[string]interface{}{"Id": data.Id, "Error": fmt.Sprintf("Error saving peer count: %v", err)})
+				continue
+			}
 		default:
 			errorDetails = append(errorDetails, map[string]interface{}{"Id": data.Id, "Error": fmt.Sprintf("Unknown telemetry type: %s", data.TelemetryType)})
 		}

telemetry/sql/000009_peer_count.up.sql

@@ -0,0 +1,10 @@
+CREATE TABLE IF NOT EXISTS peerCount (
+    id SERIAL PRIMARY KEY,
+    createdAt INTEGER NOT NULL,
+    peerCount INTEGER NOT NULL,
+    nodeName VARCHAR(255) NOT NULL,
+    nodeKeyUid VARCHAR(255) NOT NULL,
+    timestamp INTEGER NOT NULL,
+    statusVersion VARCHAR(31),
+    CONSTRAINT peerCount_unique unique(timestamp, nodeName, nodeKeyUid, statusVersion)
+);