69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
/*
|
|
* Copyright 2024 the KubeSphere Authors.
|
|
* Please refer to the LICENSE file in the root directory of the project.
|
|
* https://github.com/kubesphere/kubesphere/blob/master/LICENSE
|
|
*/
|
|
|
|
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/emicklei/go-restful/v3"
|
|
"k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/klog/v2"
|
|
)
|
|
|
|
// Avoid emitting errors that look like valid HTML. Quotes are okay.
|
|
var sanitizer = strings.NewReplacer(`&`, "&", `<`, "<", `>`, ">")
|
|
|
|
func HandleInternalError(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusInternalServerError, response, req, err)
|
|
}
|
|
|
|
// HandleBadRequest writes http.StatusBadRequest and log error
|
|
func HandleBadRequest(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusBadRequest, response, req, err)
|
|
}
|
|
|
|
func HandleNotFound(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusNotFound, response, req, err)
|
|
}
|
|
|
|
func HandleForbidden(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusForbidden, response, req, err)
|
|
}
|
|
|
|
func HandleUnauthorized(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusUnauthorized, response, req, err)
|
|
}
|
|
|
|
func HandleTooManyRequests(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusTooManyRequests, response, req, err)
|
|
}
|
|
|
|
func HandleConflict(response *restful.Response, req *restful.Request, err error) {
|
|
handle(http.StatusConflict, response, req, err)
|
|
}
|
|
|
|
func HandleError(response *restful.Response, req *restful.Request, err error) {
|
|
var statusCode int
|
|
switch t := err.(type) {
|
|
case errors.APIStatus:
|
|
statusCode = int(t.Status().Code)
|
|
case restful.ServiceError:
|
|
statusCode = t.Code
|
|
default:
|
|
statusCode = http.StatusInternalServerError
|
|
}
|
|
handle(statusCode, response, req, err)
|
|
}
|
|
|
|
func handle(statusCode int, response *restful.Response, req *restful.Request, err error) {
|
|
_, fn, line, _ := runtime.Caller(2)
|
|
klog.Errorf("%s:%d %v", fn, line, err)
|
|
http.Error(response, sanitizer.Replace(err.Error()), statusCode)
|
|
}
|