Index: go/src/infra/libs/clock/duration.go |
diff --git a/go/src/infra/libs/clock/duration.go b/go/src/infra/libs/clock/duration.go |
new file mode 100644 |
index 0000000000000000000000000000000000000000..265c0e19614a96309c36c41046a23f7e955d442a |
--- /dev/null |
+++ b/go/src/infra/libs/clock/duration.go |
@@ -0,0 +1,59 @@ |
+// Copyright (c) 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+package clock |
iannucci
2015/06/02 06:15:33
maybe this could go in
clock/clockflag
?
Then
dnj
2015/06/02 17:03:12
Done.
|
+ |
+import ( |
+ "errors" |
+ "fmt" |
+ "time" |
+) |
+ |
+// DurationFlag is a Flag- and JSON-compatible Duration value. |
+type DurationFlag time.Duration |
+ |
+// Set implements flag.Value. |
+func (d *DurationFlag) Set(value string) error { |
+ duration, err := time.ParseDuration(value) |
+ if err != nil { |
+ return err |
+ } |
+ *d = DurationFlag(duration) |
+ return nil |
+} |
+ |
+// String implements flag.Value. |
+func (d *DurationFlag) String() string { |
+ return time.Duration(*d).String() |
+} |
+ |
+// Duration returns the Duration value of the flag. |
+func (d DurationFlag) Duration() time.Duration { |
+ return time.Duration(d) |
+} |
+ |
+// IsZero tests if this DurationFlag is the zero value. |
+func (d DurationFlag) IsZero() bool { |
+ return time.Duration(d) == time.Duration(0) |
+} |
+ |
+// UnmarshalJSON implements json.Unmarshaler. |
+// |
+// Unmarshals a JSON entry into the underlying type. The entry is expected to contain |
+// a string corresponding to one of the enum's keys. |
+func (d *DurationFlag) UnmarshalJSON(data []byte) error { |
+ // Strip off leading and trailing quotes. |
+ s := string(data) |
+ if len(s) < 2 || s[0] != '"' || s[len(s)-1] != '"' { |
+ return errors.New("Duration JSON must be a valid JSON string.") |
+ } |
+ return d.Set(s[1 : len(s)-1]) |
+} |
+ |
+// MarshalJSON implements json.Marshaler. |
+// |
+// Marshals a DurationFlag into a duration string. |
+func (d DurationFlag) MarshalJSON() ([]byte, error) { |
+ return []byte(fmt.Sprintf(`"%s"`, d.Duration().String())), nil |
+} |