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
|
package types
import (
"database/sql/driver"
"encoding"
"encoding/json"
"github.com/icinga/icingadb/internal"
"github.com/pkg/errors"
)
// AcknowledgementState specifies an acknowledgement state (yes, no, sticky).
type AcknowledgementState uint8
// UnmarshalText implements the encoding.TextUnmarshaler interface.
func (as *AcknowledgementState) UnmarshalText(text []byte) error {
return as.UnmarshalJSON(text)
}
// UnmarshalJSON implements the json.Unmarshaler interface.
func (as *AcknowledgementState) UnmarshalJSON(data []byte) error {
var i uint8
if err := internal.UnmarshalJSON(data, &i); err != nil {
return err
}
a := AcknowledgementState(i)
if _, ok := acknowledgementStates[a]; !ok {
return badAcknowledgementState(data)
}
*as = a
return nil
}
// Value implements the driver.Valuer interface.
func (as AcknowledgementState) Value() (driver.Value, error) {
if v, ok := acknowledgementStates[as]; ok {
return v, nil
} else {
return nil, badAcknowledgementState(as)
}
}
// badAcknowledgementState returns an error about a syntactically, but not semantically valid AcknowledgementState.
func badAcknowledgementState(s interface{}) error {
return errors.Errorf("bad acknowledgement state: %#v", s)
}
// acknowledgementStates maps all valid AcknowledgementState values to their SQL representation.
var acknowledgementStates = map[AcknowledgementState]string{
0: "n",
1: "y",
2: "sticky",
}
// Assert interface compliance.
var (
_ encoding.TextUnmarshaler = (*AcknowledgementState)(nil)
_ json.Unmarshaler = (*AcknowledgementState)(nil)
_ driver.Valuer = AcknowledgementState(0)
)
|