Merge pull request #237 from k1LoW/store-timeout

Support `timeout:` and set default timeout

cmd/root.go

@@ -64,7 +64,6 @@ var rootCmd = &cobra.Command{
 			return printMetrics(cmd)
 		}
 
-		ctx := context.Background()
 		addPaths := []string{}
 		cmd.PrintErrf("%s version %s\n", version.Name, version.Version)
 
@@ -78,6 +77,9 @@ var rootCmd = &cobra.Command{
 			cmd.PrintErrf("%s are not found\n", strings.Join(config.DefaultConfigFilePaths, " and "))
 		}
 
+		ctx, cancel := context.WithTimeout(context.Background(), c.Timeout)
+		defer cancel()
+
 		if reportPath != "" {
 			c.Coverage.Paths = []string{reportPath}
 			c.CodeToTestRatio = nil

config/config.go

@@ -21,6 +21,7 @@ import (
 
 const defaultBadgesDatastore = "local://reports"
 const defaultReportsDatastore = "local://reports"
+const defaultTimeout = "30sec"
 const largeEnoughTime = float64(99 * time.Hour)
 
 const (
@@ -46,6 +47,7 @@ type Config struct {
 	Summary           *ConfigSummary           `yaml:"summary,omitempty"`
 	Body              *ConfigBody              `yaml:"body,omitempty"`
 	Diff              *ConfigDiff              `yaml:"diff,omitempty"`
+	Timeout           time.Duration            `yaml:"timeout,omitempty"`
 	GitRoot           string                   `yaml:"-"`
 	// working directory
 	wd string

config/yaml.go

@@ -4,6 +4,7 @@ import (
 	"regexp"
 
 	"github.com/goccy/go-yaml"
+	"github.com/k1LoW/duration"
 )
 
 var commentRe = regexp.MustCompile(`(?m)^comment:`)
@@ -23,6 +24,7 @@ func (c *Config) UnmarshalYAML(data []byte) error {
 		Summary           *ConfigSummary           `yaml:"summary,omitempty"`
 		Body              *ConfigBody              `yaml:"body,omitempty"`
 		Diff              *ConfigDiff              `yaml:"diff,omitempty"`
+		Timeout           string                   `yaml:"timeout,omitempty"`
 	}{}
 	err := yaml.Unmarshal(data, &s)
 	if err != nil {
@@ -37,6 +39,13 @@ func (c *Config) UnmarshalYAML(data []byte) error {
 	c.Summary = s.Summary
 	c.Body = s.Body
 	c.Diff = s.Diff
+	if s.Timeout == "" {
+		s.Timeout = defaultTimeout
+	}
+	c.Timeout, err = duration.Parse(s.Timeout)
+	if err != nil {
+		return err
+	}
 
 	switch v := s.Comment.(type) {
 	case nil: