Renaming CertStreamWrapper to CertStreamRaw for more clarity
This commit is contained in:
parent
e38b05de66
commit
01f7ae70b9
2
main.go
2
main.go
|
@ -46,7 +46,7 @@ func main() {
|
||||||
if data, err := models.ExtractCertFromStream(jq); err == nil {
|
if data, err := models.ExtractCertFromStream(jq); err == nil {
|
||||||
|
|
||||||
rawNode := models.WrapCertStreamData(*data)
|
rawNode := models.WrapCertStreamData(*data)
|
||||||
models.SaveCertStreamWrapper("raw_certstream.json", rawNode)
|
models.SaveCertStreamRaw("raw_certstream.json", rawNode)
|
||||||
|
|
||||||
certNode := models.BuildCertNode(rawNode)
|
certNode := models.BuildCertNode(rawNode)
|
||||||
models.SaveCertNode("cert_nodes.json", certNode)
|
models.SaveCertNode("cert_nodes.json", certNode)
|
||||||
|
|
|
@ -124,9 +124,9 @@ func SaveEdge(edge *Edge) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// CertStreamWrapper is a wrapper around the stream function to unmarshall the
|
// CertStreamRaw is a wrapper around the stream function to unmarshall the
|
||||||
// data receive in a Go structure.
|
// data receive in a Go structure.
|
||||||
type CertStreamWrapper struct {
|
type CertStreamRaw struct {
|
||||||
ID string `json:"id"`
|
ID string `json:"id"`
|
||||||
Type string `json:"type"`
|
Type string `json:"type"`
|
||||||
Data CertStreamStruct `json:"data"`
|
Data CertStreamStruct `json:"data"`
|
||||||
|
@ -149,10 +149,10 @@ type CertNode struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
// WrapCertStreamData is a wrapper around CertStreamStruct.
|
// WrapCertStreamData is a wrapper around CertStreamStruct.
|
||||||
func WrapCertStreamData(data CertStreamStruct) *CertStreamWrapper {
|
func WrapCertStreamData(data CertStreamStruct) *CertStreamRaw {
|
||||||
t := time.Now()
|
t := time.Now()
|
||||||
rfc3339time := t.Format(time.RFC3339)
|
rfc3339time := t.Format(time.RFC3339)
|
||||||
return &CertStreamWrapper{
|
return &CertStreamRaw{
|
||||||
ID: "certstream--" + uuid.New().String(),
|
ID: "certstream--" + uuid.New().String(),
|
||||||
Type: "certstream_raw",
|
Type: "certstream_raw",
|
||||||
Data: data,
|
Data: data,
|
||||||
|
@ -162,7 +162,7 @@ func WrapCertStreamData(data CertStreamStruct) *CertStreamWrapper {
|
||||||
}
|
}
|
||||||
|
|
||||||
// BuildCertNode builds a custom node based on CertStream.
|
// BuildCertNode builds a custom node based on CertStream.
|
||||||
func BuildCertNode(rawNode *CertStreamWrapper) *CertNode {
|
func BuildCertNode(rawNode *CertStreamRaw) *CertNode {
|
||||||
main := &CertNode{
|
main := &CertNode{
|
||||||
ID: "certstream--" + uuid.New().String(),
|
ID: "certstream--" + uuid.New().String(),
|
||||||
Fingerprint: rawNode.Data.Data.LeafCert.Fingerprint,
|
Fingerprint: rawNode.Data.Data.LeafCert.Fingerprint,
|
||||||
|
@ -193,8 +193,8 @@ func BuildCertNode(rawNode *CertStreamWrapper) *CertNode {
|
||||||
return main
|
return main
|
||||||
}
|
}
|
||||||
|
|
||||||
// SaveCertStreamWrapper save the raw CertStream data.
|
// SaveCertStreamRaw save the raw CertStream data.
|
||||||
func SaveCertStreamWrapper(filename string, data *CertStreamWrapper) {
|
func SaveCertStreamRaw(filename string, data *CertStreamRaw) {
|
||||||
err := utils.FileExists(filename)
|
err := utils.FileExists(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logrus.Error(err)
|
logrus.Error(err)
|
||||||
|
@ -203,7 +203,7 @@ func SaveCertStreamWrapper(filename string, data *CertStreamWrapper) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logrus.Error(err)
|
logrus.Error(err)
|
||||||
}
|
}
|
||||||
rawDatas := []CertStreamWrapper{}
|
rawDatas := []CertStreamRaw{}
|
||||||
|
|
||||||
if err := json.Unmarshal(nodeFile, &rawDatas); err != nil {
|
if err := json.Unmarshal(nodeFile, &rawDatas); err != nil {
|
||||||
logrus.Error(err)
|
logrus.Error(err)
|
||||||
|
|
Loading…
Reference in a new issue