Skip to content

Commit

Permalink
Cleanup.
Browse files Browse the repository at this point in the history
Signed-off-by: Cody Littley <[email protected]>
  • Loading branch information
cody-littley committed Dec 6, 2024
1 parent d9d898c commit 8bd8ff1
Showing 1 changed file with 23 additions and 21 deletions.
44 changes: 23 additions & 21 deletions node/grpc/v2_metrics.go
Original file line number Diff line number Diff line change
Expand Up @@ -143,31 +143,33 @@ func (m *V2Metrics) Start() {
}
}()

if m.dbSizePollPeriod.Nanoseconds() > 0 {
go func() {
ticker := time.NewTicker(m.dbSizePollPeriod)

for m.isAlive.Load() {
var size int64
err := filepath.Walk(m.dbDir, func(_ string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if !info.IsDir() {
size += info.Size()
}
return err
})
if m.dbSizePollPeriod.Nanoseconds() == 0 {
return
}
go func() {
ticker := time.NewTicker(m.dbSizePollPeriod)

for m.isAlive.Load() {
var size int64
err := filepath.Walk(m.dbDir, func(_ string, info os.FileInfo, err error) error {
if err != nil {
m.logger.Errorf("failed to get database size: %v", err)
} else {
m.dbSize.WithLabelValues().Set(float64(size))
return err
}
if !info.IsDir() {
size += info.Size()
}
<-ticker.C
return err
})

if err != nil {
m.logger.Errorf("failed to get database size: %v", err)
} else {
m.dbSize.WithLabelValues().Set(float64(size))
}
}()
}
<-ticker.C
}
}()

}

// Stop stops the metrics server.
Expand Down

0 comments on commit 8bd8ff1

Please sign in to comment.