update SendToFiretail to take header name for token as arg

Author: TheTeaCat

src/handler.go

@@ -33,5 +33,5 @@ func Handler(ctx context.Context, event events.CloudwatchLogsEvent) error {
 		log.Println(string(logBytes))
 	}
 
-	return SendToFiretail(firetailLogs, firetailApiUrl, firetailApiToken)
+	return SendToFiretail(firetailLogs, firetailApiUrl, firetailTokenHeaderName, firetailToken)
 }

src/send_to_firetail.go

@@ -9,7 +9,7 @@ import (
 	"github.com/pkg/errors"
 )
 
-func SendToFiretail(firetailLogs map[string]*FiretailLog, apiUrl, apiToken string) error {
+func SendToFiretail(firetailLogs map[string]*FiretailLog, apiUrl, tokenHeaderName, token string) error {
 	reqBytes := []byte{}
 	for _, firetailLog := range firetailLogs {
 		logBytes, err := json.Marshal(*firetailLog)
@@ -29,7 +29,7 @@ func SendToFiretail(firetailLogs map[string]*FiretailLog, apiUrl, apiToken strin
 		return err
 	}
 
-	req.Header.Set("x-ft-api-key", apiToken)
+	req.Header.Set(tokenHeaderName, token)
 
 	resp, err := http.DefaultClient.Do(req)
 	if err != nil {