golang实现路由中间件middleware

registerHandlers.go

package router
import (
 "../ctrl"
 "../funcs"
 "github.com/julienschmidt/httprouter"
 "net/http"
)
type middleWareHandler struct {
 r *httprouter.Router
}
func NewMiddleWareHandler(r *httprouter.Router) http.Handler {
 m := middleWareHandler{}
 m.r = r
 return m
}
func (m middleWareHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 //检验合法性(session)
 ok := funcs.ValidateUser(r)
 if ok == false {
 funcs.Resp(w, -1, "请先登录", nil)
 return
 }
 m.r.ServeHTTP(w, r)
}
func RegisterHandlers() *httprouter.Router {
 router := httprouter.New()
 router.ServeFiles("/asset/*filepath", http.Dir("asset")) //启动静态资源服务
 router.POST("/user/login", ctrl.Login) //登陆
 router.POST("/user/register", ctrl.Register) //注册
 router.POST("/contact/addfriend", ctrl.AddFriend) //添加好友
 router.POST("/contact/loadfriend", ctrl.LoadFriend) //加载好友
 return router
}

main.go

package main
import (
 "./funcs"
 "./router"
 "html/template"
 "log"
 "net/http"
 "runtime"
)
var (
 AllConf funcs.Confs
 err error
 tpl *template.Template
 //SessionManager *funcs.Manager
)
//func init() {
// SessionManager = funcs.Manager{}
//}
//循环渲染所有模版、路由方法
func DisplayView() {
 tpl, err = template.ParseGlob("view/**/*")
 //错误则不必再继续
 if err != nil {
 log.Fatal(err.Error())
 }
 for _, v := range tpl.Templates() {
 tplName := v.Name()
 http.HandleFunc(tplName, func(writer http.ResponseWriter, request *http.Request) {
 tpl.ExecuteTemplate(writer, tplName, nil)
 })
 }
}
func main() {
 //初始化配置
 if AllConf, err = funcs.InitConf("config.json"); err != nil {
 log.Fatal(err)
 }
 //初始化核心数
 runtime.GOMAXPROCS(runtime.NumCPU())
 //初始化数据库
 //渲染模版
 //DisplayView()
 //注册路由
 r := router.RegisterHandlers()
 mh := router.NewMiddleWareHandler(r)
 err := http.ListenAndServe(":80", mh)
 if err != nil {
 log.Fatal(err)
 }
}

本文来自:51CTO博客

感谢作者:梁十八

查看原文:golang实现路由中间件middleware

• 编辑
• 预览