From d1b5606b0004b4b7342bcd26b948701b52c51da0 Mon Sep 17 00:00:00 2001
From: Rafael Sampaio <5679073+r4f4ss@users.noreply.github.com>
Date: Thu, 17 Oct 2024 18:41:12 -0300
Subject: [PATCH] metric storage_capacity

---
 cmd/shisui/main.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/cmd/shisui/main.go b/cmd/shisui/main.go
index 6260029552..6977180a4d 100644
--- a/cmd/shisui/main.go
+++ b/cmd/shisui/main.go
@@ -42,6 +42,10 @@ import (
 	"github.com/urfave/cli/v2"
 )
 
+var (
+	storageCapacity metrics.Gauge
+)
+
 const (
 	privateKeyFileName = "clientKey"
 )
@@ -124,6 +128,11 @@ func shisui(ctx *cli.Context) error {
 	// Start system runtime metrics collection
 	go metrics.CollectProcessMetrics(3 * time.Second)
 
+	if metrics.Enabled {
+		storageCapacity = metrics.NewRegisteredGauge("portal/storage_capacity", nil)
+		storageCapacity.Update(ctx.Int64(utils.PortalDataCapacityFlag.Name))
+	}
+
 	config, err := getPortalConfig(ctx)
 	if err != nil {
 		return nil