From e242a1cc91fe4b3afea1dc88f4a42b1817f5b5b2 Mon Sep 17 00:00:00 2001
From: "peng.xu" <peng.xu@zilliz.com>
Date: Wed, 18 Sep 2019 20:16:46 +0800
Subject: [PATCH] temp support dns addr

---
 mishards/connections.py     | 1 +
 mishards/main.py            | 8 +++++++-
 mishards/service_handler.py | 2 +-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/mishards/connections.py b/mishards/connections.py
index 9201ea2b08..c6323f66f8 100644
--- a/mishards/connections.py
+++ b/mishards/connections.py
@@ -1,5 +1,6 @@
 import logging
 import threading
+import socket
 from functools import wraps
 from contextlib import contextmanager
 from milvus import Milvus
diff --git a/mishards/main.py b/mishards/main.py
index 5d96d8b499..e9c47f9edf 100644
--- a/mishards/main.py
+++ b/mishards/main.py
@@ -1,6 +1,9 @@
 import os, sys
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
+from urllib.parse import urlparse
+import socket
+
 from mishards import (
         settings,
         db, connect_mgr,
@@ -9,7 +12,10 @@ from mishards import (
 
 def main():
     discover.start()
-    connect_mgr.register('WOSERVER', settings.WOSERVER if not settings.TESTING else settings.TESTING_WOSERVER)
+    woserver = settings.WOSERVER if not settings.TESTING else settings.TESTING_WOSERVER
+    url = urlparse(woserver)
+    connect_mgr.register('WOSERVER',
+            '{}://{}:{}'.format(url.scheme, socket.gethostbyname(url.hostname), url.port))
     server.run(port=settings.SERVER_PORT)
     return 0
 
diff --git a/mishards/service_handler.py b/mishards/service_handler.py
index eb2951be5e..ac70440c47 100644
--- a/mishards/service_handler.py
+++ b/mishards/service_handler.py
@@ -145,7 +145,7 @@ class ServiceHandler(milvus_pb2_grpc.MilvusServiceServicer):
             for res in rs:
                 res.result()
 
-        reverse = table_meta.metric_type == types.MetricType.L2
+        reverse = table_meta.metric_type == types.MetricType.IP
         return self._do_merge(all_topk_results, topk, reverse=reverse)
 
     def CreateTable(self, request, context):