|
|
@@ -1,9 +1,8 @@
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
- "github.com/juju/ratelimit"
|
|
|
+ "github.com/gin-gonic/gin"
|
|
|
"io/ioutil"
|
|
|
- "net/http"
|
|
|
_ "net/http"
|
|
|
"os"
|
|
|
"rocommon/service"
|
|
|
@@ -15,9 +14,6 @@ import (
|
|
|
"runtime"
|
|
|
"strconv"
|
|
|
"syscall"
|
|
|
- "time"
|
|
|
-
|
|
|
- "github.com/gin-gonic/gin"
|
|
|
)
|
|
|
|
|
|
// /https://github.com/gin-gonic/gin
|
|
|
@@ -76,17 +72,17 @@ func main() {
|
|
|
//对外开放端口
|
|
|
go func() {
|
|
|
router := gin.Default()
|
|
|
- bucket := ratelimit.NewBucket(time.Second/10, 1)
|
|
|
-
|
|
|
- // 限流中间件
|
|
|
- router.Use(func(c *gin.Context) {
|
|
|
- if bucket.TakeAvailable(1) == 0 {
|
|
|
- c.String(http.StatusTooManyRequests, "当前玩家用户过多，请稍后再试")
|
|
|
- c.Abort()
|
|
|
- return
|
|
|
- }
|
|
|
- c.Next()
|
|
|
- })
|
|
|
+ //bucket := ratelimit.NewBucket(time.Second/10, 1)
|
|
|
+ //
|
|
|
+ //// 限流中间件
|
|
|
+ //router.Use(func(c *gin.Context) {
|
|
|
+ // if bucket.TakeAvailable(1) == 0 {
|
|
|
+ // c.String(http.StatusTooManyRequests, "当前玩家用户过多，请稍后再试")
|
|
|
+ // c.Abort()
|
|
|
+ // return
|
|
|
+ // }
|
|
|
+ // c.Next()
|
|
|
+ //})
|
|
|
router.Use(httpLogHandleFunc)
|
|
|
r1 := router.Group("/serverlist")
|
|
|
//获取服务器列表
|
