swagger修改，限流令牌桶配置化
This commit is contained in:
33
http/http.go
33
http/http.go
@@ -6,6 +6,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
_ "gitee.com/red-future---jilin-g/common/consul"
|
_ "gitee.com/red-future---jilin-g/common/consul"
|
||||||
"gitee.com/red-future---jilin-g/common/jaeger"
|
"gitee.com/red-future---jilin-g/common/jaeger"
|
||||||
|
"gitee.com/red-future---jilin-g/common/middleware"
|
||||||
"gitee.com/red-future---jilin-g/common/utils"
|
"gitee.com/red-future---jilin-g/common/utils"
|
||||||
"github.com/gogf/gf/contrib/registry/consul/v2"
|
"github.com/gogf/gf/contrib/registry/consul/v2"
|
||||||
"github.com/gogf/gf/v2/frame/g"
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
@@ -38,9 +39,33 @@ func init() {
|
|||||||
//s.Use(common.Cors) //中间件验证
|
//s.Use(common.Cors) //中间件验证
|
||||||
//s.EnablePProf() //启用性能分析
|
//s.EnablePProf() //启用性能分析
|
||||||
Httpserver.SetOpenApiPath("/api.json")
|
Httpserver.SetOpenApiPath("/api.json")
|
||||||
Httpserver.SetSwaggerPath("/swagger") //api文档访问路径
|
Httpserver.SetSwaggerPath("/docs") //api文档访问路径
|
||||||
Httpserver.SetDumpRouterMap(true) //关闭打印路由注册信息
|
Httpserver.SetDumpRouterMap(true) //关闭打印路由注册信息
|
||||||
Httpserver.BindMiddlewareDefault(ghttp.MiddlewareHandlerResponse, jaeger.NewTracer) //使用默认http返回结构
|
Httpserver.SetSwaggerUITemplate(`
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8" />
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1" />
|
||||||
|
<meta name="description" content="SwaggerUI"/>
|
||||||
|
<title>SwaggerUI</title>
|
||||||
|
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/5.10.5/swagger-ui.min.css" />
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div id="swagger-ui"></div>
|
||||||
|
<script src="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/5.10.5/swagger-ui-bundle.js" crossorigin></script>
|
||||||
|
<script>
|
||||||
|
window.onload = () => {
|
||||||
|
window.ui = SwaggerUIBundle({
|
||||||
|
url: '{SwaggerUIDocUrl}',
|
||||||
|
dom_id: '#swagger-ui',
|
||||||
|
});
|
||||||
|
};
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
`)
|
||||||
|
Httpserver.BindMiddlewareDefault(ghttp.MiddlewareCORS, ghttp.MiddlewareHandlerResponse, middleware.Limiter, jaeger.NewTracer) //使用默认http返回结构
|
||||||
go Httpserver.Run()
|
go Httpserver.Run()
|
||||||
|
|
||||||
consulCfg, _ := g.Cfg().Get(context.Background(), "consul.address")
|
consulCfg, _ := g.Cfg().Get(context.Background(), "consul.address")
|
||||||
@@ -51,7 +76,6 @@ func init() {
|
|||||||
}
|
}
|
||||||
gsvc.SetRegistry(registry)
|
gsvc.SetRegistry(registry)
|
||||||
gsel.SetBuilder(gsel.NewBuilderRoundRobin())
|
gsel.SetBuilder(gsel.NewBuilderRoundRobin())
|
||||||
Httpclient.SetHeader("Authorization", g.RequestFromCtx(context.TODO()).GetHeader("Authorization"))
|
|
||||||
Httpclient.SetDiscovery(gsvc.GetRegistry())
|
Httpclient.SetDiscovery(gsvc.GetRegistry())
|
||||||
}
|
}
|
||||||
func RouteRegister(controllers []interface{}) {
|
func RouteRegister(controllers []interface{}) {
|
||||||
@@ -71,6 +95,7 @@ func doRequest(ctx context.Context, method string, url string, target any, data
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
Httpclient.SetHeader("Authorization", g.RequestFromCtx(ctx).GetHeader("Authorization"))
|
||||||
response, err := Httpclient.DoRequest(ctx, method, url, data)
|
response, err := Httpclient.DoRequest(ctx, method, url, data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
|
|||||||
55
middleware/middleware.go
Normal file
55
middleware/middleware.go
Normal file
@@ -0,0 +1,55 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/net/ghttp"
|
||||||
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
|
"github.com/gogf/gf/v2/text/gstr"
|
||||||
|
"golang.org/x/time/rate"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Logger 中间件
|
||||||
|
func Logger(r *ghttp.Request) {
|
||||||
|
startTime := gtime.TimestampMilli()
|
||||||
|
r.Middleware.Next()
|
||||||
|
endTime := gtime.TimestampMilli()
|
||||||
|
g.Log().Infof(r.GetCtx(),
|
||||||
|
"request: %s %s | status: %d | time: %dms",
|
||||||
|
r.Method,
|
||||||
|
r.URL.Path,
|
||||||
|
r.Response.Status,
|
||||||
|
endTime-startTime,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
var rateLimit, _ = g.Cfg().Get(context.TODO(), "rate.limit")
|
||||||
|
var rateBurst, _ = g.Cfg().Get(context.TODO(), "rate.burst")
|
||||||
|
var limiter = rate.NewLimiter(rate.Limit(rateLimit.Int()), rateBurst.Int())
|
||||||
|
|
||||||
|
func Limiter(r *ghttp.Request) {
|
||||||
|
if !limiter.Allow() {
|
||||||
|
r.Response.WriteStatusExit(429) // Return 429 Too Many Requests
|
||||||
|
r.ExitAll()
|
||||||
|
}
|
||||||
|
r.Middleware.Next()
|
||||||
|
}
|
||||||
|
func Auth(r *ghttp.Request) {
|
||||||
|
token := r.Header.Get("Authorization")
|
||||||
|
if token == "" || !gstr.HasPrefix(token, "Bearer ") {
|
||||||
|
r.Response.WriteStatusExit(401, "Unauthorized")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证 token
|
||||||
|
if !validateToken(gstr.SubStrFrom(token, "7")) {
|
||||||
|
r.Response.WriteStatusExit(401, "Unauthorized")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
r.Middleware.Next()
|
||||||
|
}
|
||||||
|
func validateToken(token string) bool {
|
||||||
|
// 实现 token 验证逻辑
|
||||||
|
return token == "valid-token"
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user