def update_status(service_record):
global service_statuses, my_service_record
hostid = service_record['hostid']
servicename = service_record['servicename']
base_url = service_record.get('base_url', 'N/A')
service = '__'.join([hostid, servicename, base_url])
with session_scope() as dbsession:
db_service_record = db_services.get(hostid,
servicename,
base_url,
session=dbsession)
logger.debug("db service record: {}".format(db_service_record))
if db_service_record:
my_service_record = db_service_record
my_service_record['heartbeat'] = time.time()
if service_statuses[service]['up'] and service_statuses[service][
'available']:
my_service_record['status'] = True
else:
my_service_record['status'] = False
my_service_record['short_description'] = json.dumps(
service_statuses[service])
db_services.update_record(my_service_record, session=dbsession)
else:
db_services.add(hostid,
servicename,
service_record,
session=dbsession)
return True
def update_status(service_record):
global service_statuses, my_service_record
hostid = service_record["hostid"]
servicename = service_record["servicename"]
base_url = service_record.get("base_url", "N/A")
service = "__".join([hostid, servicename, base_url])
with session_scope() as dbsession:
db_service_record = db_services.get(
hostid, servicename, base_url, session=dbsession
)
logger.debug("db service record: {}".format(db_service_record))
if db_service_record:
my_service_record = db_service_record
my_service_record["heartbeat"] = time.time()
if (
service_statuses[service]["up"]
and service_statuses[service]["available"]
):
my_service_record["status"] = True
else:
my_service_record["status"] = False
my_service_record["short_description"] = json.dumps(
service_statuses[service]
)
db_services.update_record(my_service_record, session=dbsession)
else:
db_services.add(hostid, servicename, service_record, session=dbsession)
return True
def _register(self):
if not self.is_enabled:
logger.error(
'Service not enabled in config, not registering service: ' +
self.name)
raise Exception('No service enabled, cannot continue bootstrap')
logger.info('Registering service: {}'.format(self.name))
service_template = {
'type': 'anchore',
'base_url': 'N/A',
'status_base_url': 'N/A',
'version': 'v1',
'short_description': ''
}
hstring = 'http'
if 'external_tls' in self.configuration:
if self.configuration.get('external_tls', False):
hstring = 'https'
elif 'ssl_enable' in self.configuration:
if self.configuration.get('ssl_enable', False):
hstring = 'https'
endpoint_hostname = endpoint_port = endpoint_hostport = None
if self.configuration.get('external_hostname', False):
endpoint_hostname = self.configuration.get('external_hostname')
elif self.configuration.get('endpoint_hostname', False):
endpoint_hostname = self.configuration.get('endpoint_hostname')
if self.configuration.get('external_port', False):
endpoint_port = int(self.configuration.get('external_port'))
elif self.configuration.get('port', False):
endpoint_port = int(self.configuration.get('port'))
if endpoint_hostname:
endpoint_hostport = endpoint_hostname
if endpoint_port:
endpoint_hostport = endpoint_hostport + ":" + str(
endpoint_port)
if endpoint_hostport:
service_template['base_url'] = "{}://{}".format(
hstring, endpoint_hostport)
else:
raise Exception(
"could not construct service base_url - please check service configuration for hostname/port settings"
)
try:
service_template['status'] = False
service_template['status_message'] = taskstate.base_state(
'service_status')
with session_scope() as dbsession:
service_records = db_services.get_byname(self.__service_name__,
session=dbsession)
# fail if trying to add a service that must be unique in the system, but one already is registered in DB
if self.__is_unique_service__:
if len(service_records) > 1:
raise Exception(
'more than one entry for service type (' +
str(self.__service_name__) +
') exists in DB, but service must be unique - manual DB intervention required'
)
for service_record in service_records:
if service_record and (service_record['hostid'] !=
self.instance_id):
raise Exception(
'service type (' + str(self.__service_name__) +
') already exists in system with different host_id - detail: my_host_id='
+ str(self.instance_id) + ' db_host_id=' +
str(service_record['hostid']))
# if all checks out, then add/update the registration
ret = db_services.add(self.instance_id,
self.__service_name__,
service_template,
session=dbsession)
try:
my_service_record = {
'hostid': self.instance_id,
'servicename': self.__service_name__,
}
my_service_record.update(service_template)
servicestatus.set_my_service_record(my_service_record)
self.service_record = my_service_record
except Exception as err:
logger.warn(
'could not set local service information - exception: {}'
.format(str(err)))
except Exception as err:
raise err
service_record = servicestatus.get_my_service_record()
servicestatus.set_status(service_record,
up=True,
available=True,
update_db=True,
versions=self.versions)
logger.info('Service registration complete')
return True
def registerService(sname, config, enforce_unique=True):
ret = False
myconfig = config['services'][sname]
# TODO add version support/detection here
service_template = {'type': 'anchore', 'base_url': 'N/A', 'version': 'v1'}
if 'ssl_enable' in myconfig and myconfig['ssl_enable']:
hstring = "https"
else:
hstring = "http"
endpoint_hostname = endpoint_port = endpoint_hostport = None
if 'endpoint_hostname' in myconfig:
endpoint_hostname = myconfig['endpoint_hostname']
service_template[
'base_url'] = hstring + "://" + myconfig['endpoint_hostname']
if 'port' in myconfig:
endpoint_port = int(myconfig['port'])
service_template['base_url'] += ":" + str(endpoint_port)
if endpoint_hostname:
endpoint_hostport = endpoint_hostname
if endpoint_port:
endpoint_hostport = endpoint_hostport + ":" + str(endpoint_port)
try:
service_template['status'] = True
service_template['status_message'] = "registered"
with session_scope() as dbsession:
service_records = db_services.get_byname(sname, session=dbsession)
# fail if trying to add a service that must be unique in the system, but one already is registered in DB
if enforce_unique:
if len(service_records) > 1:
raise Exception(
"more than one entry for service type (" + str(sname) +
") exists in DB, but service must be unique - manual DB intervention required"
)
for service_record in service_records:
if service_record and (service_record['hostid'] !=
config['host_id']):
raise Exception(
"service type (" + str(sname) +
") already exists in system with different host_id - detail: my_host_id="
+ str(config['host_id']) + " db_host_id=" +
str(service_record['hostid']))
# in any case, check if another host is registered that has the same endpoint
for service_record in service_records:
if service_record[
'base_url'] and service_record['base_url'] != 'N/A':
service_hostport = re.sub("^http.//", "",
service_record['base_url'])
# if a different host_id has the same endpoint, fail
if (service_hostport == endpoint_hostport) and (
config['host_id'] != service_record['hostid']):
raise Exception(
"trying to add new host but found conflicting endpoint from another host in DB - detail: my_host_id="
+ str(config['host_id']) + " db_host_id=" +
str(service_record['hostid']) +
" my_host_endpoint=" + str(endpoint_hostport) +
" db_host_endpoint=" + str(service_hostport))
7
# if all checks out, then add/update the registration
ret = db_services.add(config['host_id'],
sname,
service_template,
session=dbsession)
except Exception as err:
raise err
return (ret)
def registerService(sname, config, enforce_unique=True):
ret = False
myconfig = config['services'][sname]
# TODO add version support/detection here
service_template = {
'type': 'anchore',
'base_url': 'N/A',
'status_base_url': 'N/A',
'version': 'v1',
'short_description': ''
}
#if 'ssl_enable' in myconfig and myconfig['ssl_enable']:
if myconfig.get('ssl_enable', False) or myconfig.get(
'external_tls', False):
hstring = "https"
else:
hstring = "http"
endpoint_hostname = endpoint_port = endpoint_hostport = None
if 'endpoint_hostname' in myconfig:
endpoint_hostname = myconfig['endpoint_hostname']
service_template[
'base_url'] = hstring + "://" + myconfig['endpoint_hostname']
if 'port' in myconfig:
endpoint_port = int(myconfig['port'])
service_template['base_url'] += ":" + str(endpoint_port)
if endpoint_hostname:
endpoint_hostport = endpoint_hostname
if endpoint_port:
endpoint_hostport = endpoint_hostport + ":" + str(endpoint_port)
try:
service_template['status'] = False
service_template['status_message'] = taskstate.base_state(
'service_status')
with session_scope() as dbsession:
service_records = db_services.get_byname(sname, session=dbsession)
# fail if trying to add a service that must be unique in the system, but one already is registered in DB
if enforce_unique:
if len(service_records) > 1:
raise Exception(
"more than one entry for service type (" + str(sname) +
") exists in DB, but service must be unique - manual DB intervention required"
)
for service_record in service_records:
if service_record and (service_record['hostid'] !=
config['host_id']):
raise Exception(
"service type (" + str(sname) +
") already exists in system with different host_id - detail: my_host_id="
+ str(config['host_id']) + " db_host_id=" +
str(service_record['hostid']))
# if all checks out, then add/update the registration
ret = db_services.add(config['host_id'],
sname,
service_template,
session=dbsession)
try:
my_service_record = {
'hostid': config['host_id'],
'servicename': sname,
}
my_service_record.update(service_template)
servicestatus.set_my_service_record(my_service_record)
except Exception as err:
logger.warn(
"could not set local service information - exception: {}".
format(str(err)))
except Exception as err:
raise err
return (ret)
def _register(self):
if not self.is_enabled:
logger.error(
"Service not enabled in config, not registering service: " +
self.name)
raise Exception("No service enabled, cannot continue bootstrap")
logger.info("Registering service: {}".format(self.name))
service_template = {
"type": "anchore",
"base_url": "N/A",
"status_base_url": "N/A",
"version": "v1",
"short_description": "",
}
hstring = "http"
if "external_tls" in self.configuration:
if self.configuration.get("external_tls", False):
hstring = "https"
elif "ssl_enable" in self.configuration:
if self.configuration.get("ssl_enable", False):
hstring = "https"
endpoint_hostname = endpoint_port = endpoint_hostport = None
if self.configuration.get("external_hostname", False):
endpoint_hostname = self.configuration.get("external_hostname")
elif self.configuration.get("endpoint_hostname", False):
endpoint_hostname = self.configuration.get("endpoint_hostname")
if self.configuration.get("external_port", False):
endpoint_port = int(self.configuration.get("external_port"))
elif self.configuration.get("port", False):
endpoint_port = int(self.configuration.get("port"))
if endpoint_hostname:
endpoint_hostport = endpoint_hostname
if endpoint_port:
endpoint_hostport = endpoint_hostport + ":" + str(
endpoint_port)
if endpoint_hostport:
service_template["base_url"] = "{}://{}".format(
hstring, endpoint_hostport)
else:
raise Exception(
"could not construct service base_url - please check service configuration for hostname/port settings"
)
try:
service_template["status"] = False
service_template["status_message"] = taskstate.base_state(
"service_status")
with session_scope() as dbsession:
service_records = db_services.get_byname(self.__service_name__,
session=dbsession)
# fail if trying to add a service that must be unique in the system, but one already is registered in DB
if self.__is_unique_service__:
if len(service_records) > 1:
raise Exception(
"more than one entry for service type (" +
str(self.__service_name__) +
") exists in DB, but service must be unique - manual DB intervention required"
)
for service_record in service_records:
if service_record and (service_record["hostid"] !=
self.instance_id):
raise Exception(
"service type (" + str(self.__service_name__) +
") already exists in system with different host_id - detail: my_host_id="
+ str(self.instance_id) + " db_host_id=" +
str(service_record["hostid"]))
# if all checks out, then add/update the registration
ret = db_services.add(
self.instance_id,
self.__service_name__,
service_template,
session=dbsession,
)
try:
my_service_record = {
"hostid": self.instance_id,
"servicename": self.__service_name__,
}
my_service_record.update(service_template)
servicestatus.set_my_service_record(my_service_record)
self.service_record = my_service_record
except Exception as err:
logger.warn(
"could not set local service information - exception: {}"
.format(str(err)))
except Exception as err:
raise err
service_record = servicestatus.get_my_service_record()
servicestatus.set_status(
service_record,
up=True,
available=True,
update_db=True,
versions=self.versions,
)
logger.info("Service registration complete")
return True
