Add new httpClient parameter to NewCollector (can be nil)

dmwm · Sep 11, 2024 · 7f8c3f6 · 7f8c3f6
1 parent 19f4aaa
commit 7f8c3f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/logging/collector.go b/logging/collector.go
@@ -22,13 +22,16 @@ type Collector struct {
 }
 
 // NewCollector initializes and returns a new Collector
-func NewCollector(maxSize int, endpoint, login, password string) *Collector {
+func NewCollector(maxSize int, endpoint, login, password string, httpClient *http.Client) *Collector {
  auth := base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s:%s", login, password)))
+ if httpClient == nil {
+ httpClient = &http.Client{}
+ }
  return &Collector{
  records: make([]HTTPRecord, 0, maxSize),
  maxSize: maxSize,
  endpoint: endpoint,
- httpClient: &http.Client{},
+ httpClient: httpClient,
  authHeader: "Basic " + auth,
  }
 }

diff --git a/logging/logging.go b/logging/logging.go
@@ -242,7 +242,7 @@ func LogRequest(w http.ResponseWriter, r *http.Request, start time.Time, cauth s
  if maxSize == 0 {
  maxSize = 1000
  }
- LogCollector = NewCollector(maxSize, CollectorURL, CollectorLogin, CollectorPassword)
+ LogCollector = NewCollector(maxSize, CollectorURL, CollectorLogin, CollectorPassword, nil)
  }
 
  // our apache configuration