-
Notifications
You must be signed in to change notification settings - Fork 22
/
write_mode.go
53 lines (45 loc) · 1.03 KB
/
write_mode.go
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
package specs
import (
"bytes"
"encoding/json"
"fmt"
)
type WriteMode int
const (
WriteModeOverwriteDeleteStale WriteMode = iota
WriteModeOverwrite
WriteModeAppend
)
var (
writeModeStrings = []string{"overwrite-delete-stale", "overwrite", "append"}
)
func (m WriteMode) String() string {
return writeModeStrings[m]
}
func (m WriteMode) MarshalJSON() ([]byte, error) {
buffer := bytes.NewBufferString(`"`)
buffer.WriteString(m.String())
buffer.WriteString(`"`)
return buffer.Bytes(), nil
}
func (m *WriteMode) UnmarshalJSON(data []byte) (err error) {
var writeMode string
if err := json.Unmarshal(data, &writeMode); err != nil {
return err
}
if *m, err = WriteModeFromString(writeMode); err != nil {
return err
}
return nil
}
func WriteModeFromString(s string) (WriteMode, error) {
switch s {
case "append":
return WriteModeAppend, nil
case "overwrite":
return WriteModeOverwrite, nil
case "overwrite-delete-stale":
return WriteModeOverwriteDeleteStale, nil
}
return 0, fmt.Errorf("invalid write mode: %s", s)
}