diff --git a/common/middleware/demo.go b/common/middleware/demo.go index f2cc9533..93623874 100644 --- a/common/middleware/demo.go +++ b/common/middleware/demo.go @@ -8,19 +8,21 @@ import ( func DemoEvn() gin.HandlerFunc { return func(c *gin.Context) { method := c.Request.Method - if method == "GET" || - method == "OPTIONS" || - c.Request.RequestURI == "/api/v1/login" || - c.Request.RequestURI == "/api/v1/logout" { - c.Next() - } else { - c.JSON(http.StatusOK, gin.H{ - "code": 500, - "msg": "谢谢您的参与,但为了大家更好的体验,所以本次提交就算了吧!\U0001F600\U0001F600\U0001F600", - }) - c.Abort() - return + if config.ApplicationConfig.Mode == "demo" { + if method == "GET" || + method == "OPTIONS" || + c.Request.RequestURI == "/api/v1/login" || + c.Request.RequestURI == "/api/v1/logout" { + c.Next() + } else { + c.JSON(http.StatusOK, gin.H{ + "code": 500, + "msg": "谢谢您的参与,但为了大家更好的体验,所以本次提交就算了吧!\U0001F600\U0001F600\U0001F600", + }) + c.Abort() + return + } } - + c.Next() } -} \ No newline at end of file +}