老师如何给web请求添加一个授权中间件？

你在学习Golang相关的知识吗？本文《老师如何给web请求添加一个授权中间件？》，主要介绍的内容就涉及到golang、中间件，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

问题内容

老师如何给web请求添加一个授权中间件？

正确答案

中间件就是在请求中添加一个层，实现你想要的能够，验证成功进行一下步，失败返回。

package main

import (
	"io"
	"net/http"
)

const (
	// 授权key
	MyAPIKey = "MY_EXAMPLE_KEY"
)

func main() {
	// 处理请求
	http.Handle("/", Middleware(
		http.HandlerFunc(ExampleHandler),
		AuthMiddleware,
	))

	// 8080端口监听
	if err := http.ListenAndServe(":8080", nil); err != nil {
		panic(err)
	}
}

func Middleware(h http.Handler, middleware ...func(http.Handler) http.Handler) http.Handler {
	for _, mw := range middleware {
		h = mw(h)
	}
	return h
}

// 授权中间件
func AuthMiddleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// 获得key
		requestKey := r.URL.Query().Get("key")
		if len(requestKey) == 0 || requestKey != MyAPIKey {
			// 授权失败
			w.Header().Add("Content-Type", "application/json")
			w.WriteHeader(http.StatusUnauthorized)
			io.WriteString(w, `{"error":"invalid_key"}`)
			return
		}
		next.ServeHTTP(w, r)
	})
}

func ExampleHandler(w http.ResponseWriter, r *http.Request) {
	w.Header().Add("Content-Type", "application/json")
	io.WriteString(w, `{"status":"ok"}`)
}

好了，本文到此结束，带大家了解了《老师如何给web请求添加一个授权中间件？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！