missing response time in docker provider

umputun · Feb 6, 2022 · b5e3919 · b5e3919
1 parent a6e41b3
commit b5e3919
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/app/status/external/docker_provider.go b/app/status/external/docker_provider.go
@@ -23,6 +23,7 @@ type DockerProvider struct {
 // i.e. docker:///var/run/docker.sock?containers=foo,bar
 func (d *DockerProvider) Status(req Request) (*Response, error) {
 
+	st := time.Now()
 	u := strings.Replace(req.URL, "docker://", "tcp://", 1)
 	if strings.HasPrefix(req.URL, "docker:///") { // i.e. docker:///var/run/docker.sock
 		u = strings.Replace(req.URL, "docker://", "unix://", 1)
@@ -68,9 +69,10 @@ func (d *DockerProvider) Status(req Request) (*Response, error) {
 	}
 
 	result := Response{
-		Name:       req.Name,
-		StatusCode: resp.StatusCode,
-		Body:       dkinfo,
+		Name:         req.Name,
+		StatusCode:   resp.StatusCode,
+		Body:         dkinfo,
+		ResponseTime: time.Since(st).Milliseconds(),
 	}
 	return &result, nil
 }

diff --git a/app/status/external/docker_provider_test.go b/app/status/external/docker_provider_test.go
@@ -33,6 +33,7 @@ func TestDockerProvider_Status(t *testing.T) {
 	resp, err := p.Status(Request{Name: "d1", URL: strings.Replace(ts.URL, "http://", "tcp://", 1)})
 	require.NoError(t, err)
 	assert.Equal(t, 200, resp.StatusCode)
+	assert.True(t, resp.ResponseTime > 1, resp.ResponseTime)
 }
 
 func TestDockerProvider_StatusWithRequired(t *testing.T) {