2020-01-14 09:43:12 +01:00
|
|
|
package main
|
|
|
|
|
2020-01-14 14:49:14 +01:00
|
|
|
import (
|
2020-01-15 16:46:13 +01:00
|
|
|
"fmt"
|
|
|
|
|
2020-01-14 14:49:14 +01:00
|
|
|
"github.com/CaliDog/certstream-go"
|
2020-01-16 15:56:57 +01:00
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"gitlab.dcso.lolcat/LABS/styx/broker"
|
2020-01-14 16:31:57 +01:00
|
|
|
"gitlab.dcso.lolcat/LABS/styx/models"
|
|
|
|
"gitlab.dcso.lolcat/LABS/styx/utils"
|
2020-01-14 14:49:14 +01:00
|
|
|
)
|
|
|
|
|
2020-01-14 09:43:12 +01:00
|
|
|
func main() {
|
2020-01-14 14:49:14 +01:00
|
|
|
// The false flag specifies that we want heartbeat messages.
|
|
|
|
stream, errStream := certstream.CertStreamEventStream(false)
|
2020-01-15 16:46:13 +01:00
|
|
|
fmt.Println("Starting to get data from CertStream...")
|
2020-01-16 15:56:57 +01:00
|
|
|
Conn, err := broker.SetUpKafkaConnecter()
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2020-01-14 14:49:14 +01:00
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case jq := <-stream:
|
2020-01-14 16:31:57 +01:00
|
|
|
if data, err := utils.ExtractCertFromStream(jq); err == nil {
|
2020-01-16 15:56:57 +01:00
|
|
|
node, err := models.BuildNode("certstream", *data)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
fmt.Println(node)
|
|
|
|
|
|
|
|
err = models.SendEventToKafka(Conn, *node)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2020-01-14 16:31:57 +01:00
|
|
|
}
|
|
|
|
|
2020-01-14 14:49:14 +01:00
|
|
|
case err := <-errStream:
|
2020-01-16 15:56:57 +01:00
|
|
|
logrus.Error(err)
|
2020-01-14 14:49:14 +01:00
|
|
|
}
|
|
|
|
}
|
2020-01-14 09:43:12 +01:00
|
|
|
}
|