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

还可以更灵活： main.go

package main

import (
	"./funcs"
	"html/template"
	"log"
	"net/http"
	"runtime"
	"./ctrl"
)

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

	//初始化数据库

	http.Handle("/user/login", http.HandlerFunc(ctrl.Login)) //登陆 post
	http.Handle("/user/register", http.HandlerFunc(ctrl.Register)) //注册 post
	http.Handle("/contact/addfriend", funcs.AuthLogin(http.HandlerFunc(ctrl.AddFriend))) //添加好友 get
	http.Handle("/contact/loadfriend", funcs.AuthLogin(http.HandlerFunc(ctrl.LoadFriend))) //加载好友 get

	err := http.ListenAndServe(":80", nil)
	if err != nil {
		log.Fatal(err)
	}
}

middleWare.go

package funcs

import (
	"../defs"
	"../model/redis"
	"encoding/json"
	"net/http"
)

//验证是否登陆
func AuthLogin(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		//验证...

	})
}

ctrl/addFriend.go

package ctrl

import (
	"../funcs"
	"../model"
	"fmt"
	"net/http"
	"strconv"
)

var contactModel model.Contact

func AddFriend(writer http.ResponseWriter, request *http.Request) {

	fmt.Println("addFriend")
	return

	
}