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