Fixes errors

This commit is contained in:
Manu Mtz-Almeida 2015-05-12 15:22:13 +02:00
parent 421793bfba
commit 99694bb716
9 changed files with 42 additions and 42 deletions

View File

@ -40,13 +40,18 @@ type Params []Param
// ByName returns the value of the first Param which key matches the given name. // ByName returns the value of the first Param which key matches the given name.
// If no matching Param is found, an empty string is returned. // If no matching Param is found, an empty string is returned.
func (ps Params) ByName(name string) string { func (ps Params) Get(name string) (string, bool) {
for _, entry := range ps { for _, entry := range ps {
if entry.Key == name { if entry.Key == name {
return entry.Value return entry.Value, true
} }
} }
return "" return "", false
}
func (ps Params) ByName(name string) (va string) {
va, _ = ps.Get(name)
return
} }
// Context is the most important part of gin. It allows us to pass variables between middleware, // Context is the most important part of gin. It allows us to pass variables between middleware,
@ -138,9 +143,9 @@ func (c *Context) Fail(code int, err error) {
func (c *Context) ErrorTyped(err error, typ int, meta interface{}) { func (c *Context) ErrorTyped(err error, typ int, meta interface{}) {
c.Errors = append(c.Errors, errorMsg{ c.Errors = append(c.Errors, errorMsg{
Err: err.Error(), Error: err,
Type: typ, Flags: typ,
Meta: meta, Meta: meta,
}) })
} }
@ -154,7 +159,7 @@ func (c *Context) Error(err error, meta interface{}) {
func (c *Context) LastError() error { func (c *Context) LastError() error {
nuErrors := len(c.Errors) nuErrors := len(c.Errors)
if nuErrors > 0 { if nuErrors > 0 {
return errors.New(c.Errors[nuErrors-1].Err) return c.Errors[nuErrors-1].Error
} }
return nil return nil
} }
@ -203,8 +208,7 @@ func (c *Context) DefaultParamValue(key, defaultValue string) string {
} }
func (c *Context) paramValue(key string) (string, bool) { func (c *Context) paramValue(key string) (string, bool) {
va := c.Params.ByName(key) return c.Params.Get(key)
return va, len(va) > 0
} }
func (c *Context) formValue(key string) (string, bool) { func (c *Context) formValue(key string) (string, bool) {
@ -231,17 +235,17 @@ func (c *Context) postFormValue(key string) (string, bool) {
// Sets a new pair key/value just for the specified context. // Sets a new pair key/value just for the specified context.
// It also lazy initializes the hashmap. // It also lazy initializes the hashmap.
func (c *Context) Set(key string, item interface{}) { func (c *Context) Set(key string, value interface{}) {
if c.Keys == nil { if c.Keys == nil {
c.Keys = make(map[string]interface{}) c.Keys = make(map[string]interface{})
} }
c.Keys[key] = item c.Keys[key] = value
} }
// Get returns the value for the given key or an error if the key does not exist. // Get returns the value for the given key or an error if the key does not exist.
func (c *Context) Get(key string) (value interface{}, ok bool) { func (c *Context) Get(key string) (value interface{}, exists bool) {
if c.Keys != nil { if c.Keys != nil {
value, ok = c.Keys[key] value, exists = c.Keys[key]
} }
return return
} }

View File

@ -318,13 +318,13 @@ func TestContextError(t *testing.T) {
assert.Equal(t, c.Errors.String(), "Error #01: first error\n Meta: some data\n"+ assert.Equal(t, c.Errors.String(), "Error #01: first error\n Meta: some data\n"+
"Error #02: second error\n Meta: some data 2\n") "Error #02: second error\n Meta: some data 2\n")
assert.Equal(t, c.Errors[0].Err, "first error") assert.Equal(t, c.Errors[0].Error, errors.New("first error"))
assert.Equal(t, c.Errors[0].Meta, "some data") assert.Equal(t, c.Errors[0].Meta, "some data")
assert.Equal(t, c.Errors[0].Type, ErrorTypeExternal) assert.Equal(t, c.Errors[0].Flags, ErrorTypeExternal)
assert.Equal(t, c.Errors[1].Err, "second error") assert.Equal(t, c.Errors[1].Error, errors.New("second error"))
assert.Equal(t, c.Errors[1].Meta, "some data 2") assert.Equal(t, c.Errors[1].Meta, "some data 2")
assert.Equal(t, c.Errors[1].Type, ErrorTypeExternal) assert.Equal(t, c.Errors[1].Flags, ErrorTypeExternal)
} }
func TestContextTypedError(t *testing.T) { func TestContextTypedError(t *testing.T) {
@ -337,11 +337,11 @@ func TestContextTypedError(t *testing.T) {
c.ErrorTyped(errors.New("interno 2"), ErrorTypeInternal, nil) c.ErrorTyped(errors.New("interno 2"), ErrorTypeInternal, nil)
for _, err := range c.Errors.ByType(ErrorTypeExternal) { for _, err := range c.Errors.ByType(ErrorTypeExternal) {
assert.Equal(t, err.Type, ErrorTypeExternal) assert.Equal(t, err.Flags, ErrorTypeExternal)
} }
for _, err := range c.Errors.ByType(ErrorTypeInternal) { for _, err := range c.Errors.ByType(ErrorTypeInternal) {
assert.Equal(t, err.Type, ErrorTypeInternal) assert.Equal(t, err.Flags, ErrorTypeInternal)
} }
} }

View File

@ -10,18 +10,15 @@ import (
) )
const ( const (
ErrorTypePrivate = 1 << iota ErrorTypeInternal = 1 << iota
ErrorTypePublic = 1 << iota ErrorTypeExternal = 1 << iota
) ErrorTypeAny = 0xffffffff
const (
ErrorMaskAny = 0xffffffff
) )
// Used internally to collect errors that occurred during an http request. // Used internally to collect errors that occurred during an http request.
type errorMsg struct { type errorMsg struct {
Error error `json:"error"` Error error `json:"error"`
Type int `json:"-"` Flags int `json:"-"`
Meta interface{} `json:"meta"` Meta interface{} `json:"meta"`
} }
@ -33,7 +30,7 @@ func (a errorMsgs) ByType(typ int) errorMsgs {
} }
result := make(errorMsgs, 0, len(a)) result := make(errorMsgs, 0, len(a))
for _, msg := range a { for _, msg := range a {
if msg.Type&typ > 0 { if msg.Flags&typ > 0 {
result = append(result, msg) result = append(result, msg)
} }
} }
@ -44,11 +41,11 @@ func (a errorMsgs) Errors() []string {
if len(a) == 0 { if len(a) == 0 {
return []string{} return []string{}
} }
errors := make([]string, len(a)) errorStrings := make([]string, len(a))
for i, err := range a { for i, err := range a {
errors[i] = err.Error.Error() errorStrings[i] = err.Error.Error()
} }
return errors return errorStrings
} }
func (a errorMsgs) String() string { func (a errorMsgs) String() string {

View File

@ -22,7 +22,7 @@ var (
) )
func ErrorLogger() HandlerFunc { func ErrorLogger() HandlerFunc {
return ErrorLoggerT(ErrorTypeAll) return ErrorLoggerT(ErrorTypeAny)
} }
func ErrorLoggerT(typ int) HandlerFunc { func ErrorLoggerT(typ int) HandlerFunc {
@ -31,17 +31,17 @@ func ErrorLoggerT(typ int) HandlerFunc {
if !c.Writer.Written() { if !c.Writer.Written() {
if errs := c.Errors.ByType(typ); len(errs) > 0 { if errs := c.Errors.ByType(typ); len(errs) > 0 {
c.JSON(-1, errs) c.JSON(-1, errs.Errors())
} }
} }
} }
} }
func Logger() HandlerFunc { func Logger() HandlerFunc {
return LoggerWithFile(DefaultWriter) return LoggerWithWriter(DefaultWriter)
} }
func LoggerWithFile(out io.Writer) HandlerFunc { func LoggerWithWriter(out io.Writer) HandlerFunc {
return func(c *Context) { return func(c *Context) {
// Start timer // Start timer
start := time.Now() start := time.Now()

View File

@ -24,7 +24,7 @@ func init() {
func TestLogger(t *testing.T) { func TestLogger(t *testing.T) {
buffer := new(bytes.Buffer) buffer := new(bytes.Buffer)
router := New() router := New()
router.Use(LoggerWithFile(buffer)) router.Use(LoggerWithWriter(buffer))
router.GET("/example", func(c *Context) {}) router.GET("/example", func(c *Context) {})
performRequest(router, "GET", "/example") performRequest(router, "GET", "/example")

View File

@ -23,10 +23,10 @@ var (
// Recovery returns a middleware that recovers from any panics and writes a 500 if there was one. // Recovery returns a middleware that recovers from any panics and writes a 500 if there was one.
// While Gin is in development mode, Recovery will also output the panic as HTML. // While Gin is in development mode, Recovery will also output the panic as HTML.
func Recovery() HandlerFunc { func Recovery() HandlerFunc {
return RecoveryWithFile(DefaultWriter) return RecoveryWithWriter(DefaultWriter)
} }
func RecoveryWithFile(out io.Writer) HandlerFunc { func RecoveryWithWriter(out io.Writer) HandlerFunc {
var logger *log.Logger var logger *log.Logger
if out != nil { if out != nil {
logger = log.New(out, "", log.LstdFlags) logger = log.New(out, "", log.LstdFlags)

View File

@ -15,7 +15,7 @@ import (
func TestPanicInHandler(t *testing.T) { func TestPanicInHandler(t *testing.T) {
buffer := new(bytes.Buffer) buffer := new(bytes.Buffer)
router := New() router := New()
router.Use(RecoveryWithFile(buffer)) router.Use(RecoveryWithWriter(buffer))
router.GET("/recovery", func(_ *Context) { router.GET("/recovery", func(_ *Context) {
panic("Oupps, Houston, we have a problem") panic("Oupps, Houston, we have a problem")
}) })
@ -30,7 +30,7 @@ func TestPanicInHandler(t *testing.T) {
// TestPanicWithAbort assert that panic has been recovered even if context.Abort was used. // TestPanicWithAbort assert that panic has been recovered even if context.Abort was used.
func TestPanicWithAbort(t *testing.T) { func TestPanicWithAbort(t *testing.T) {
router := New() router := New()
router.Use(RecoveryWithFile(nil)) router.Use(RecoveryWithWriter(nil))
router.GET("/recovery", func(c *Context) { router.GET("/recovery", func(c *Context) {
c.AbortWithStatus(400) c.AbortWithStatus(400)
panic("Oupps, Houston, we have a problem") panic("Oupps, Houston, we have a problem")

View File

@ -6,7 +6,6 @@ package gin
import ( import (
"bufio" "bufio"
"log"
"net" "net"
"net/http" "net/http"
) )
@ -46,7 +45,7 @@ func (w *responseWriter) WriteHeader(code int) {
if code > 0 { if code > 0 {
w.status = code w.status = code
if w.Written() { if w.Written() {
log.Println("[GIN] WARNING. Headers were already written!") debugPrint("[WARNING] Headers were already written")
} }
} }
} }

View File

@ -50,7 +50,7 @@ func TestFunctionName(t *testing.T) {
} }
func somefunction() { func somefunction() {
// this empty function is used by TestFunctionName()
} }
func TestJoinPaths(t *testing.T) { func TestJoinPaths(t *testing.T) {