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
77
78
79
|
package types
import (
"database/sql/driver"
"encoding"
"encoding/json"
"github.com/icinga/icingadb/internal"
"github.com/pkg/errors"
"strconv"
)
// CommentType specifies a comment's origin's kind.
type CommentType uint8
// UnmarshalJSON implements the json.Unmarshaler interface.
func (ct *CommentType) UnmarshalJSON(data []byte) error {
var i uint8
if err := internal.UnmarshalJSON(data, &i); err != nil {
return err
}
c := CommentType(i)
if _, ok := commentTypes[c]; !ok {
return badCommentType(data)
}
*ct = c
return nil
}
// UnmarshalText implements the encoding.TextUnmarshaler interface.
func (ct *CommentType) UnmarshalText(text []byte) error {
s := string(text)
i, err := strconv.ParseUint(s, 10, 64)
if err != nil {
return internal.CantParseUint64(err, s)
}
c := CommentType(i)
if uint64(c) != i {
// Truncated due to above cast, obviously too high
return badCommentType(s)
}
if _, ok := commentTypes[c]; !ok {
return badCommentType(s)
}
*ct = c
return nil
}
// Value implements the driver.Valuer interface.
func (ct CommentType) Value() (driver.Value, error) {
if v, ok := commentTypes[ct]; ok {
return v, nil
} else {
return nil, badCommentType(ct)
}
}
// badCommentType returns an error about a syntactically, but not semantically valid CommentType.
func badCommentType(t interface{}) error {
return errors.Errorf("bad comment type: %#v", t)
}
// commentTypes maps all valid CommentType values to their SQL representation.
var commentTypes = map[CommentType]string{
1: "comment",
4: "ack",
}
// Assert interface compliance.
var (
_ json.Unmarshaler = (*CommentType)(nil)
_ encoding.TextUnmarshaler = (*CommentType)(nil)
_ driver.Valuer = CommentType(0)
)
|
