Newer
Older
Angelo Capossele
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
package prometheus
import (
"os"
"path/filepath"
"github.com/iotaledger/goshimmer/plugins/config"
"github.com/iotaledger/goshimmer/plugins/database"
"github.com/prometheus/client_golang/prometheus"
)
var (
dbSize prometheus.Gauge
)
func registerDBMetrics() {
dbSize = prometheus.NewGauge(
prometheus.GaugeOpts{
Name: "db_size_bytes",
Help: "DB size in bytes.",
},
)
registry.MustRegister(dbSize)
addCollect(collectDBSize)
}
func collectDBSize() {
size, err := directorySize(config.Node.GetString(database.CfgDatabaseDir))
if err == nil {
dbSize.Set(float64(size))
}
}
func directorySize(path string) (int64, error) {
var size int64
err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if !info.IsDir() {
size += info.Size()
}
return err
})
return size, err
}