Remove Health Checker
This commit deletes the health_checker and the associated unit test. Change-Id: I0a1e4d2609c42607b021a1a5b1d3e7f29cbfea5f
This commit is contained in:
Jenna Kleiman
parent
f973b8acfc
commit
6fe9c32e4a
@ -1,163 +0,0 @@
|
||||
#
|
||||
# Copyright 2014-2017 AT&T Intellectual Property
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import uuid
|
||||
|
||||
from oslo_config import cfg
|
||||
|
||||
from valet.common.conf import get_logger
|
||||
from valet.common.music import REST
|
||||
from valet.engine.conf import init_engine
|
||||
|
||||
CONF = cfg.CONF
|
||||
LOG = get_logger(__name__)
|
||||
|
||||
|
||||
class HealthCheck(object):
|
||||
|
||||
rest = None
|
||||
|
||||
def __init__(self, hosts=[]):
|
||||
# default health_timeout=10
|
||||
self.tries = CONF.engine.health_timeout * 2
|
||||
self.uuid = str(uuid.uuid4())
|
||||
|
||||
kwargs = {
|
||||
'hosts': hosts,
|
||||
'port': CONF.music.port,
|
||||
'path': '/MUSIC/rest',
|
||||
'timeout': CONF.music.interval,
|
||||
}
|
||||
self.rest = REST(**kwargs)
|
||||
|
||||
def ping(self):
|
||||
|
||||
engine_id = None
|
||||
try:
|
||||
if self._send():
|
||||
engine_id = self._read_response()
|
||||
finally:
|
||||
self._delete_result()
|
||||
return engine_id
|
||||
|
||||
def _send(self):
|
||||
request = [
|
||||
{
|
||||
"action": "ping",
|
||||
"stack_id": self.uuid
|
||||
}
|
||||
]
|
||||
|
||||
data = {
|
||||
"values": {
|
||||
"stack_id": self.uuid,
|
||||
"request": request
|
||||
},
|
||||
"consistencyInfo": {
|
||||
"type": "eventual"
|
||||
}
|
||||
}
|
||||
|
||||
path = '/keyspaces/%(keyspace)s/tables/%(table)s/rows' % {
|
||||
'keyspace': CONF.music.keyspace,
|
||||
'table': CONF.music.request_table,
|
||||
}
|
||||
response = self.rest.request(method='post', path=path, data=data)
|
||||
|
||||
return response.status_code == 204 if response else False
|
||||
|
||||
def _read_response(self):
|
||||
|
||||
engine_id = None
|
||||
pre = '/keyspaces/%(keyspace)s/tables/%(table)s/rows?stack_id=%(uid)s'
|
||||
path = pre % {
|
||||
'keyspace': CONF.music.keyspace,
|
||||
'table': CONF.music.response_table,
|
||||
'uid': self.uuid,
|
||||
}
|
||||
|
||||
# default 20 tries * 0.5 sec = 10 sec. timeout
|
||||
for i in range(self.tries):
|
||||
time.sleep(0.5)
|
||||
try:
|
||||
response = self.rest.request(method='get', path=path)
|
||||
|
||||
if response.status_code == 200 and len(response.text) > 3:
|
||||
|
||||
j = json.loads(response.text)
|
||||
if j['row 0']['stack_id'] != self.uuid:
|
||||
continue
|
||||
|
||||
placement = json.loads(j['row 0']['placement'])
|
||||
engine_id = placement['resources']['id']
|
||||
break
|
||||
except Exception as e:
|
||||
LOG.warning("HealthCheck exception in read response, ", str(e))
|
||||
|
||||
return engine_id
|
||||
|
||||
def _delete_result(self):
|
||||
# leave a clean table - delete from requests and responses
|
||||
data = {
|
||||
"consistencyInfo": {"type": "eventual"}
|
||||
}
|
||||
|
||||
base = '/keyspaces/%(keyspace)s/tables/%(table)s/rows?stack_id=%(uid)s'
|
||||
try:
|
||||
path = base % {
|
||||
'keyspace': CONF.music.keyspace,
|
||||
'table': CONF.music.request_table,
|
||||
'uid': self.uuid
|
||||
}
|
||||
self.rest.request(method='delete', path=path, data=data)
|
||||
except Exception as e:
|
||||
LOG.warning("HealthCheck exception in delete request, ", str(e))
|
||||
|
||||
try:
|
||||
path = base % {
|
||||
'keyspace': CONF.music.keyspace,
|
||||
'table': CONF.music.response_table,
|
||||
'uid': self.uuid
|
||||
}
|
||||
self.rest.request(method='delete', path=path, data=data)
|
||||
except Exception as e:
|
||||
LOG.warning("HealthCheck exception in delete response, ", str(e))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
respondent_id = None
|
||||
code = 0
|
||||
init_engine(default_config_files=['/etc/valet/valet.conf'])
|
||||
|
||||
if os.path.exists(CONF.engine.pid):
|
||||
respondent_id = HealthCheck().ping()
|
||||
|
||||
if respondent_id == CONF.engine.priority:
|
||||
code = CONF.engine.priority
|
||||
LOG.info("HealthCheck - Alive, "
|
||||
"respondent instance id: {}".format(respondent_id))
|
||||
else:
|
||||
LOG.warning("HealthCheck - pid file exists, "
|
||||
"engine {} did not respond in a timely manner "
|
||||
"(respondent id {})".format(CONF.engine.priority,
|
||||
respondent_id))
|
||||
else:
|
||||
LOG.info("HealthCheck - no pid file, engine is not running!")
|
||||
sys.exit(code)
|
||||
@ -1,93 +0,0 @@
|
||||
#
|
||||
# Copyright 2014-2017 AT&T Intellectual Property
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
import mock
|
||||
|
||||
import valet.engine.optimizer.ostro_server.health_checker as ping
|
||||
from valet.engine.optimizer.ostro_server.health_checker import HealthCheck
|
||||
from valet.tests.base import Base
|
||||
|
||||
json = (r'{"row 0":{"placement": "{\"status\": {\"message\": '
|
||||
r'\"ping\", \"type\": \"ok\"},\"resources\": '
|
||||
r'{\"ip\": \"localhost\", \"id\": %d}}","stack_id":"%s"}}')
|
||||
|
||||
|
||||
class TestHealthCheck(Base):
|
||||
|
||||
def setUp(self):
|
||||
super(TestHealthCheck, self).setUp()
|
||||
ping.CONF = mock.MagicMock()
|
||||
ping.REST = mock.MagicMock()
|
||||
self.pingger = HealthCheck()
|
||||
|
||||
@mock.patch.object(HealthCheck, '_send')
|
||||
@mock.patch.object(HealthCheck, '_read_response')
|
||||
def test_ping(self, mock_read, mock_send):
|
||||
mock_send.return_value = True
|
||||
mock_read.return_value = True
|
||||
|
||||
self.validate_test(self.pingger.ping() == 1)
|
||||
|
||||
@mock.patch.object(HealthCheck, '_send')
|
||||
@mock.patch.object(HealthCheck, '_read_response')
|
||||
def test_ping_unhappy(self, mock_read, mock_send):
|
||||
mock_send.return_value = False
|
||||
mock_read.return_value = True
|
||||
|
||||
self.validate_test(self.pingger.ping() is None)
|
||||
|
||||
@mock.patch.object(HealthCheck, '_send')
|
||||
@mock.patch.object(HealthCheck, '_read_response')
|
||||
def test_ping_unhappy_2(self, mock_read, mock_send):
|
||||
mock_send.return_value = True
|
||||
mock_read.return_value = False
|
||||
|
||||
self.validate_test(not self.pingger.ping())
|
||||
|
||||
def test_send(self):
|
||||
self.pingger.rest.request.return_value.status_code = 204
|
||||
self.validate_test(self.pingger._send())
|
||||
|
||||
def test_send_unhappy(self):
|
||||
self.pingger.rest.request.return_value.status_code = 200
|
||||
self.validate_test(not self.pingger._send())
|
||||
|
||||
def test_read_response(self):
|
||||
mid = 1
|
||||
self.pingger.rest.request.return_value.status_code = 200
|
||||
self.pingger.rest.request.return_value.text = json % (
|
||||
mid, self.pingger.uuid)
|
||||
self.validate_test(self.pingger._read_response())
|
||||
|
||||
def test_read_response_from_other_engine(self):
|
||||
my_id = 1
|
||||
self.pingger.rest.request.return_value.status_code = 200
|
||||
self.pingger.rest.request.return_value.text = json % (
|
||||
my_id, self.pingger.uuid)
|
||||
self.validate_test(not self.pingger._read_response() == 2)
|
||||
|
||||
def test_read_response_unhappy_wrong_res_code(self):
|
||||
self.pingger.rest.request.return_value.status_code = 204
|
||||
self.pingger.rest.request.return_value.text = self.pingger.uuid
|
||||
self.validate_test(not self.pingger._read_response())
|
||||
|
||||
def test_read_response_unhappy_wrong_body(self):
|
||||
self.pingger.rest.request.return_value.status_code = 200
|
||||
self.pingger.rest.request.return_value.text = ""
|
||||
self.validate_test(not self.pingger._read_response())
|
||||
|
||||
def test_read_response_unhappy_wrong_both(self):
|
||||
self.pingger.rest.request.return_value.status_code = 204
|
||||
self.pingger.rest.request.return_value.text = ""
|
||||
self.validate_test(not self.pingger._read_response())
|
||||
Loading…
Reference in New Issue
Block a user