polycule-connect/polyculeconnect/helpers/helpers.go
Melora Hugues 1e8b9ef161
All checks were successful
continuous-integration/drone/push Build is passing
Set correct login flow
2023-10-15 20:11:50 +02:00

31 lines
753 B
Go

package helpers
import (
"context"
"net/http"
"github.com/sirupsen/logrus"
)
type ContextKey string
const ResponseInfoKey ContextKey = "response_info"
type ResponseInfo struct {
ReturnCode int
ContentLength int
}
func HandleResponse(w http.ResponseWriter, r *http.Request, returncode int, content []byte, l *logrus.Logger) {
w.WriteHeader(returncode)
n, err := w.Write(content)
if err != nil {
l.Errorf("Failed to write content to response: %q", err.Error())
}
if n != len(content) {
l.Errorf("Failed to write the entire response (%d/%d)", n, len(content))
}
contextedReq := r.WithContext(context.WithValue(r.Context(), ResponseInfoKey, ResponseInfo{ReturnCode: returncode, ContentLength: len(content)}))
*r = *contextedReq
}