2024-04-02 14:51:18 +00:00
|
|
|
package firehose
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2024-06-10 14:55:59 +00:00
|
|
|
"html"
|
2024-04-02 14:51:18 +00:00
|
|
|
"net/http"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
// WriteSuccessResponse writes success response for AWS Firehose request.
|
|
|
|
//
|
|
|
|
// See https://docs.aws.amazon.com/firehose/latest/dev/httpdeliveryrequestresponse.html#responseformat
|
|
|
|
func WriteSuccessResponse(w http.ResponseWriter, r *http.Request) {
|
|
|
|
requestID := r.Header.Get("X-Amz-Firehose-Request-Id")
|
|
|
|
if requestID == "" {
|
2024-06-10 14:55:59 +00:00
|
|
|
// This isn't an AWS firehose request - just return an empty response in this case.
|
2024-04-02 14:51:18 +00:00
|
|
|
w.WriteHeader(http.StatusOK)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2024-06-10 14:55:59 +00:00
|
|
|
requestID = html.EscapeString(requestID)
|
2024-04-02 14:51:18 +00:00
|
|
|
body := fmt.Sprintf(`{"requestId":%q,"timestamp":%d}`, requestID, time.Now().UnixMilli())
|
|
|
|
|
|
|
|
h := w.Header()
|
|
|
|
h.Set("Content-Type", "application/json")
|
|
|
|
h.Set("Content-Length", fmt.Sprintf("%d", len(body)))
|
|
|
|
w.Write([]byte(body))
|
|
|
|
}
|