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)

　　}无锡引产打胎哪家好 http://xmobile.wxbhnk120.com/

　　//初始化核心数

　　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

　　}

巴特西

golang实现路由中间件middleware

最新文章

热门文章