ls

main.go

@@ -3,12 +3,9 @@ package main
 import (
 	"flag"
 	"fmt"
-	"gitee.com/zr233/bsf"
-	"github.com/gin-gonic/gin"
-	"github.com/sirupsen/logrus"
-	ginSwagger "github.com/swaggo/gin-swagger"
-	"github.com/swaggo/gin-swagger/swaggerFiles"
 	"os"
+	"strconv"
+	"strings"
 	"video_course/controller"
 	"video_course/docs"
 	"video_course/global"
@@ -16,6 +13,12 @@ import (
 	"video_course/middleware"
 	"video_course/repository/postgres"
 	"video_course/repository/redis"
+
+	"gitee.com/zr233/bsf"
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+	ginSwagger "github.com/swaggo/gin-swagger"
+	"github.com/swaggo/gin-swagger/swaggerFiles"
 )
 
 var _VERSION_ = "unknown"
@@ -95,7 +98,17 @@ func main() {
 	if global.Project.Debug {
 		gin.SetMode(gin.DebugMode)
 		//设置swagger文档主机
-		docs.SwaggerInfo.Host = fmt.Sprintf("%s:%d", global.Project.Host, global.Port)
+		strSilce := strings.Split(global.Project.Host, ":")
+		host := strSilce[0]
+		port := global.Port
+		if len(strSilce) > 1 {
+			port_, err := strconv.Atoi(strSilce[1])
+			if err == nil {
+				port = port_
+			}
+		}
+
+		docs.SwaggerInfo.Host = fmt.Sprintf("%s:%d", host, port)
 		// 注册swagger文档
 		// 查看地址为 /v1/docs/index.html     /v1/SignIn/docs/doc.json
 		r.GET("/v1/docs/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))