Interface health for AI agents

# -*- coding: utf-8 -*-

# Zato
from zato.common.audit_log.api import AuditOutcome, AuditSource
from zato.server.service import Service

# #####################################################################
# #####################################################################

class InterfaceHealth(Service):
    """ Returns, per interface, the HL7 messages that failed since a given
    time, with the control ID and error of each failure.
    """
    name = 'interface.health'

    input  = 'since'
    output = 'since', 'has_failures', 'interfaces'

    def handle(self) -> 'None':

        # Read the start of the requested period ..
        since = self.request.input.since

        # .. look up every failure recorded since then ..
        failures = self.audit.search(
            source=AuditSource.MLLP_Channel,
            outcome=AuditOutcome.Error,
            time_from=since,
        )

        # .. group the failures by the channel they arrived on ..
        by_interface = {}

        for failure in failures:

            # .. find or start the group of this failure's channel ..
            interface = failure['object_name']
            if interface not in by_interface:
                by_interface[interface] = []

            # .. copy only the fields needed by the caller ..
            by_interface[interface].append({
                'control_id': failure['msg_id'],
                'error': failure['status'],
                'received_at': failure['event_time_iso'],
            })

        # .. turn each group into one interface result ..
        interfaces = []

        for interface, interface_failures in by_interface.items():

            # .. with the failures of that interface and their count ..
            interfaces.append({
                'interface': interface,
                'failure_count': len(interface_failures),
                'failures': interface_failures,
            })

        # .. determine whether the period contains failures ..
        has_failures = len(interfaces) > 0

        # .. and return the declared tool response.
        self.response.payload = {
            'since': since,
            'has_failures': has_failures,
            'interfaces': interfaces,
        }