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)
}
}