Gorilla Simple Handler Filter 实现

c4pt0r · · 2438 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

今晚简单撸了一个,感觉蛮简洁实用,可以 share 出来

mux_filter.go

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package utils

import (
    "net/http"

    "github.com/gorilla/mux"
)

var router *mux.Router

type HttpFunc func(w http.ResponseWriter, r *http.Request) bool

type RouteFilter struct {
    filters []HttpFunc
    hdlr    http.HandlerFunc
}

func NewRouteFilter() *RouteFilter {
    return &RouteFilter{}
}

func (f *RouteFilter) AddFilter(h HttpFunc) *RouteFilter {
    f.filters = append(f.filters, h)
    return f
}

func (f *RouteFilter) Handler(h http.HandlerFunc) *RouteFilter {
    f.hdlr = h
    return f
}

func (f *RouteFilter) ServeHTTP(w http.ResponseWriter, req *http.Request) {
    for _, hdlr := range f.filters {
        if !hdlr(w, req) {
            return
        }
    }
    f.hdlr(w, req)
}

使用方法 (实现一个 Basic Auth 的 Filter):

utils.go

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
package common

import (
	"net/http"
	"strings"
)

type AuthFunc func(string, string) bool

func AuthFilterWrapper(authFunc AuthFunc) func(w http.ResponseWriter, r *http.Request) bool {
	return func(w http.ResponseWriter, r *http.Request) bool {
		_, ok := r.Header["Authorization"]
		if !ok {
			http.Error(w, "Need Authorization Header", http.StatusBadRequest)
			return false
		}

		auth := strings.SplitN(r.Header["Authorization"][0], " ", 2)

		if len(auth) != 2 {
			http.Error(w, "Bad Syntax", http.StatusBadRequest)
			return false
		}

		if !authFunc(auth[0], auth[1]) {
			http.Error(w, "Authorization Failed", http.StatusUnauthorized)
			return false
		}
		return true
	}
}

app.go

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
func UserAuth(method, payload string) bool {
    switch method {
    case "Basic":
        p, _ := base64.StdEncoding.DecodeString(payload)
        pair := strings.SplitN(string(p), ":", 2)
        if len(pair) != 2 {
            return false
        }
        return true
    case "Token":
        return true
    }
    return false
}

func InitHandlers(r *mux.Router) {
    s := r.PathPrefix(URL_PREFIX).Subrouter()

    s.Handle("/test", utils.NewRouteFilter().
        AddFilter(common.AuthFilterWrapper(UserAuth)).
        Handler(testHandler))
}

func testHandler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("OK"))
}

有疑问加站长微信联系(非本文作者)

本文来自:dongxu.h

感谢作者:c4pt0r

查看原文:Gorilla Simple Handler Filter 实现

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:701969077

2438 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传