-
Notifications
You must be signed in to change notification settings - Fork 188
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
9 changed files
with
461 additions
and
11 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,32 @@ | ||
| <!-- | ||
| If you suspect your issue is a bug, please edit your issue description to | ||
| include the BUG REPORT INFORMATION shown below. | ||
| --------------------------------------------------- | ||
| BUG REPORT INFORMATION | ||
| --------------------------------------------------- | ||
| Use the commands below to provide key information from your environment: | ||
| You do NOT have to include this information if this is a FEATURE REQUEST | ||
| --> | ||
|
|
||
| **Description** | ||
|
|
||
| <!-- | ||
| Briefly describe the problem you are having in a few paragraphs. | ||
| --> | ||
|
|
||
| **Steps to reproduce the issue:** | ||
| 1. | ||
| 2. | ||
| 3. | ||
|
|
||
| **Describe the results you received:** | ||
|
|
||
|
|
||
| **Describe the results you expected:** | ||
|
|
||
|
|
||
| **Additional information you deem important (e.g. issue happens only occasionally):** | ||
|
|
||
|
|
||
| **Additional environment details (AWS, VirtualBox, physical, etc.):** |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
| package server | ||
|
|
||
| import ( | ||
| "net/http" | ||
| "fmt" | ||
| "os" | ||
| "strings" | ||
| "io/ioutil" | ||
| "bytes" | ||
| ) | ||
|
|
||
| // TODO: Test with haproxy_exporter and Prometheus | ||
| // NOTE: It does not work until the first service is added | ||
| // TODO: Check whether data should be aggregated | ||
| // TODO: Document | ||
|
|
||
| type Metricer interface { | ||
| Get(w http.ResponseWriter, req *http.Request) | ||
| GetMetricsUrl() string | ||
| } | ||
|
|
||
| type Metrics struct { | ||
| metricsUrl string | ||
| } | ||
|
|
||
| func NewMetrics(metricsUrl string) Metricer { | ||
| if len(metricsUrl) == 0 { | ||
| metricsUrl = fmt.Sprintf("http://%slocalhost/admin?stats;csv", getCreds()) | ||
| } | ||
| return &Metrics{metricsUrl: metricsUrl} | ||
| } | ||
|
|
||
| func (m *Metrics) GetMetricsUrl() string { | ||
| return m.metricsUrl | ||
| } | ||
|
|
||
| func (m *Metrics) Get(w http.ResponseWriter, req *http.Request) { | ||
| contentType := "text/html" | ||
| if strings.EqualFold(req.URL.Query().Get("distribute"), "true") { | ||
| dns := fmt.Sprintf("tasks.%s", os.Getenv("SERVICE_NAME")) | ||
| ips, err := lookupHost(dns) | ||
| if err != nil { | ||
| logPrintf(err.Error()) | ||
| w.WriteHeader(http.StatusInternalServerError) | ||
| } else { | ||
| body, err := m.getAllHaProxyMetrics(req, ips) | ||
| if err != nil { | ||
| w.WriteHeader(http.StatusInternalServerError) | ||
| } | ||
| w.Write(body) | ||
| } | ||
| } else { | ||
| body, err := m.getHaProxyMetrics() | ||
| if err != nil { | ||
| w.WriteHeader(http.StatusInternalServerError) | ||
| } | ||
| w.Write(body) | ||
| } | ||
| httpWriterSetContentType(w, contentType) | ||
| } | ||
|
|
||
| func (m *Metrics) getHaProxyMetrics() ([]byte, error) { | ||
| resp, err := http.Get(m.metricsUrl) | ||
| if err != nil { | ||
| logPrintf("Failed to fetch metrics from %s\nERROR: %s", m.metricsUrl, err.Error()) | ||
| return []byte(""), err | ||
| } | ||
| defer resp.Body.Close() | ||
| return ioutil.ReadAll(resp.Body) | ||
| } | ||
|
|
||
| func (m *Metrics) getAllHaProxyMetrics(req *http.Request, ips []string) ([]byte, error) { | ||
| msg := []byte("") | ||
| for _, ip := range ips { | ||
| values := req.URL.Query() | ||
| values.Set("distribute", "false") | ||
| req.URL.RawQuery = values.Encode() | ||
| port := "" | ||
| if !strings.Contains(ip, ":") { | ||
| port = ":8080" | ||
| } | ||
| addr := fmt.Sprintf("http://%s%s/v1/docker-flow-proxy/metrics?%s", ip, port, req.URL.RawQuery) | ||
| resp, err := http.Get(addr) | ||
| if err != nil { | ||
| logPrintf("Failed to fetch metrics from %s\nERROR: %s", addr, err.Error()) | ||
| return []byte(""), err | ||
| } else { | ||
| defer resp.Body.Close() | ||
| if resp.StatusCode >= 300 { | ||
| return []byte(""), fmt.Errorf("Got response status %d", resp.StatusCode) | ||
| } | ||
| body, _ := ioutil.ReadAll(resp.Body) | ||
| msg = append(msg, body...) | ||
| if !bytes.HasSuffix(msg, []byte("\n")) { | ||
| msg = append(msg, byte('\n')) | ||
| } | ||
| } | ||
| } | ||
| return msg, nil | ||
| } | ||
|
|
||
| func getCreds() string { | ||
| statsUser := getSecretOrEnvVar(os.Getenv("STATS_USER_ENV"), "") | ||
| statsPass := getSecretOrEnvVar(os.Getenv("STATS_PASS_ENV"), "") | ||
| if len(statsUser) > 0 && !strings.EqualFold(statsUser, "none") && len(statsPass) > 0 && !strings.EqualFold(statsPass, "none") { | ||
| return fmt.Sprintf("%s:%s@", statsUser, statsPass) | ||
| } | ||
| return "" | ||
| } |
Oops, something went wrong.