SCFF-37 added the custom Headers regarding the Metadata for sumo

Author: nmillalaf

main.go

@@ -31,9 +31,9 @@ var (
 	tickerTime, errT     = time.ParseDuration("60s") //Default
 	eventsBatchSize      = kingpin.Flag("log-events-batch-size", "Log Events Batch Size").OverrideDefaultFromEnvar("LOG_EVENTS_BATCH_SIZE").Int()
 	sumoPostMinimumDelay = kingpin.Flag("sumo-post-minimum-delay", "Sumo Post Minimum Delay").OverrideDefaultFromEnvar("SUMO_POST_MINIMUM_DELAY").Duration()
-	sumoCategory         = kingpin.Flag("sumo-category", "Sumo Category").OverrideDefaultFromEnvar("SUMO_CATEGORY").String()
-	sumoName             = kingpin.Flag("sumo-name", "Sumo Name").OverrideDefaultFromEnvar("SUMO_NAME").String()
-	sumoClient           = kingpin.Flag("sumo-client", "Sumo Client").OverrideDefaultFromEnvar("SUMO_CLIENT").String()
+	sumoCategory         = kingpin.Flag("sumo-category", "Sumo Category").Default("").OverrideDefaultFromEnvar("SUMO_CATEGORY").String()
+	sumoName             = kingpin.Flag("sumo-name", "Sumo Name").Default("").OverrideDefaultFromEnvar("SUMO_NAME").String()
+	sumoClient           = kingpin.Flag("sumo-client", "Sumo Client").Default("").OverrideDefaultFromEnvar("SUMO_CLIENT").String()
 )
 
 var (
@@ -81,7 +81,7 @@ func main() {
 
 	logging.Info.Println("Creating queue")
 	queue := eventQueue.NewQueue(make([]*events.Event, 100))
-	loggingClientSumo := sumoCFFirehose.NewSumoLogicAppender(*sumoEndpoint, 1000, &queue, *eventsBatchSize, *sumoPostMinimumDelay)
+	loggingClientSumo := sumoCFFirehose.NewSumoLogicAppender(*sumoEndpoint, 1000, &queue, *eventsBatchSize, *sumoPostMinimumDelay, *sumoCategory, *sumoName, *sumoClient)
 	go loggingClientSumo.Start() //multi
 
 	logging.Info.Println("Creating Events")

sumoCFFirehose/sumoLogicAppender.go

@@ -22,6 +22,9 @@ type SumoLogicAppender struct {
 	eventsBatchSize      int
 	sumoPostMinimumDelay time.Duration
 	timerBetweenPost     time.Time
+	sumoCategory         string
+	sumoName             string
+	sumoClient           string
 }
 
 type SumoBuffer struct {
@@ -30,14 +33,17 @@ type SumoBuffer struct {
 	timerIdlebuffer          time.Time
 }
 
-func NewSumoLogicAppender(urlValue string, connectionTimeoutValue int, nozzleQueue *eventQueue.Queue, eventsBatchSize int, sumoPostMinimumDelay time.Duration) *SumoLogicAppender {
+func NewSumoLogicAppender(urlValue string, connectionTimeoutValue int, nozzleQueue *eventQueue.Queue, eventsBatchSize int, sumoPostMinimumDelay time.Duration, sumoCategory string, sumoName string, sumoClient string) *SumoLogicAppender {
 	return &SumoLogicAppender{
 		url:                  urlValue,
 		connectionTimeout:    connectionTimeoutValue,
 		httpClient:           http.Client{Timeout: time.Duration(connectionTimeoutValue * int(time.Millisecond))},
 		nozzleQueue:          nozzleQueue,
 		eventsBatchSize:      eventsBatchSize,
 		sumoPostMinimumDelay: sumoPostMinimumDelay,
+		sumoCategory:         sumoCategory,
+		sumoName:             sumoName,
+		sumoClient:           sumoClient,
 	}
 }
 
@@ -115,7 +121,6 @@ func (s *SumoLogicAppender) AppendLogs(buffer *SumoBuffer) {
 }
 
 func (s *SumoLogicAppender) SendToSumo(logStringToSend string) {
-
 	var buf bytes.Buffer
 	g := gzip.NewWriter(&buf)
 	g.Write([]byte(logStringToSend))
@@ -131,8 +136,18 @@ func (s *SumoLogicAppender) SendToSumo(logStringToSend string) {
 		logging.Error.Printf("http.NewRequest() error: %v\n", err)
 		return
 	}
+
 	request.Header.Add("Content-Encoding", "gzip")
-	//request.SetBasicAuth("admin", "admin")
+
+	if s.sumoName != "" {
+		request.Header.Add("X-Sumo-Name", s.sumoName)
+	}
+	if s.sumoClient != "" {
+		request.Header.Add("X-Sumo-Client", s.sumoClient)
+	}
+	if s.sumoCategory != "" {
+		request.Header.Add("X-Sumo-Category", s.sumoCategory)
+	}
 	response, err := s.httpClient.Do(request)
 
 	if (err != nil) || (response.StatusCode != 200 && response.StatusCode != 302 && response.StatusCode < 500) {
@@ -147,7 +162,18 @@ func (s *SumoLogicAppender) SendToSumo(logStringToSend string) {
 			if err != nil {
 				logging.Error.Printf("http.NewRequest() error: %v\n", err)
 			}
+			request.Header.Add("X-Sumo-Name", "HTTPTestString")
 			request.Header.Add("Content-Encoding", "gzip")
+
+			if s.sumoName != "" {
+				request.Header.Add("X-Sumo-Name", s.sumoName)
+			}
+			if s.sumoClient != "" {
+				request.Header.Add("X-Sumo-Client", s.sumoClient)
+			}
+			if s.sumoCategory != "" {
+				request.Header.Add("X-Sumo-Category", s.sumoCategory)
+			}
 			response, errRetry = s.httpClient.Do(request)
 			if errRetry != nil {
 				logging.Error.Printf("http.Do() error: %v\n", errRetry)