Setting Get metadata method to return both an interface as well as an error to remove panic.

This commit is contained in:
Nick Gerakines 2014-07-03 15:17:24 -04:00
parent 5eb0e10a78
commit 108bfb4470
2 changed files with 8 additions and 13 deletions

View File

@ -38,14 +38,14 @@ func main() {
}))
authorized.POST("admin", func(c *gin.Context) {
user := c.Get("user").(string)
user, _ := c.Get("user")
// Parse JSON
var json struct {
Value string `json:"value" binding:"required"`
}
if c.EnsureBody(&json) {
DB[user] = json.Value
DB[user.(string)] = json.Value
c.JSON(200, gin.H{"status": "ok"})
}
})

17
gin.go
View File

@ -8,7 +8,6 @@ import (
"fmt"
"github.com/julienschmidt/httprouter"
"html/template"
"log"
"math"
"net/http"
"path"
@ -283,18 +282,14 @@ func (c *Context) Set(key string, item interface{}) {
// Returns the value for the given key.
// It panics if the value doesn't exist.
func (c *Context) Get(key string) interface{} {
var ok bool
var item interface{}
func (c *Context) Get(key string) (interface{}, error) {
if c.Keys != nil {
item, ok = c.Keys[key]
} else {
item, ok = nil, false
item, ok := c.Keys[key]
if ok {
return item, nil
}
}
if !ok || item == nil {
log.Panicf("Key %s doesn't exist", key)
}
return item
return nil, errors.New("Key does not exist.")
}
/************************************/