Merge pull request #62 from qkgo/feat/env-cors

cors setup

Author: zhy1

pkg/util/http-util.go

@@ -186,10 +186,6 @@ func GetRouter(
 	gin.SetMode(gin.ReleaseMode)
 	gin.DefaultWriter = ioutil.Discard
 	router := gin.Default()
-	corsDefault := cors.DefaultConfig()
-	corsDefault.AllowCredentials = true
-	corsDefault.AllowHeaders = []string{"Origin", "token", "Content-Length", "Content-Type", "session", "DNT", "content-type", "s", "timezone", "tz", "specify", "order"}
-	corsDefault.AllowAllOrigins = true
 	if os.Getenv("IGNORE_ROOT") == "" {
 		router.GET("/", DefaultView)
 	}
@@ -201,8 +197,14 @@ func GetRouter(
 	if os.Getenv("HTTP_DETAIL") == "" {
 		router.Use(RequestLogger())
 	}
+	if os.Getenv("NEED_CORS") != "" {
+		corsDefault := cors.DefaultConfig()
+		corsDefault.AllowCredentials = true
+		corsDefault.AllowHeaders = []string{"Origin", "token", "Content-Length", "Content-Type", "session", "DNT", "content-type", "s", "timezone", "tz", "specify", "order"}
+		corsDefault.AllowAllOrigins = true
+		router.Use(cors.New(corsDefault))
+	}
 	router.Use(GinToLogrus())
-	router.Use(cors.New(corsDefault))
 	c := gin.LoggerConfig{
 		Output:    ioutil.Discard,
 		SkipPaths: []string{"*"},