2019-05-31 00:52:10 +02:00
|
|
|
package http
|
|
|
|
|
|
|
|
import (
|
2020-03-07 13:06:57 +02:00
|
|
|
"encoding/json"
|
|
|
|
"fmt"
|
|
|
|
"log"
|
2019-05-31 00:52:10 +02:00
|
|
|
"net/http"
|
|
|
|
|
2020-01-30 13:39:00 +02:00
|
|
|
"github.com/micro/go-micro/v2/client/selector"
|
|
|
|
"github.com/micro/go-micro/v2/registry"
|
2019-05-31 00:52:10 +02:00
|
|
|
)
|
|
|
|
|
2020-03-07 13:06:57 +02: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))
|
|
|
|
}
|
|
|
|
|
2019-05-31 00:52:10 +02:00
|
|
|
func NewRoundTripper(opts ...Option) http.RoundTripper {
|
|
|
|
options := Options{
|
|
|
|
Registry: registry.DefaultRegistry,
|
|
|
|
}
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
|
|
|
|
|
|
|
return &roundTripper{
|
|
|
|
rt: http.DefaultTransport,
|
|
|
|
st: selector.Random,
|
|
|
|
opts: options,
|
|
|
|
}
|
|
|
|
}
|