mirror of
https://gitee.com/johng/gf
synced 2026-07-06 21:45:34 +08:00
43 lines
784 B
Go
43 lines
784 B
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gogf/gf/v2/os/glog"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
)
|
|
|
|
func MiddlewareAuth(r *ghttp.Request) {
|
|
token := r.Get("token")
|
|
if token == "123456" {
|
|
r.Middleware.Next()
|
|
} else {
|
|
r.Response.WriteStatus(http.StatusForbidden)
|
|
}
|
|
}
|
|
|
|
func MiddlewareCORS(r *ghttp.Request) {
|
|
r.Response.CORSDefault()
|
|
r.Middleware.Next()
|
|
}
|
|
|
|
func MiddlewareLog(r *ghttp.Request) {
|
|
r.Middleware.Next()
|
|
g.Log().Print(r.Response.Status, r.URL.Path)
|
|
}
|
|
|
|
func main() {
|
|
s := g.Server()
|
|
s.Use(MiddlewareLog)
|
|
s.Group("/api.v2", func(group *ghttp.RouterGroup) {
|
|
group.Middleware(MiddlewareAuth, MiddlewareCORS)
|
|
group.ALL("/user/list", func(r *ghttp.Request) {
|
|
panic("custom error")
|
|
})
|
|
})
|
|
s.SetPort(8199)
|
|
s.Run()
|
|
}
|