Merge pull request #229 from openinfradev/change_swagger_url

trivial. minor fix
openinfradev · Feb 8, 2024 · ff35f44 · ff35f44
2 parents 99f571a + c5cdeec
commit ff35f44
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net/http"
 	"strconv"
+	"strings"
 
 	"github.com/spf13/pflag"
 	"github.com/spf13/viper"
@@ -74,7 +75,13 @@ func init() {
 		log.Error(err)
 	}
 
-	swagger.SwaggerInfo.Host = viper.GetString("external-address")
+	address := viper.GetString("external-address")
+	arr := strings.Split(address, "//")
+	if len(arr) >= 2 {
+		address = arr[1]
+	}
+
+	swagger.SwaggerInfo.Host = address
 }
 
 // @title       tks-api service