Merge pull request #475 from AliyunContainerService/start_period

Support start_period for healthcheck in Docker Compose

cli/compose/convert/service.go

@@ -366,7 +366,6 @@ func convertHealthcheck(healthcheck *composetypes.HealthCheckConfig) (*container
 		return nil, nil
 	}
 	var (
-		err                            error
 		timeout, interval, startPeriod time.Duration
 		retries                        int
 	)
@@ -379,23 +378,14 @@ func convertHealthcheck(healthcheck *composetypes.HealthCheckConfig) (*container
 		}, nil
 
 	}
-	if healthcheck.Timeout != "" {
+	if healthcheck.Timeout != nil {
-		timeout, err = time.ParseDuration(healthcheck.Timeout)
+		timeout = *healthcheck.Timeout
-		if err != nil {
-			return nil, err
-		}
 	}
-	if healthcheck.Interval != "" {
+	if healthcheck.Interval != nil {
-		interval, err = time.ParseDuration(healthcheck.Interval)
+		interval = *healthcheck.Interval
-		if err != nil {
-			return nil, err
-		}
 	}
-	if healthcheck.StartPeriod != "" {
+	if healthcheck.StartPeriod != nil {
-		startPeriod, err = time.ParseDuration(healthcheck.StartPeriod)
+		startPeriod = *healthcheck.StartPeriod
-		if err != nil {
-			return nil, err
-		}
 	}
 	if healthcheck.Retries != nil {
 		retries = int(*healthcheck.Retries)

cli/compose/convert/service_test.go

@@ -109,16 +109,18 @@ func TestConvertResourcesOnlyMemory(t *testing.T) {
 
 func TestConvertHealthcheck(t *testing.T) {
 	retries := uint64(10)
+	timeout := 30 * time.Second
+	interval := 2 * time.Millisecond
 	source := &composetypes.HealthCheckConfig{
 		Test:     []string{"EXEC", "touch", "/foo"},
-		Timeout:  "30s",
+		Timeout:  &timeout,
-		Interval: "2ms",
+		Interval: &interval,
 		Retries:  &retries,
 	}
 	expected := &container.HealthConfig{
 		Test:     source.Test,
-		Timeout:  30 * time.Second,
+		Timeout:  timeout,
-		Interval: 2 * time.Millisecond,
+		Interval: interval,
 		Retries:  10,
 	}
 

cli/compose/loader/full-example.yml

@@ -114,6 +114,7 @@ services:
       interval: 10s
       timeout: 1s
       retries: 5
+      start_period: 15s
 
     # Any valid image reference - repo, tag, id, sha
     image: redis

cli/compose/loader/loader_test.go

@@ -756,10 +756,11 @@ func TestFullExample(t *testing.T) {
 			"somehost":  "162.242.195.82",
 		},
 		HealthCheck: &types.HealthCheckConfig{
-			Test:     types.HealthCheckTest([]string{"CMD-SHELL", "echo \"hello world\""}),
+			Test:        types.HealthCheckTest([]string{"CMD-SHELL", "echo \"hello world\""}),
-			Interval: "10s",
+			Interval:    durationPtr(10 * time.Second),
-			Timeout:  "1s",
+			Timeout:     durationPtr(1 * time.Second),
-			Retries:  uint64Ptr(5),
+			Retries:     uint64Ptr(5),
+			StartPeriod: durationPtr(15 * time.Second),
 		},
 		Hostname: "foo",
 		Image:    "redis",

cli/compose/schema/data/config_schema_v3.4.json

@@ -316,7 +316,7 @@
       "additionalProperties": false,
       "properties": {
         "disable": {"type": "boolean"},
-        "interval": {"type": "string"},
+        "interval": {"type": "string", "format": "duration"},
         "retries": {"type": "number"},
         "test": {
           "oneOf": [
@@ -324,7 +324,8 @@
             {"type": "array", "items": {"type": "string"}}
           ]
         },
-        "timeout": {"type": "string"}
+        "timeout": {"type": "string", "format": "duration"},
+        "start_period": {"type": "string", "format": "duration"}
       }
     },
     "deployment": {

cli/compose/types/types.go

@@ -170,10 +170,10 @@ type DeployConfig struct {
 // HealthCheckConfig the healthcheck configuration for a service
 type HealthCheckConfig struct {
 	Test        HealthCheckTest
-	Timeout     string
+	Timeout     *time.Duration
-	Interval    string
+	Interval    *time.Duration
 	Retries     *uint64
-	StartPeriod string
+	StartPeriod *time.Duration `mapstructure:"start_period"`
 	Disable     bool
 }