You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
go-zero/rest/httpx/responses.go

58 lines
1.2 KiB
Go

4 years ago
package httpx
import (
"encoding/json"
"net/http"
"sync"
4 years ago
"github.com/tal-tech/go-zero/core/logx"
4 years ago
)
var (
errorHandler = defaultErrorHandler
lock sync.RWMutex
)
func Error(w http.ResponseWriter, err error) {
lock.RLock()
code, body := errorHandler(err)
lock.RUnlock()
WriteJson(w, code, body)
}
func Ok(w http.ResponseWriter) {
w.WriteHeader(http.StatusOK)
}
4 years ago
func OkJson(w http.ResponseWriter, v interface{}) {
WriteJson(w, http.StatusOK, v)
}
func SetErrorHandler(handler func(error) (int, interface{})) {
lock.Lock()
defer lock.Unlock()
errorHandler = handler
}
4 years ago
func WriteJson(w http.ResponseWriter, code int, v interface{}) {
w.Header().Set(ContentType, ApplicationJson)
w.WriteHeader(code)
if bs, err := json.Marshal(v); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
} else if n, err := w.Write(bs); err != nil {
// http.ErrHandlerTimeout has been handled by http.TimeoutHandler,
// so it's ignored here.
if err != http.ErrHandlerTimeout {
logx.Errorf("write response failed, error: %s", err)
}
} else if n < len(bs) {
logx.Errorf("actual bytes: %d, written bytes: %d", len(bs), n)
}
}
func defaultErrorHandler(err error) (int, interface{}) {
return http.StatusBadRequest, err
}