diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 01:21:45 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 01:21:45 -0500 |
commit | 07390eeb960df3ca10b758c9f62459cf2ee9aa3f (patch) | |
tree | 6612ba4fe2f7252a3096bc2bb250aca11d86636f /internal/feed/dataformat.go | |
parent | a824a9732154f9bee2ce530f753d5747de78aa3f (diff) | |
download | cvez-07390eeb960df3ca10b758c9f62459cf2ee9aa3f.tar.bz2 cvez-07390eeb960df3ca10b758c9f62459cf2ee9aa3f.zip |
add internal/feed/dataformat.go
Diffstat (limited to 'internal/feed/dataformat.go')
-rw-r--r-- | internal/feed/dataformat.go | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/internal/feed/dataformat.go b/internal/feed/dataformat.go new file mode 100644 index 0000000..bb3f8f8 --- /dev/null +++ b/internal/feed/dataformat.go @@ -0,0 +1,36 @@ +package feed + +//go:generate stringer -linecomment -type=DataFormat + +import ( + "encoding/json" + "fmt" +) + +// Data format for NVD feeds and feed items. +type DataFormat byte + +const ( + MitreFormat DataFormat = iota // MITRE +) + +// Unmarshal DataFormat from JSON. +func (me *DataFormat) 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 "MITRE": + *me = MitreFormat + default: + // return error + return fmt.Errorf("unknown data format: %s", s) + } + + // return success + return nil +} |