metricator/internal/application/fetcher.go

86 lines
1.7 KiB
Go

package application
import (
"io/ioutil"
"net/http"
"time"
)
// Fetches data from remote endpoint, parses it and updates in storage.
func (a *Application) fetch() {
// Do not do anything if fetching is running.
// ToDo: maybe another approach?
a.fetchIsRunningMutex.RLock()
isFetching := a.fetchIsRunning
a.fetchIsRunningMutex.RUnlock()
if isFetching {
return
}
a.fetchIsRunningMutex.Lock()
a.fetchIsRunning = true
a.fetchIsRunningMutex.Unlock()
a.logger.Infoln("Fetching data for", a.name)
req, err := http.NewRequestWithContext(a.ctx, "GET", a.config.Endpoint, nil)
if err != nil {
a.logger.Infoln("Failed to create request for", a.name, "metrics:", err.Error())
return
}
for header, value := range a.config.Headers {
req.Header.Add(header, value)
}
resp, err := a.httpClient.Do(req)
if err != nil {
a.logger.Infoln("Failed to execute request for", a.name, "metrics:", err.Error())
return
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
a.logger.Infoln("Failed to read response body for", a.name, "metrics:", err.Error())
return
}
data := a.parse(string(body))
a.storage.Put(data)
a.fetchIsRunningMutex.Lock()
a.fetchIsRunning = false
a.fetchIsRunningMutex.Unlock()
}
// Configures and starts Prometheus data fetching goroutine.
func (a *Application) startFetcher() {
fetchTicker := time.NewTicker(a.config.TimeBetweenRequests)
// nolint:exhaustivestruct
a.httpClient = &http.Client{
Timeout: time.Second * 5,
}
defer a.logger.Debugln("Fetcher for", a.name, "completed")
// First fetch should be executed ASAP.
a.fetch()
for {
select {
case <-a.ctx.Done():
return
case <-fetchTicker.C:
a.fetch()
}
}
}