feat(cmd): add timeouts support.

support timeout-related commands

Author: notmaxx

cmd/timeouts.go

@@ -0,0 +1,136 @@
+package cmd
+
+import (
+  "fmt"
+  "regexp"
+
+  "github.com/deis/pkg/prettyprint"
+
+  "github.com/notmaxx/controller-sdk-go/api"
+  "github.com/notmaxx/controller-sdk-go/config"
+)
+
+// LimitsList lists an app's timeouts.
+func (d *DeisCmd) TimeoutsList(appID string) error {
+  s, appID, err := load(d.ConfigFile, appID)
+
+  if err != nil {
+    return err
+  }
+
+  config, err := config.List(s.Client, appID)
+  if d.checkAPICompatibility(s.Client, err) != nil {
+    return err
+  }
+
+  d.Printf("=== %s Timeouts (sec)\n\n", appID)
+
+  if len(config.Timeout) == 0 {
+    d.Println("default (30 sec) or controlled by env KUBERNETES_POD_TERMINATION_GRACE_PERIOD_SECONDS")
+  } else {
+    timeoutsMap := make(map[string]string)
+
+    for key, value := range config.Timeout {
+      timeoutsMap[key] = fmt.Sprintf("%v", value)
+    }
+
+    d.Print(prettyprint.PrettyTabs(timeoutsMap, 5))
+  }
+
+  return nil
+}
+
+// LimitsSet sets an app's timeouts.
+func (d *DeisCmd) TimeoutsSet(appID string, timeouts []string) error {
+  s, appID, err := load(d.ConfigFile, appID)
+
+  if err != nil {
+    return err
+  }
+
+  timeoutsMap, err := parseTimeouts(timeouts)
+  if err != nil {
+    return err
+  }
+
+  d.Print("Applying timeouts... ")
+
+  quit := progress(d.WOut)
+  configObj := api.Config{}
+
+  configObj.Timeout = timeoutsMap
+
+  _, err = config.Set(s.Client, appID, configObj)
+  quit <- true
+  <-quit
+  if d.checkAPICompatibility(s.Client, err) != nil {
+    return err
+  }
+
+  d.Print("done\n\n")
+
+  return d.TimeoutsList(appID)
+}
+
+// LimitsUnset removes an app's timeouts.
+func (d *DeisCmd) TimeoutsUnset(appID string, timeouts []string) error {
+  s, appID, err := load(d.ConfigFile, appID)
+
+  if err != nil {
+    return err
+  }
+
+  d.Print("Applying timeouts... ")
+
+  quit := progress(d.WOut)
+
+  configObj := api.Config{}
+
+  valuesMap := make(map[string]interface{})
+
+  for _, timeout := range timeouts {
+    valuesMap[timeout] = nil
+  }
+
+  configObj.Timeout = valuesMap
+
+  _, err = config.Set(s.Client, appID, configObj)
+  quit <- true
+  <-quit
+  if d.checkAPICompatibility(s.Client, err) != nil {
+    return err
+  }
+
+  d.Print("done\n\n")
+
+  return d.TimeoutsList(appID)
+}
+
+func parseTimeouts(timeouts []string) (map[string]interface{}, error) {
+  timeoutsMap := make(map[string]interface{})
+
+  for _, timeout := range timeouts {
+    key, value, err := parseTimeout(timeout)
+
+    if err != nil {
+      return nil, err
+    }
+
+    timeoutsMap[key] = value
+  }
+
+  return timeoutsMap, nil
+}
+
+func parseTimeout(timeout string) (string, string, error) {
+  regex := regexp.MustCompile("^[0-9]*$")
+
+  if !regex.MatchString(timeout) {
+    return "", "", fmt.Errorf(`%s doesn't fit format type=#
+Examples: web=30 worker=300`, timeout)
+  }
+
+  capture := regex.FindStringSubmatch(timeout)
+
+  return capture[1], capture[2], nil
+}