budibase/packages/datastores/tests/diagnosticPlugin.js

import {eventsList} from "budibase-core";
import {filter,union,has,map} from "lodash/fp";
import records from "./records";

const allEventsOfType = type => 
    filter(
        e => e.endsWith(`:${type}`)
    )(eventsList);

const getEventNamespace = ev => {
    const parts = ev.split(":");
    return `${parts[0]}:${parts[1]}`;
}

const hasRecord = has("record");

export const register = (app, logTimeElapsed, eventNamespaces = []) => {
    const onCompleteEvents = 
        eventNamespaces.length === 0
        ? allEventsOfType("onComplete")
        : map(e => `${e}:onComplete`)(eventNamespaces)

    const onErrorEvents = 
        eventNamespaces.length === 0
        ? allEventsOfType("onError")
        : map(e => `${e}:onError`)(eventNamespaces)

    for(let ev of union(onCompleteEvents)(onErrorEvents)) {
        app.subscribe(ev, (_, ctx) => {
            const info = 
                hasRecord(ctx)
                ? ctx.record.type()
                : "";

            logTimeElapsed(
                ev, ctx.elapsed, info);
        });
    }
};