Skip to content

Commit beb6de1

Browse files
author
urmi
committed
run make lint
Signed-off-by: urmi <[email protected]>
1 parent 53c16a8 commit beb6de1

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

milvus/kubedb_client_builder.go

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -71,19 +71,21 @@ func (o *KubeDBClientBuilder) GetMilvusClient() (*milvusclient.Client, error) {
7171
config.Password = string(pass)
7272
}
7373

74-
conn, err := grpc.Dial(config.Address,
75-
grpc.WithTransportCredentials(insecure.NewCredentials()),
76-
grpc.WithBlock(),
77-
grpc.WithTimeout(10*time.Second),
78-
)
74+
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
75+
defer cancel()
76+
77+
conn, err := grpc.NewClient(config.Address, grpc.WithTransportCredentials(insecure.NewCredentials()))
7978
if err != nil {
8079
klog.Warningf("gRPC dial failed: %v", err)
8180
return nil, err
8281
}
83-
conn.Close()
82+
defer conn.Close()
8483

85-
ctx, cancel := context.WithTimeout(o.ctx, 30*time.Second)
86-
defer cancel()
84+
if err != nil {
85+
klog.Warningf("gRPC dial failed: %v", err)
86+
return nil, err
87+
}
88+
conn.Close()
8789

8890
client, err := milvusclient.New(ctx, config)
8991
if err != nil {

0 commit comments

Comments
 (0)