blob: e77d7b780a17f02dbfcaf7a6b665cf592a4f1677 (
plain)
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
|
// SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
package common
import (
"encoding/json"
"fmt"
"strings"
)
type Annotator struct {
Annotator string
// including AnnotatorType: one of "Person", "Organization" or "Tool"
AnnotatorType string
}
// UnmarshalJSON takes an annotator in the typical one-line format and parses it into an Annotator struct.
// This function is also used when unmarshalling YAML
func (a *Annotator) UnmarshalJSON(data []byte) error {
// annotator will simply be a string
annotatorStr := string(data)
annotatorStr = strings.Trim(annotatorStr, "\"")
annotatorFields := strings.SplitN(annotatorStr, ": ", 2)
if len(annotatorFields) != 2 {
return fmt.Errorf("failed to parse Annotator '%s'", annotatorStr)
}
a.AnnotatorType = annotatorFields[0]
a.Annotator = annotatorFields[1]
return nil
}
// MarshalJSON converts the receiver into a slice of bytes representing an Annotator in string form.
// This function is also used when marshalling to YAML
func (a Annotator) MarshalJSON() ([]byte, error) {
if a.Annotator != "" {
return json.Marshal(fmt.Sprintf("%s: %s", a.AnnotatorType, a.Annotator))
}
return []byte{}, nil
}
|
