mirror of https://github.com/milvus-io/milvus.git
update for better test
parent
43bc2cc60c
commit
3ddd181dd2
|
@ -20,7 +20,7 @@ def app(request):
|
|||
@pytest.fixture
|
||||
def started_app(app):
|
||||
app.on_pre_run()
|
||||
app.start(app.port)
|
||||
app.start(settings.SERVER_TEST_PORT)
|
||||
|
||||
yield app
|
||||
|
||||
|
|
|
@ -91,7 +91,7 @@ class Server:
|
|||
handler_class(tracer=self.tracer,
|
||||
router=self.router), self.server_impl)
|
||||
self.server_impl.add_insecure_port("[::]:{}".format(
|
||||
str(port or self._port)))
|
||||
str(port or self.port)))
|
||||
self.server_impl.start()
|
||||
|
||||
def run(self, port):
|
||||
|
|
|
@ -25,6 +25,7 @@ TIMEOUT = env.int('TIMEOUT', 60)
|
|||
MAX_RETRY = env.int('MAX_RETRY', 3)
|
||||
|
||||
SERVER_PORT = env.int('SERVER_PORT', 19530)
|
||||
SERVER_TEST_PORT = env.int('SERVER_TEST_PORT', 19530)
|
||||
WOSERVER = env.str('WOSERVER')
|
||||
|
||||
SD_PROVIDER_SETTINGS = None
|
||||
|
@ -36,11 +37,13 @@ if SD_PROVIDER == 'Kubernetes':
|
|||
in_cluster=env.bool('SD_IN_CLUSTER', False),
|
||||
poll_interval=env.int('SD_POLL_INTERVAL', 5),
|
||||
pod_patt=env.str('SD_ROSERVER_POD_PATT', ''),
|
||||
label_selector=env.str('SD_LABEL_SELECTOR', ''))
|
||||
label_selector=env.str('SD_LABEL_SELECTOR', ''),
|
||||
port=env.int('SD_PORT', 19530))
|
||||
elif SD_PROVIDER == 'Static':
|
||||
from sd.static_provider import StaticProviderSettings
|
||||
SD_PROVIDER_SETTINGS = StaticProviderSettings(
|
||||
hosts=env.list('SD_STATIC_HOSTS', []))
|
||||
hosts=env.list('SD_STATIC_HOSTS', []),
|
||||
port=env.int('SD_STATIC_PORT', 19530))
|
||||
|
||||
# TESTING_WOSERVER = env.str('TESTING_WOSERVER', 'tcp://127.0.0.1:19530')
|
||||
|
||||
|
|
|
@ -226,12 +226,13 @@ class EventHandler(threading.Thread):
|
|||
|
||||
class KubernetesProviderSettings:
|
||||
def __init__(self, namespace, pod_patt, label_selector, in_cluster,
|
||||
poll_interval, **kwargs):
|
||||
poll_interval, port=None, **kwargs):
|
||||
self.namespace = namespace
|
||||
self.pod_patt = pod_patt
|
||||
self.label_selector = label_selector
|
||||
self.in_cluster = in_cluster
|
||||
self.poll_interval = poll_interval
|
||||
self.port = int(port) if port else 19530
|
||||
|
||||
|
||||
@singleton
|
||||
|
@ -245,6 +246,7 @@ class KubernetesProvider(object):
|
|||
self.label_selector = settings.label_selector
|
||||
self.in_cluster = settings.in_cluster
|
||||
self.poll_interval = settings.poll_interval
|
||||
self.port = settings.port
|
||||
self.kwargs = kwargs
|
||||
self.queue = queue.Queue()
|
||||
|
||||
|
@ -279,7 +281,7 @@ class KubernetesProvider(object):
|
|||
**kwargs)
|
||||
|
||||
def add_pod(self, name, ip):
|
||||
self.conn_mgr.register(name, 'tcp://{}:19530'.format(ip))
|
||||
self.conn_mgr.register(name, 'tcp://{}:{}'.format(ip, self.port))
|
||||
|
||||
def delete_pod(self, name):
|
||||
self.conn_mgr.unregister(name)
|
||||
|
|
|
@ -9,8 +9,9 @@ from sd import ProviderManager
|
|||
|
||||
|
||||
class StaticProviderSettings:
|
||||
def __init__(self, hosts):
|
||||
def __init__(self, hosts, port=None):
|
||||
self.hosts = hosts
|
||||
self.port = int(port) if port else 19530
|
||||
|
||||
|
||||
@singleton
|
||||
|
@ -21,6 +22,7 @@ class KubernetesProvider(object):
|
|||
def __init__(self, settings, conn_mgr, **kwargs):
|
||||
self.conn_mgr = conn_mgr
|
||||
self.hosts = [socket.gethostbyname(host) for host in settings.hosts]
|
||||
self.port = settings.port
|
||||
|
||||
def start(self):
|
||||
for host in self.hosts:
|
||||
|
@ -31,7 +33,7 @@ class KubernetesProvider(object):
|
|||
self.delete_pod(host)
|
||||
|
||||
def add_pod(self, name, ip):
|
||||
self.conn_mgr.register(name, 'tcp://{}:19530'.format(ip))
|
||||
self.conn_mgr.register(name, 'tcp://{}:{}'.format(ip, self.port))
|
||||
|
||||
def delete_pod(self, name):
|
||||
self.conn_mgr.unregister(name)
|
||||
|
|
Loading…
Reference in New Issue