From 837338a632734c9765d8607dd2585e72debc5c31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=A7=89?= Date: Mon, 13 Nov 2023 19:40:14 +0800 Subject: [PATCH] Refactor Milvus connection initialization --- models/milvus.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/milvus.go b/models/milvus.go index c7ed90d..f94c985 100644 --- a/models/milvus.go +++ b/models/milvus.go @@ -2,8 +2,8 @@ package models import ( "context" + "fmt" "log" - "strconv" "github.com/milvus-io/milvus-sdk-go/v2/client" ) @@ -20,9 +20,9 @@ func (m *MilvusConnection) Init() (err error) { log.Println("Milvus connection init") host := Viper.Get("milvus.host").(string) port := Viper.Get("milvus.port").(int) - address_of_milvus := host + ":" + strconv.Itoa(port) - log.Println("Milvus address:", address_of_milvus) - m.Client, err = client.NewGrpcClient(context.Background(), address_of_milvus) + address := fmt.Sprintf("%s:%d", host, port) + log.Println("Milvus address:", address) + m.Client, err = client.NewGrpcClient(context.Background(), address) if err != nil { log.Println("Milvus connection failed:", err) return