diff --git a/internal/proxynode/insert_channels.go b/internal/proxynode/insert_channels.go index 90451614bdfd208157da1fe81409fc869d943b0c..deda7ef03c957858ebf3a92ad92c1d894d22aa3c 100644 --- a/internal/proxynode/insert_channels.go +++ b/internal/proxynode/insert_channels.go @@ -134,10 +134,12 @@ func (m *InsertChannelsMap) closeInsertMsgStream(collID UniqueID) error { m.usageHistogram[loc]-- if m.usageHistogram[loc] <= 0 { m.insertMsgStreams[loc].Close() - m.droppedBitMap[loc] = 1 - delete(m.collectionID2InsertChannels, collID) - log.Print("close insert message stream ...") } + log.Print("close insert message stream ...") + + m.droppedBitMap[loc] = 1 + delete(m.collectionID2InsertChannels, collID) + return nil } diff --git a/internal/querynode/param_table.go b/internal/querynode/param_table.go index a2254f1bdb710f62e83878171aca97c13860e050..d34a04e6ea3005a96cd7f9dcb034c7c4eada1701 100644 --- a/internal/querynode/param_table.go +++ b/internal/querynode/param_table.go @@ -4,8 +4,8 @@ import ( "log" "os" "strconv" - "strings" + "github.com/zilliztech/milvus-distributed/internal/util/funcutil" "github.com/zilliztech/milvus-distributed/internal/util/paramtable" ) @@ -87,22 +87,13 @@ func (p *ParamTable) Init() { } } - queryNodeAddress := os.Getenv("QUERY_NODE_ADDRESS") - if queryNodeAddress == "" { + queryNodeIP := os.Getenv("QUERY_NODE_IP") + if queryNodeIP == "" { p.QueryNodeIP = "localhost" - p.QueryNodePort = 20010 } else { - ipAndPort := strings.Split(queryNodeAddress, ":") - if len(ipAndPort) != 2 { - panic("illegal query node address") - } - p.QueryNodeIP = ipAndPort[0] - port, err := strconv.ParseInt(ipAndPort[1], 10, 64) - if err != nil { - panic(err) - } - p.QueryNodePort = port + p.QueryNodeIP = queryNodeIP } + p.QueryNodePort = int64(funcutil.GetAvailablePort()) err = p.LoadYaml("advanced/common.yaml") if err != nil { diff --git a/tests/python/requirements.txt b/tests/python/requirements.txt index a4136e23e77e33a092b21452c4cfad575d97d6b7..fcf08cf5d07356c9e2a31351229ead6dd82400a3 100644 --- a/tests/python/requirements.txt +++ b/tests/python/requirements.txt @@ -2,7 +2,7 @@ grpcio==1.26.0 grpcio-tools==1.26.0 numpy==1.18.1 pytest-cov==2.8.1 -pymilvus-distributed==0.0.17 +pymilvus-distributed==0.0.19 sklearn==0.0 pytest==4.5.0 pytest-timeout==1.3.3