1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2014-2015 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package timeout
import (
"encoding/json"
"time"
)
// Timeout is a time.Duration that knows how to roundtrip to json and yaml
type Timeout time.Duration
// DefaultTimeout specifies the timeout for services that do not specify StopTimeout
var DefaultTimeout = Timeout(30 * time.Second)
// MarshalJSON is from the json.Marshaler interface
func (t Timeout) MarshalJSON() ([]byte, error) {
return json.Marshal(t.String())
}
// UnmarshalJSON is from the json.Unmarshaler interface
func (t *Timeout) UnmarshalJSON(buf []byte) error {
var str string
if err := json.Unmarshal(buf, &str); err != nil {
return err
}
dur, err := time.ParseDuration(str)
if err != nil {
return err
}
*t = Timeout(dur)
return nil
}
func (t *Timeout) UnmarshalYAML(unmarshal func(any) error) error {
var str string
if err := unmarshal(&str); err != nil {
return err
}
dur, err := time.ParseDuration(str)
if err != nil {
return err
}
*t = Timeout(dur)
return nil
}
// String returns a string representing the duration
func (t Timeout) String() string {
return time.Duration(t).String()
}
// Seconds returns the duration as a floating point number of seconds.
func (t Timeout) Seconds() float64 {
return time.Duration(t).Seconds()
}
|
