加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 大数据 > 正文

golang实现路由中间件middleware

发布时间:2020-12-16 09:23:55 所属栏目:大数据 来源:网络整理
导读:registerHandlers.go package routerimport ( "../ctrl" "../funcs" "github.com/julienschmidt/httprouter" "net/http")type middleWareHandler struct { r *httprouter.Router}func NewMiddleWareHandler(r *httprouter.Router) http.Handler { m := middl
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)
    }
}

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读