Skip to content

Commit 9d557be

Browse files
authored
Add observability report wrapper for scrapers and use it everywhere (#1276)
Signed-off-by: Bogdan Drutu <[email protected]>
1 parent 97d9d86 commit 9d557be

File tree

25 files changed

+357
-67
lines changed

25 files changed

+357
-67
lines changed

receiver/hostmetricsreceiver/internal/scraper/cpuscraper/cpu_scraper.go

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ import (
2020

2121
"github.com/shirou/gopsutil/cpu"
2222
"github.com/shirou/gopsutil/host"
23-
"go.opencensus.io/trace"
2423

2524
"go.opentelemetry.io/collector/consumer/pdata"
2625
)
@@ -53,10 +52,7 @@ func (s *scraper) Close(_ context.Context) error {
5352
}
5453

5554
// ScrapeMetrics
56-
func (s *scraper) ScrapeMetrics(ctx context.Context) (pdata.MetricSlice, error) {
57-
_, span := trace.StartSpan(ctx, "cpuscraper.ScrapeMetrics")
58-
defer span.End()
59-
55+
func (s *scraper) ScrapeMetrics(_ context.Context) (pdata.MetricSlice, error) {
6056
metrics := pdata.NewMetricSlice()
6157

6258
cpuTimes, err := cpu.Times(s.config.ReportPerCPU)

receiver/hostmetricsreceiver/internal/scraper/cpuscraper/factory.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"go.uber.org/zap"
2121

2222
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal"
23+
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal/scraper/obsreportscraper"
2324
)
2425

2526
// This file implements Factory for CPU scraper.
@@ -45,5 +46,5 @@ func (f *Factory) CreateMetricsScraper(
4546
config internal.Config,
4647
) (internal.Scraper, error) {
4748
cfg := config.(*Config)
48-
return newCPUScraper(ctx, cfg), nil
49+
return obsreportscraper.WrapScraper(newCPUScraper(ctx, cfg), TypeStr), nil
4950
}

receiver/hostmetricsreceiver/internal/scraper/diskscraper/disk_scraper.go

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ import (
2020

2121
"github.com/shirou/gopsutil/disk"
2222
"github.com/shirou/gopsutil/host"
23-
"go.opencensus.io/trace"
2423

2524
"go.opentelemetry.io/collector/consumer/pdata"
2625
)
@@ -53,10 +52,7 @@ func (s *scraper) Close(_ context.Context) error {
5352
}
5453

5554
// ScrapeMetrics
56-
func (s *scraper) ScrapeMetrics(ctx context.Context) (pdata.MetricSlice, error) {
57-
_, span := trace.StartSpan(ctx, "diskscraper.ScrapeMetrics")
58-
defer span.End()
59-
55+
func (s *scraper) ScrapeMetrics(_ context.Context) (pdata.MetricSlice, error) {
6056
metrics := pdata.NewMetricSlice()
6157

6258
ioCounters, err := disk.IOCounters()

receiver/hostmetricsreceiver/internal/scraper/diskscraper/factory.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"go.uber.org/zap"
2121

2222
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal"
23+
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal/scraper/obsreportscraper"
2324
)
2425

2526
// This file implements Factory for Disk scraper.
@@ -45,5 +46,5 @@ func (f *Factory) CreateMetricsScraper(
4546
config internal.Config,
4647
) (internal.Scraper, error) {
4748
cfg := config.(*Config)
48-
return newDiskScraper(ctx, cfg), nil
49+
return obsreportscraper.WrapScraper(newDiskScraper(ctx, cfg), TypeStr), nil
4950
}

receiver/hostmetricsreceiver/internal/scraper/filesystemscraper/factory.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"go.uber.org/zap"
2121

2222
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal"
23+
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal/scraper/obsreportscraper"
2324
)
2425

2526
// This file implements Factory for FileSystem scraper.
@@ -50,5 +51,5 @@ func (f *Factory) CreateMetricsScraper(
5051
config internal.Config,
5152
) (internal.Scraper, error) {
5253
cfg := config.(*Config)
53-
return newFileSystemScraper(ctx, cfg), nil
54+
return obsreportscraper.WrapScraper(newFileSystemScraper(ctx, cfg), TypeStr), nil
5455
}

receiver/hostmetricsreceiver/internal/scraper/filesystemscraper/filesystem_scraper.go

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@ import (
1919
"time"
2020

2121
"github.com/shirou/gopsutil/disk"
22-
"go.opencensus.io/trace"
2322

2423
"go.opentelemetry.io/collector/component/componenterror"
2524
"go.opentelemetry.io/collector/consumer/pdata"
@@ -51,16 +50,11 @@ func (s *scraper) Close(_ context.Context) error {
5150
}
5251

5352
// ScrapeMetrics
54-
func (s *scraper) ScrapeMetrics(ctx context.Context) (pdata.MetricSlice, error) {
55-
_, span := trace.StartSpan(ctx, "filesystemscraper.ScrapeMetrics")
56-
defer span.End()
57-
53+
func (s *scraper) ScrapeMetrics(_ context.Context) (pdata.MetricSlice, error) {
5854
metrics := pdata.NewMetricSlice()
5955

6056
// omit logical (virtual) filesystems (not relevant for windows)
61-
all := false
62-
63-
partitions, err := disk.Partitions(all)
57+
partitions, err := disk.Partitions( /*all=*/ false)
6458
if err != nil {
6559
return metrics, err
6660
}

receiver/hostmetricsreceiver/internal/scraper/loadscraper/factory.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"go.uber.org/zap"
2121

2222
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal"
23+
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal/scraper/obsreportscraper"
2324
)
2425

2526
// This file implements Factory for Load scraper.
@@ -45,5 +46,5 @@ func (f *Factory) CreateMetricsScraper(
4546
config internal.Config,
4647
) (internal.Scraper, error) {
4748
cfg := config.(*Config)
48-
return newLoadScraper(ctx, logger, cfg), nil
49+
return obsreportscraper.WrapScraper(newLoadScraper(ctx, logger, cfg), TypeStr), nil
4950
}

receiver/hostmetricsreceiver/internal/scraper/loadscraper/load_scraper.go

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ import (
1818
"context"
1919
"time"
2020

21-
"go.opencensus.io/trace"
2221
"go.uber.org/zap"
2322

2423
"go.opentelemetry.io/collector/consumer/pdata"
@@ -46,10 +45,7 @@ func (s *scraper) Close(ctx context.Context) error {
4645
}
4746

4847
// ScrapeMetrics
49-
func (s *scraper) ScrapeMetrics(ctx context.Context) (pdata.MetricSlice, error) {
50-
_, span := trace.StartSpan(ctx, "loadscraper.ScrapeMetrics")
51-
defer span.End()
52-
48+
func (s *scraper) ScrapeMetrics(_ context.Context) (pdata.MetricSlice, error) {
5349
metrics := pdata.NewMetricSlice()
5450

5551
avgLoadValues, err := getSampledLoadAverages()

receiver/hostmetricsreceiver/internal/scraper/memoryscraper/factory.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"go.uber.org/zap"
2121

2222
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal"
23+
"go.opentelemetry.io/collector/receiver/hostmetricsreceiver/internal/scraper/obsreportscraper"
2324
)
2425

2526
// This file implements Factory for Memory scraper.
@@ -45,5 +46,5 @@ func (f *Factory) CreateMetricsScraper(
4546
config internal.Config,
4647
) (internal.Scraper, error) {
4748
cfg := config.(*Config)
48-
return newMemoryScraper(ctx, cfg), nil
49+
return obsreportscraper.WrapScraper(newMemoryScraper(ctx, cfg), TypeStr), nil
4950
}

receiver/hostmetricsreceiver/internal/scraper/memoryscraper/memory_scraper.go

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@ import (
1919
"time"
2020

2121
"github.com/shirou/gopsutil/mem"
22-
"go.opencensus.io/trace"
2322

2423
"go.opentelemetry.io/collector/consumer/pdata"
2524
)
@@ -45,10 +44,7 @@ func (s *scraper) Close(_ context.Context) error {
4544
}
4645

4746
// ScrapeMetrics
48-
func (s *scraper) ScrapeMetrics(ctx context.Context) (pdata.MetricSlice, error) {
49-
_, span := trace.StartSpan(ctx, "memoryscraper.ScrapeMetrics")
50-
defer span.End()
51-
47+
func (s *scraper) ScrapeMetrics(_ context.Context) (pdata.MetricSlice, error) {
5248
metrics := pdata.NewMetricSlice()
5349

5450
memInfo, err := mem.VirtualMemory()

0 commit comments

Comments
 (0)