def __retrieve_entity_events()

in google-datacatalog-apache-atlas-connector/src/google/datacatalog_connectors/apache_atlas/scrape/metadata_event_scraper.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

    def __retrieve_entity_events(self, entity_events=None):
        # In case we receive the entity_events we don't create
        # the event consumer.
        if entity_events:
            return entity_events, None

        if not entity_events:
            entity_events = []

            event_consumer = self.__apache_atlas_event_facade.\
                create_event_consumer()

            for msg in event_consumer:
                if msg:
                    logging.info("Event %s:%s:%s: key=%s ", msg.topic,
                                 msg.partition, msg.offset, msg.key)
                    entity_event = json.loads(msg.value)
                    entity_events.append(entity_event['message'])
            return entity_events, event_consumer