add vendoring with go dep

vendor/github.com/influxdata/influxdb/toml/toml.go

@@ -0,0 +1,74 @@
+// Package toml adds support to marshal and unmarshal types not in the official TOML spec.
+package toml // import "github.com/influxdata/influxdb/toml"
+
+import (
+	"fmt"
+	"strconv"
+	"time"
+)
+
+// maxInt is the largest integer representable by a word (architecture dependent).
+const maxInt = int64(^uint(0) >> 1)
+
+// Duration is a TOML wrapper type for time.Duration.
+type Duration time.Duration
+
+// String returns the string representation of the duration.
+func (d Duration) String() string {
+	return time.Duration(d).String()
+}
+
+// UnmarshalText parses a TOML value into a duration value.
+func (d *Duration) UnmarshalText(text []byte) error {
+	// Ignore if there is no value set.
+	if len(text) == 0 {
+		return nil
+	}
+
+	// Otherwise parse as a duration formatted string.
+	duration, err := time.ParseDuration(string(text))
+	if err != nil {
+		return err
+	}
+
+	// Set duration and return.
+	*d = Duration(duration)
+	return nil
+}
+
+// MarshalText converts a duration to a string for decoding toml
+func (d Duration) MarshalText() (text []byte, err error) {
+	return []byte(d.String()), nil
+}
+
+// Size represents a TOML parseable file size.
+// Users can specify size using "m" for megabytes and "g" for gigabytes.
+type Size int
+
+// UnmarshalText parses a byte size from text.
+func (s *Size) UnmarshalText(text []byte) error {
+	// Parse numeric portion of value.
+	length := len(string(text))
+	size, err := strconv.ParseInt(string(text[:length-1]), 10, 64)
+	if err != nil {
+		return err
+	}
+
+	// Parse unit of measure ("m", "g", etc).
+	switch suffix := text[len(text)-1]; suffix {
+	case 'm':
+		size *= 1 << 20 // MB
+	case 'g':
+		size *= 1 << 30 // GB
+	default:
+		return fmt.Errorf("unknown size suffix: %c", suffix)
+	}
+
+	// Check for overflow.
+	if size > maxInt {
+		return fmt.Errorf("size %d cannot be represented by an int", size)
+	}
+
+	*s = Size(size)
+	return nil
+}

vendor/github.com/influxdata/influxdb/toml/toml_test.go

@@ -0,0 +1,45 @@
+package toml_test
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/BurntSushi/toml"
+	"github.com/influxdata/influxdb/cmd/influxd/run"
+	itoml "github.com/influxdata/influxdb/toml"
+)
+
+// Ensure that megabyte sizes can be parsed.
+func TestSize_UnmarshalText_MB(t *testing.T) {
+	var s itoml.Size
+	if err := s.UnmarshalText([]byte("200m")); err != nil {
+		t.Fatalf("unexpected error: %s", err)
+	} else if s != 200*(1<<20) {
+		t.Fatalf("unexpected size: %d", s)
+	}
+}
+
+// Ensure that gigabyte sizes can be parsed.
+func TestSize_UnmarshalText_GB(t *testing.T) {
+	var s itoml.Size
+	if err := s.UnmarshalText([]byte("1g")); err != nil {
+		t.Fatalf("unexpected error: %s", err)
+	} else if s != 1073741824 {
+		t.Fatalf("unexpected size: %d", s)
+	}
+}
+
+func TestConfig_Encode(t *testing.T) {
+	var c run.Config
+	c.Coordinator.WriteTimeout = itoml.Duration(time.Minute)
+	buf := new(bytes.Buffer)
+	if err := toml.NewEncoder(buf).Encode(&c); err != nil {
+		t.Fatal("Failed to encode: ", err)
+	}
+	got, search := buf.String(), `write-timeout = "1m0s"`
+	if !strings.Contains(got, search) {
+		t.Fatalf("Encoding config failed.\nfailed to find %s in:\n%s\n", search, got)
+	}
+}