updates for v0.4.0 release. Slight refactor/organization.
This commit is contained in:
@@ -10,6 +10,7 @@ import (
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
utils "github.com/analogj/go-util/utils"
|
||||
@@ -113,7 +114,10 @@ OPTIONS:
|
||||
}
|
||||
|
||||
if c.IsSet("api-endpoint") {
|
||||
config.Set("api.endpoint", c.String("api-endpoint"))
|
||||
//if the user is providing an api-endpoint with a basepath (eg. http://localhost:8080/scrutiny),
|
||||
//we need to ensure the basepath has a trailing slash, otherwise the url.Parse() path concatenation doesnt work.
|
||||
apiEndpoint := strings.TrimSuffix(c.String("api-endpoint"), "/") + "/"
|
||||
config.Set("api.endpoint", apiEndpoint)
|
||||
}
|
||||
|
||||
collectorLogger := logrus.WithFields(logrus.Fields{
|
||||
|
||||
@@ -48,7 +48,7 @@ func (mc *MetricsCollector) Run() error {
|
||||
}
|
||||
|
||||
apiEndpoint, _ := url.Parse(mc.apiEndpoint.String())
|
||||
apiEndpoint.Path = "/api/devices/register"
|
||||
apiEndpoint, _ = apiEndpoint.Parse("api/devices/register") //this acts like filepath.Join()
|
||||
|
||||
deviceRespWrapper := new(models.DeviceWrapper)
|
||||
|
||||
@@ -73,6 +73,7 @@ func (mc *MetricsCollector) Run() error {
|
||||
|
||||
if !deviceRespWrapper.Success {
|
||||
mc.logger.Errorln("An error occurred while retrieving filtered devices")
|
||||
mc.logger.Debugln(deviceRespWrapper)
|
||||
return errors.ApiServerCommunicationError("An error occurred while retrieving filtered devices")
|
||||
} else {
|
||||
mc.logger.Debugln(deviceRespWrapper)
|
||||
@@ -146,7 +147,7 @@ func (mc *MetricsCollector) Publish(deviceWWN string, payload []byte) error {
|
||||
mc.logger.Infof("Publishing smartctl results for %s\n", deviceWWN)
|
||||
|
||||
apiEndpoint, _ := url.Parse(mc.apiEndpoint.String())
|
||||
apiEndpoint.Path = fmt.Sprintf("/api/device/%s/smart", strings.ToLower(deviceWWN))
|
||||
apiEndpoint, _ = apiEndpoint.Parse(fmt.Sprintf("api/device/%s/smart", strings.ToLower(deviceWWN)))
|
||||
|
||||
resp, err := httpClient.Post(apiEndpoint.String(), "application/json", bytes.NewBuffer(payload))
|
||||
if err != nil {
|
||||
|
||||
@@ -0,0 +1,38 @@
|
||||
package collector
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/require"
|
||||
"net/url"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestApiEndpointParse(t *testing.T) {
|
||||
baseURL, _ := url.Parse("http://localhost:8080/")
|
||||
|
||||
url1, _ := baseURL.Parse("d/e")
|
||||
require.Equal(t, "http://localhost:8080/d/e", url1.String())
|
||||
|
||||
url2, _ := baseURL.Parse("/d/e")
|
||||
require.Equal(t, "http://localhost:8080/d/e", url2.String())
|
||||
}
|
||||
|
||||
func TestApiEndpointParse_WithBasepathWithoutTrailingSlash(t *testing.T) {
|
||||
baseURL, _ := url.Parse("http://localhost:8080/scrutiny")
|
||||
|
||||
//This testcase is unexpected and can cause issues. We need to ensure the apiEndpoint always has a trailing slash.
|
||||
url1, _ := baseURL.Parse("d/e")
|
||||
require.Equal(t, "http://localhost:8080/d/e", url1.String())
|
||||
|
||||
url2, _ := baseURL.Parse("/d/e")
|
||||
require.Equal(t, "http://localhost:8080/d/e", url2.String())
|
||||
}
|
||||
|
||||
func TestApiEndpointParse_WithBasepathWithTrailingSlash(t *testing.T) {
|
||||
baseURL, _ := url.Parse("http://localhost:8080/scrutiny/")
|
||||
|
||||
url1, _ := baseURL.Parse("d/e")
|
||||
require.Equal(t, "http://localhost:8080/scrutiny/d/e", url1.String())
|
||||
|
||||
url2, _ := baseURL.Parse("/d/e")
|
||||
require.Equal(t, "http://localhost:8080/d/e", url2.String())
|
||||
}
|
||||
Reference in New Issue
Block a user