iris前置自定义读取请求数据数据处理

目录结构

主目录readCustomPerType

  1. —— main.go
  2. —— main_test.go

代码示例

main.go

  1. package main
  2. import (
  3. "gopkg.in/yaml.v2"
  4. "github.com/kataras/iris"
  5. )
  6. func main() {
  7. app := newApp()
  8. //使用Postman或其他什么来做POST请求
  9. //(但是你总是可以自由地使用app.Get和GET http方法请求来读取请求值)
  10. //使用RAW BODY到http//localhost:8080:
  11. /*
  12. addr: localhost:8080
  13. serverName: Iris
  14. */
  15. //响应应该是:
  16. //收到: main.config{Addr:"localhost:8080", ServerName:"Iris"}
  17. app.Run(iris.Addr(":8080"), iris.WithoutServerError(iris.ErrServerClosed), iris.WithOptimizations)
  18. }
  19. func newApp() *iris.Application {
  20. app := iris.New()
  21. app.Post("/", handler)
  22. return app
  23. }
  24. //简单的yaml内容,请访问https://github.com/go-yaml/yaml阅读更多内容
  25. type config struct {
  26. Addr string `yaml:"addr"`
  27. ServerName string `yaml:"serverName"`
  28. }
  29. //Decode实现`kataras/iris/context#BodyDecoder`可选接口
  30. //任何go类型都可以实现,以便在读取请求的主体时进行自解码。
  31. func (c *config) Decode(body []byte) error {
  32. return yaml.Unmarshal(body, c)
  33. }
  34. func handler(ctx iris.Context) {
  35. var c config
  36. // 注意:第二个参数是nil,因为我们的&c实现了`context#BodyDecoder`
  37. //优先于上下文#Unmarshaler(可以是读取请求主体全局选项)
  38. //参见`http_request/read-custom-via-unmarshaler/main.go`示例,了解如何使用上下文#Unmarshaler。
  39. // 注意2:如果你因任何原因需要多次读取访问body
  40. //你应该通过`app.Run(...,iris.WithoutBodyConsumptionOnUnmarshal)消费body
  41. if err := ctx.UnmarshalBody(&c, nil); err != nil {
  42. ctx.StatusCode(iris.StatusBadRequest)
  43. ctx.WriteString(err.Error())
  44. return
  45. }
  46. ctx.Writef("Received: %#+v", c)
  47. }

main_test.go

  1. package main
  2. import (
  3. "testing"
  4. "github.com/kataras/iris/httptest"
  5. )
  6. func TestReadCustomPerType(t *testing.T) {
  7. app := newApp()
  8. e := httptest.New(t, app)
  9. expectedResponse := `Received: main.config{Addr:"localhost:8080", ServerName:"Iris"}`
  10. e.POST("/").WithText("addr: localhost:8080\nserverName: Iris").Expect().
  11. Status(httptest.StatusOK).Body().Equal(expectedResponse)
  12. }

提示

  • 记得是RAW BODY格式
  • 这个例子是需要config类型实现Decode接口