micro/util/http/http.go

70 lines
1.7 KiB
Go
Raw Normal View History

package http // import "go.unistack.org/micro/v3/util/http"
2019-05-31 01:52:10 +03:00
import (
"context"
"encoding/json"
"fmt"
"log"
2019-05-31 01:52:10 +03:00
"net/http"
"strings"
2019-05-31 01:52:10 +03:00
"go.unistack.org/micro/v3/metadata"
"go.unistack.org/micro/v3/selector/random"
2019-05-31 01:52:10 +03:00
)
// Write sets the status and body on a http ResponseWriter
func Write(w http.ResponseWriter, contentType string, status int, body string) {
w.Header().Set("Content-Length", fmt.Sprintf("%v", len(body)))
w.Header().Set("Content-Type", contentType)
w.WriteHeader(status)
fmt.Fprintf(w, `%v`, body)
}
// WriteBadRequestError sets a 400 status code
func WriteBadRequestError(w http.ResponseWriter, err error) {
rawBody, err := json.Marshal(map[string]string{
"error": err.Error(),
})
if err != nil {
WriteInternalServerError(w, err)
return
}
Write(w, "application/json", 400, string(rawBody))
}
// WriteInternalServerError sets a 500 status code
func WriteInternalServerError(w http.ResponseWriter, err error) {
rawBody, err := json.Marshal(map[string]string{
"error": err.Error(),
})
if err != nil {
log.Println(err)
return
}
Write(w, "application/json", 500, string(rawBody))
}
// NewRoundTripper creates new http RoundTripper
2019-05-31 01:52:10 +03:00
func NewRoundTripper(opts ...Option) http.RoundTripper {
options := Options{}
2019-05-31 01:52:10 +03:00
for _, o := range opts {
o(&options)
}
return &roundTripper{
rt: http.DefaultTransport,
st: random.NewSelector(),
2019-05-31 01:52:10 +03:00
opts: options,
}
}
// RequestToContext puts the `Authorization` header bearer token into context
// so calls to services will be authorized.
func RequestToContext(r *http.Request) context.Context {
md := metadata.New(len(r.Header))
for k, v := range r.Header {
md.Set(k, strings.Join(v, ","))
}
return metadata.NewIncomingContext(r.Context(), md)
}