golang 库 go-restful 中 https basic authentication 基础认证实例

实例如下:

package main

import (
	"github.com/emicklei/go-restful"
	"log"
	"net/http"
	"os"
)

func main() {
	wsContainer := restful.NewContainer()
	wsContainer.Router(restful.CurlyRouter{})

	ws := new(restful.WebService)
	ws.
		Path("/api").
		Consumes(restful.MIME_JSON).
		Produces(restful.MIME_JSON) // 可单独设置每一个方法

	//	最终地址 https://127.0.0.1:7443/api/config
	ws.Route(ws.POST("/config").Filter(basicAuthenticate).To(config))

	ws.Filter(basicAuthenticate)//basic auth filter

	wsContainer.Add(ws)
	log.Printf("start listening on localhost:7443")
	server := &http.Server{Addr: ":7443", Handler: wsContainer}

	//证书生成方式参考https://blog.csdn.net/u011411069/article/details/79994716
	crtPath := "/home/leen/Desktop/certificate.crt"//crt 证书
	_, err := os.Stat(crtPath)
	if err != nil {
		log.Println("crt file not exist!")
		return
	}

	keyPath := "/home/leen/Desktop/certificate.key"//key 证书
	_, err = os.Stat(crtPath)
	if err != nil {
		log.Println("key file not exist!")
		return
	}

	err = server.ListenAndServeTLS(crtPath, keyPath)
	if err != nil {
		log.Println(err.Error())
	}
}

//basic auth 验证过滤
func basicAuthenticate(req *restful.Request, resp *restful.Response, chain *restful.FilterChain) {
	// usr/pwd = admin/admin
	u, p, ok := req.Request.BasicAuth()
	if !ok || u != "admin" || p != "admin" {
		resp.AddHeader("WWW-Authenticate", "Basic realm=Protected Area")
		resp.WriteErrorString(401, "401: Not Authorized")
		return
	}
	chain.ProcessFilter(req, resp)
}

//rest请求处理方法
func config(request *restful.Request, response *restful.Response) {
	entity := new(model.TestEntity)
	err := request.ReadEntity(entity)

	if err != nil {
		response.WriteEntity(model.NewResult(500, err.Error()))
	} else {
		response.WriteEntity(model.NewResult(0, "config successful!"))
	}
}

猜你喜欢

转载自blog.csdn.net/whatday/article/details/113750952