feat(outputs/http): Allow keepalive configuration (#138)

ethpandaops · Jul 13, 2023 · ab333a0 · ab333a0
1 parent 6c01149
commit ab333a0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pkg/output/http/config.go b/pkg/output/http/config.go
@@ -13,6 +13,7 @@ type Config struct {
  ExportTimeout time.Duration `yaml:"exportTimeout" default:"30s"`
  MaxExportBatchSize int `yaml:"maxExportBatchSize" default:"512"`
  Compression CompressionStrategy `yaml:"compression" default:"none"`
+ KeepAlive bool `yaml:"keepAlive" default:"true"`
 }
 
 func (c *Config) Validate() error {

diff --git a/pkg/output/http/exporter.go b/pkg/output/http/exporter.go
@@ -21,12 +21,18 @@ type ItemExporter struct {
 }
 
 func NewItemExporter(name string, config *Config, log logrus.FieldLogger) (ItemExporter, error) {
+ t := http.DefaultTransport.(*http.Transport).Clone()
+ if !config.KeepAlive {
+ t.DisableKeepAlives = true
+ }
+
  return ItemExporter{
  config: config,
  log: log.WithField("output_name", name).WithField("output_type", SinkType),
 
  client: &http.Client{
- Timeout: config.ExportTimeout,
+ Transport: t,
+ Timeout: config.ExportTimeout,
  },
  }, nil
 }