2017-04-08 86 views
1
package controllers 

import (
    "encoding/json" 
    "errors" 
    "io" 
    "io/ioutil" 
    "reflect" 
) 

func GetTypeFromReq(c *App, ty interface{}) (interface{}, error) { 
    //get the type we are going to marshall into 
    item := reflect.ValueOf(ty) 

    //define and set the error that we will be returning to null 
    var retErr error 
    retErr = nil 

    //extract the body from the request and defer closing of the body 
    body, err := ioutil.ReadAll(io.LimitReader(c.Request.Body, 1048576)) 
    defer c.Request.Body.Close() 

    //handle errors and unmarshal our data 
    if err != nil { 
     retErr = errors.New("Failed to Read body: " + err.Error()) 
    } else if err = json.Unmarshal(body, &item); err != nil { 
     retErr = errors.New("Unmarshal Failed: " + err.Error()) 
    } 

    return item, retErr 
} 

我想一个类型和请求传递到一个函数,那么该函数内部和解组请求到一个变量,并将其返回。如何一般马歇尔请求到JSON在golang

我认为我的做法是错误的,因为当我尝试这样做:

inter, err := GetTypeFromReq(&c, models.User{}) 
if err != nil { 
    revel.ERROR.Println(err.Error()) 
} 
user := inter.(models.User) 

我得到的错误“接口转换:接口{}是reflect.Value,不models.User”

关于如何解决这个问题的任何提示?最后一行的

回答

1

下面是如何修改的功能,使其按预期工作:

func GetTypeFromReq(c *App, ty interface{}) (interface{}, error) { 
    // Allocate new value with same type as ty 
    v := reflect.New(reflect.TypeOf(ty)) 

    //define and set the error that we will be returning to null 
    var retErr error 
    retErr = nil 

    //extract the body from the request and defer closing of the body 
    body, err := ioutil.ReadAll(io.LimitReader(c.Request.Body, 1048576)) 
    defer c.Request.Body.Close() 

    //handle errors and unmarshal our data 
    if err != nil { 
    retErr = errors.New("Failed to Read body: " + err.Error()) 
    } else if err = json.Unmarshal(body, v.Interface()); err != nil { 
    retErr = errors.New("Unmarshal Failed: " + err.Error()) 
    } 

    // v holds a pointer, call Elem() to get the value. 
    return v.Elem().Interface(), retErr 
} 

注到Interface()的呼吁得到reflect.Value的电流值。

下面是避免反射并键入断言的方法:

func GetFromReq(c *App, item interface{}) error { 
    //extract the body from the request and defer closing of the body 
    body, err := ioutil.ReadAll(io.LimitReader(c.Request.Body, 1048576)) 
    defer c.Request.Body.Close() 

    //handle errors and unmarshal our data 
    if err != nil { 
    retErr = errors.New("Failed to Read body: " + err.Error()) 
    } else if err = json.Unmarshal(body, item); err != nil { 
    retErr = errors.New("Unmarshal Failed: " + err.Error()) 
    } 
    return retErr 
} 

使用方法如下:

var user models.User 
err := GetFromReq(&c, &user) 
if err != nil { 
    revel.ERROR.Println(err.Error()) 
} 

使用JSON decoder以简化代码:

func GetFromReq(c *App, item interface{}) error { 
    defer c.Request.Body.Close() 
    return json.NewDecoder(io.LimitReader(c.Request.Body, 1048576)).Deocode(item) 
} 

如果c.Request*http.Requestc.Responsehttp.ResponseWriter,然后写入功能:

func GetFromReq(c *App, item interface{}) error { 
    return json.NewDecoder(http.MaxBytesReaer(c.Response, c.Request.Body, 1048576)).Deocode(item) 
} 

没有必要关闭请求主体在网/ HTTP服务器。使用MaxBytesReader而不是io.LimitReader来防止客户端意外或恶意发送大量请求并浪费服务器资源。

+0

真的很棒的解释,非常感谢你 –

+1

完成,woops。我以为我做到了。 –

0

修改代码:变化user := inter.(models.User)user := inter.Interface().(models.User),试试看!

0

“接口转换:接口{}是reflect.Value,不models.User”

相当平直向前有关消息错误。你的itemreflect.Value它不是models.User

所以我觉得在你的代码,你可以改变itemto models.User

但我认为你是绑来创建将与所有类型的模型的工作,在这种情况下models.User{}功能。因为它使用interface

你的方法是昂贵的。你可以在incoming request直接转换是这样的:

func GetTypeFromReq(c *App, ty models.User) (models.User, error) { 
    //get the type we are going to marshall into 
    var item models.User 

    //define and set the error that we will be returning to nil 
    var retErr error // this var if the value not define then it is nil. Because error is interface 

    //extract the body from the request and defer closing of the body 
    body, err := ioutil.ReadAll(io.LimitReader(c.Request.Body, 1048576)) 
    defer c.Request.Body.Close() 

    //handle errors and unmarshal our data 
    if err != nil { 
     retErr = errors.New("Failed to Read body: " + err.Error()) 
    } else if err = json.Unmarshal(body, &item); err != nil { 
     retErr = errors.New("Unmarshal Failed: " + err.Error()) 
    } 

    return item, retErr 
} 

如果您body具有相同的结构模型,它会给你的价值,如果不是那么它是error

请注意,您需要使用interface时要小心。你可以在这个article看到一些指导。使用接口:

  • 当API的用户需要提供实现细节时。
  • 当API有多个实现时,他们需要在内部进行维护。
  • 时可以改变API的部分已经确定,并要求脱钩。

你的功能你models.User的值转换为interface,然后返回interface值。这就是为什么它很昂贵。