From 4a82d065a2f5ea77bd160588ddf6b520af140cc3 Mon Sep 17 00:00:00 2001
From: Termina1 <terminal2010@gmail.com>
Date: Thu, 9 Jan 2025 13:22:52 +0200
Subject: [PATCH] add drain time statistics

---
 chotki.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/chotki.go b/chotki.go
index dd79ba1..4691048 100644
--- a/chotki.go
+++ b/chotki.go
@@ -65,6 +65,12 @@ var EventsBatchSize = prometheus.NewHistogram(prometheus.HistogramOpts{
 	Buckets:   []float64{0, 1, 10, 50, 100, 500, 1000, 10000, 100000, 1000000},
 })
 
+var DrainTime = prometheus.NewHistogramVec(prometheus.HistogramOpts{
+	Namespace: "chotki",
+	Name:      "drain_time",
+	Buckets:   prometheus.DefBuckets,
+}, []string{"type"})
+
 type Options struct {
 	pebble.Options
 
@@ -501,6 +507,10 @@ func (cho *Chotki) Broadcast(ctx context.Context, records protocol.Records, exce
 
 // Here new packets are timestamped and queued for save
 func (cho *Chotki) CommitPacket(ctx context.Context, lit byte, ref rdx.ID, body protocol.Records) (id rdx.ID, err error) {
+	now := time.Now()
+	defer func() {
+		DrainTime.WithLabelValues("commit+broadcast").Observe(float64(time.Since(now)) / float64(time.Millisecond))
+	}()
 	cho.commitMutex.Lock()
 	defer cho.commitMutex.Unlock()
 
@@ -513,6 +523,7 @@ func (cho *Chotki) CommitPacket(ctx context.Context, lit byte, ref rdx.ID, body
 	packet := protocol.Record(lit, i, r, protocol.Join(body...))
 	recs := protocol.Records{packet}
 	err = cho.drain(ctx, recs)
+	DrainTime.WithLabelValues("commit").Observe(float64(time.Since(now)) / float64(time.Millisecond))
 	cho.Broadcast(ctx, recs, "")
 	return
 }
@@ -557,6 +568,7 @@ func (cho *Chotki) Metrics() []prometheus.Collector {
 		OpenedIterators,
 		OpenedSnapshots,
 		SessionsStates,
+		DrainTime,
 	}
 }
 
@@ -662,6 +674,10 @@ func (cho *Chotki) drain(ctx context.Context, recs protocol.Records) (err error)
 }
 
 func (cho *Chotki) Drain(ctx context.Context, recs protocol.Records) (err error) {
+	now := time.Now()
+	defer func() {
+		DrainTime.WithLabelValues("drain").Observe(float64(time.Since(now)) / float64(time.Millisecond))
+	}()
 	cho.lock.RLock()
 	defer cho.lock.RUnlock()
 	if cho.db == nil {