com.dnastack.audit.emitter.AirliftAuditEventLogEmitter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dnastack-audit-event-logger Show documentation

Logs audit events.

The newest version!

package com.dnastack.audit.emitter;

import com.dnastack.audit.ObjectMapperFactory;
import com.dnastack.audit.exception.FailedToWriteJsonException;
import com.dnastack.audit.model.AuditEventBody;
import com.dnastack.audit.model.JsonPrinterStyle;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectWriter;
import io.airlift.log.Logger;

/**
 * Emits audit events to Airlift logs (used in Trino plugins)
 */
public class AirliftAuditEventLogEmitter implements AuditEventLogEmitter {

    private final Logger logger = Logger.get(AirliftAuditEventLogEmitter.class);
    private final ObjectWriter objectWriter;

    public AirliftAuditEventLogEmitter(JsonPrinterStyle jsonPrinterStyle) {
        this.objectWriter = ObjectMapperFactory.createWrite(jsonPrinterStyle);
    }

    @Override
    public void emit(AuditEventBody auditEvent) {
        try {
            logger.info("%s", objectWriter.writeValueAsString(auditEvent));
        } catch (JsonProcessingException e) {
            String errorMessage = String.format(
                "[%s | %s] Could not write JSON from %s",
                auditEvent.getContext().getTraceId(),
                auditEvent.getContext().getSpanId(),
                auditEvent);
            throw new FailedToWriteJsonException(errorMessage, e);
        }
    }

}