iris/_examples/mvc/error-handler-preflight/main.go

102 lines
2.0 KiB
Go
Raw Normal View History

2020-08-18 22:14:11 +02:00
package main
import (
"fmt"
"time"
"github.com/kataras/iris/v12"
"github.com/kataras/iris/v12/mvc"
)
func main() {
app := iris.New()
app.RegisterView(iris.HTML("./views", ".html"))
m := mvc.New(app)
m.Handle(new(controller))
app.Listen(":8080")
}
type controller struct{}
// Generic response type for JSON results.
type response struct {
ID uint64 `json:"id,omitempty"`
Data interface{} `json:"data,omitempty"` // {data: result } on fetch actions.
Code int `json:"code,omitempty"`
Message string `json:"message,omitempty"`
Timestamp int64 `json:"timestamp,omitempty"`
}
func (r *response) Preflight(ctx iris.Context) error {
2021-01-06 05:13:30 +01:00
if r.ID > 0 {
r.Timestamp = time.Now().Unix()
2020-08-18 22:14:11 +02:00
}
if code := r.Code; code > 0 {
// You can call ctx.View or mvc.View{...}.Dispatch
2020-08-18 22:14:11 +02:00
// to render HTML on Code != 200
// but in order to not proceed with the response resulting
// as JSON you MUST return the iris.ErrStopExecution error.
// Example:
if code != 200 {
mvc.View{
/* calls the ctx.StatusCode */
Code: code,
/* use any r.Data as the template data
OR the whole "response" as its data. */
2021-01-06 05:13:30 +01:00
Data: r,
2020-08-18 22:14:11 +02:00
/* automatically pick the template per error (just for the sake of the example) */
Name: fmt.Sprintf("%d", code),
}.Dispatch(ctx)
return iris.ErrStopExecution
}
2021-01-06 05:13:30 +01:00
ctx.StatusCode(r.Code)
2020-08-18 22:14:11 +02:00
}
return nil
}
type user struct {
ID uint64 `json:"id"`
}
func (c *controller) GetBy(userid uint64) *response {
2020-08-18 22:14:11 +02:00
if userid != 1 {
return &response{
2020-08-18 22:14:11 +02:00
Code: iris.StatusNotFound,
Message: "User Not Found",
}
}
return &response{
2020-08-18 22:14:11 +02:00
ID: userid,
Data: user{ID: userid},
}
}
/*
You can use that `response` structure on non-mvc applications too, using handlers:
c := app.ConfigureContainer()
c.Get("/{id:uint64}", getUserByID)
func getUserByID(id uint64) response {
if userid != 1 {
return response{
Code: iris.StatusNotFound,
Message: "User Not Found",
}
}
return response{
ID: userid,
Data: user{ID: userid},
}
}
*/