execution-time-logger.js

import { getTraceIdFromLocalStorage } from "./local-storage.js"

export default async (name, callback) => {
    const now = Date.now()
    const result = callback.apply()

    if (result && typeof result.finally === 'function') {
        return result.finally(logExecutionTime(now, name))
    } else {
        logExecutionTime(now, name)
        return result
    }
}

function logExecutionTime(now, name) {
    return () => {
        const then = Date.now()
        const duration = then - now

        const message = `Finished ${name} execution in ${duration}ms with id ${getTraceIdFromLocalStorage()}`
        if (duration > 1_000) {
            console.warn(message)
        } else if (duration > 5_000) {
            console.error(message)
        } else {
            console.info(message)
        }
    }
}