diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 07:56:12 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 07:56:12 -0500 |
commit | 5dbf57134000f5ff2c5c12342ac4439b6d395603 (patch) | |
tree | a3a8bb5abcb77d434c18cca61c2c6eba66f7b423 /internal/feed/nodeop.go | |
parent | 4ab54153c1448b6df9701efcd4b86e63e71c0760 (diff) | |
download | cvez-5dbf57134000f5ff2c5c12342ac4439b6d395603.tar.bz2 cvez-5dbf57134000f5ff2c5c12342ac4439b6d395603.zip |
internal/feed: add nodeop tests
Diffstat (limited to 'internal/feed/nodeop.go')
-rw-r--r-- | internal/feed/nodeop.go | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/internal/feed/nodeop.go b/internal/feed/nodeop.go new file mode 100644 index 0000000..8bfa0a0 --- /dev/null +++ b/internal/feed/nodeop.go @@ -0,0 +1,39 @@ +package feed + +//go:generate stringer -linecomment -type=NodeOp + +import ( + "encoding/json" + "fmt" +) + +// Node boolean operator. +type NodeOp byte + +const ( + OrOp NodeOp = iota // OR + AndOp // AND +) + +// Unmarshal DataVersion from JSON. +func (me *NodeOp) UnmarshalJSON(b []byte) error { + // decode string, check for error + var s string + if err := json.Unmarshal(b, &s); err != nil { + return err + } + + // check value + switch s { + case "AND": + *me = AndOp + case "OR": + *me = OrOp + default: + // return error + return fmt.Errorf("unknown operator: %s", s) + } + + // return success + return nil +} |