Fix query node address

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>

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
 }
 

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 {

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