Initial commit

pkg/errors/code.go

@@ -0,0 +1,17 @@
+package errors
+
+type Code int
+
+const (
+	OK Code = iota
+	Canceled
+	Unknown
+	InvalidArgument
+	Internal // 5
+	Unavailable
+	AlreadyExists
+	NotFound
+	NotImplement
+	VerifyFailed
+	Conflict
+)

pkg/errors/code_string.go

@@ -0,0 +1,16 @@
+// Code generated by "stringer -type=Code"; DO NOT EDIT.
+
+package errors
+
+import "strconv"
+
+const _Code_name = "OKCanceledUnknownInvalidArgumentInternalUnavailableAlreadyExistsWTFNotFoundNotImplementVerifyFailed"
+
+var _Code_index = [...]uint8{0, 2, 10, 17, 32, 40, 51, 64, 67, 75, 87, 99}
+
+func (i Code) String() string {
+	if i < 0 || i >= Code(len(_Code_index)-1) {
+		return "Code(" + strconv.FormatInt(int64(i), 10) + ")"
+	}
+	return _Code_name[_Code_index[i]:_Code_index[i+1]]
+}

pkg/errors/code_test.go

@@ -0,0 +1,7 @@
+package errors
+
+import "testing"
+
+func TestCode_String(t *testing.T) {
+	t.Log(Code(1).String())
+}

pkg/errors/errors.go

@@ -0,0 +1,114 @@
+package errors
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net"
+	"net/http"
+	"reflect"
+
+	k8sError "k8s.io/apimachinery/pkg/api/errors"
+
+	"github.com/emicklei/go-restful"
+	"github.com/go-sql-driver/mysql"
+	"github.com/golang/glog"
+)
+
+type Error struct {
+	Code    Code   `json:"code"`
+	Message string `json:"message"`
+}
+
+var None = New(OK, "success")
+
+func (e *Error) Error() string {
+	return fmt.Sprintf("error: %v,message: %v", e.Code.String(), e.Message)
+}
+func (e *Error) HttpStatusCode() int {
+	switch e.Code {
+	case OK:
+		return http.StatusOK
+	case InvalidArgument:
+		return http.StatusBadRequest
+	case AlreadyExists:
+		return http.StatusConflict
+	case Unavailable:
+		return http.StatusServiceUnavailable
+	case NotImplement:
+		return http.StatusNotImplemented
+	case VerifyFailed:
+		return http.StatusBadRequest
+	case Conflict:
+		return http.StatusConflict
+	case Internal:
+		fallthrough
+	case Unknown:
+		fallthrough
+	default:
+		return http.StatusInternalServerError
+	}
+}
+
+func New(code Code, message string) error {
+	if message == "" {
+		message = code.String()
+	}
+	return &Error{Code: code, Message: message}
+}
+
+func HandlerError(err error, resp *restful.Response) bool {
+
+	if err == nil {
+		return false
+	}
+
+	glog.Errorln(reflect.TypeOf(err), err)
+
+	resp.WriteHeaderAndEntity(wrapper(err))
+
+	return true
+}
+
+func wrapper(err error) (int, interface{}) {
+	switch err.(type) {
+	case *Error:
+	case *json.UnmarshalTypeError:
+		err = New(InvalidArgument, err.Error())
+	case *mysql.MySQLError:
+		err = wrapperMysqlError(err.(*mysql.MySQLError))
+	case *net.OpError:
+		err = New(Internal, err.Error())
+	default:
+		if k8sError.IsNotFound(err) {
+			err = New(NotFound, err.Error())
+		} else {
+			err = New(Unknown, err.Error())
+		}
+
+	}
+	return err.(*Error).HttpStatusCode(), err
+}
+
+func wrapperMysqlError(sqlError *mysql.MySQLError) error {
+	switch sqlError.Number {
+	case 1062:
+		return New(AlreadyExists, sqlError.Message)
+	default:
+		return New(Unknown, sqlError.Message)
+	}
+}
+
+func Wrap(data []byte) error {
+	var j map[string]string
+	err := json.Unmarshal(data, &j)
+	if err != nil {
+		return errors.New(string(data))
+	} else if message := j["message"]; message != "" {
+		return errors.New(message)
+	} else if message := j["Error"]; message != "" {
+		return errors.New(message)
+	} else {
+		return errors.New(string(data))
+	}
+}