Refactor: Use context in Fetch method

elastic · Dec 10, 2024 · aba1559 · aba1559
1 parent 79260f5
commit aba1559
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/metricbeat/module/pgbouncer/lists/lists.go b/metricbeat/module/pgbouncer/lists/lists.go
@@ -53,8 +53,7 @@ func New(base mb.BaseMetricSet) (mb.MetricSet, error) {
 
 // Fetch methods implements the data gathering and data conversion to the right format
 // It publishes the event which is then forwarded to the output. In case of an error, an error is reported.
-func (m *MetricSet) Fetch(reporter mb.ReporterV2) error {
-	ctx := context.Background()
+func (m *MetricSet) Fetch(ctx context.Context, reporter mb.ReporterV2) error {
 	results, err := m.QueryStats(ctx, "SHOW LISTS;")
 	if err != nil {
 		return fmt.Errorf("error in QueryStats: %w", err)

diff --git a/metricbeat/module/pgbouncer/pools/pools.go b/metricbeat/module/pgbouncer/pools/pools.go
@@ -50,8 +50,7 @@ func New(base mb.BaseMetricSet) (mb.MetricSet, error) {
 
 // Fetch methods implements the data gathering and data conversion to the right format
 // It publishes the event which is then forwarded to the output. In case of an error, an error is reported.
-func (m *MetricSet) Fetch(reporter mb.ReporterV2) error {
-	ctx := context.Background()
+func (m *MetricSet) Fetch(ctx context.Context, reporter mb.ReporterV2) error {
 	results, err := m.QueryStats(ctx, "SHOW POOLS;")
 	if err != nil {
 		return fmt.Errorf("error in QueryStats: %w", err)

diff --git a/metricbeat/module/pgbouncer/stats/stats.go b/metricbeat/module/pgbouncer/stats/stats.go
@@ -50,8 +50,7 @@ func New(base mb.BaseMetricSet) (mb.MetricSet, error) {
 
 // Fetch methods implements the data gathering and data conversion to the right format
 // It publishes the event which is then forwarded to the output. In case of an error, an error is reported.
-func (m *MetricSet) Fetch(reporter mb.ReporterV2) error {
-	ctx := context.Background()
+func (m *MetricSet) Fetch(ctx context.Context, reporter mb.ReporterV2) error {
 	results, err := m.QueryStats(ctx, "SHOW STATS;")
 	if err != nil {
 		return fmt.Errorf("error in QueryStats: %w", err)