tango/error.go
Lunny Xiao 2f5221bc91
Add query functions and upgrade circleci to 2 (#51)
* add query functions and upgrade circleci to 2

* add go mod
2019-01-25 17:50:54 +08:00

94 lines
2.2 KiB
Go

// Copyright 2015 The Tango Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package tango
import (
"fmt"
"net/http"
)
// AbortError defines an interface to describe HTTP error
type AbortError interface {
error
Code() int
}
type abortError struct {
code int
content string
}
func (a *abortError) Code() int {
return a.code
}
func (a *abortError) Error() string {
return fmt.Sprintf("%v", a.content)
}
// Abort returns an AbortError
func Abort(code int, content ...string) AbortError {
if len(content) >= 1 {
return &abortError{code, content[0]}
}
return &abortError{code, http.StatusText(code)}
}
// NotFound returns not found HTTP error
func NotFound(content ...string) AbortError {
return Abort(http.StatusNotFound, content...)
}
// NotSupported returns not supported HTTP error
func NotSupported(content ...string) AbortError {
return Abort(http.StatusMethodNotAllowed, content...)
}
// InternalServerError returns internal server HTTP error
func InternalServerError(content ...string) AbortError {
return Abort(http.StatusInternalServerError, content...)
}
// Forbidden returns forbidden HTTP error
func Forbidden(content ...string) AbortError {
return Abort(http.StatusForbidden, content...)
}
// Unauthorized returns unauthorized HTTP error
func Unauthorized(content ...string) AbortError {
return Abort(http.StatusUnauthorized, content...)
}
// Errors returns default errorhandler, you can use your self handler
func Errors() HandlerFunc {
return func(ctx *Context) {
switch res := ctx.Result.(type) {
case AbortError:
ctx.WriteHeader(res.Code())
ctx.WriteString(res.Error())
case error:
ctx.WriteHeader(http.StatusInternalServerError)
ctx.WriteString(res.Error())
default:
ctx.WriteHeader(http.StatusInternalServerError)
ctx.WriteString(http.StatusText(http.StatusInternalServerError))
}
}
}
// ErrorWithCode descripts an error that with error code
type ErrorWithCode interface {
error
ErrorCode() int
}
type ErrorKeyIsNotExist struct {
Key string
}
func (e ErrorKeyIsNotExist) Error() string {
return fmt.Sprintf("Key %s is not exist", e.Key)
}