@squidcloud/local-backend 1.0.49 → 1.0.51

package/dist/internal-common/src/kafka/kafka.manager.d.ts

@@ -1,22 +1,22 @@
-import { HeartbeatProvider } from '@squidcloud/common';
-import { KafkaConnectionConfig, MessageEnvelope, TopicId } from './kafka.types';
-import { Observable } from 'rxjs';
-export declare class KafkaManager implements HeartbeatProvider {
-    private readonly connectionConfig;
-    private kafka;
-    private readonly producerMap;
-    private readonly pubSubSubjectMap;
-    private readonly testMessageObs;
-    constructor(connectionConfig: KafkaConnectionConfig);
-    publishMessage<T>(topicId: TopicId, message: T, key: string): Promise<void>;
-    observePubSubTopic<T>(topicId: TopicId): Observable<T>;
-    observeTopic<T>(topicId: TopicId): Observable<MessageEnvelope<T>>;
-    observeBatchTopic<T>(topicId: TopicId): Observable<Array<MessageEnvelope<T>>>;
-    sendMessage<T>(topicId: TopicId, message: T, key: string): Promise<void>;
-    sendMessages<T>(topicId: TopicId, messages: Array<{
-        message: T;
-        key: string;
-    }>): Promise<void>;
-    private subscribeBatchAndRun;
-    checkHeartbeat(): Promise<void>;
-}
+import { HeartbeatProvider } from '@squidcloud/common';
+import { KafkaConnectionConfig, MessageEnvelope, TopicId } from './kafka.types';
+import { Observable } from 'rxjs';
+export declare class KafkaManager implements HeartbeatProvider {
+    private readonly connectionConfig;
+    private kafka;
+    private readonly producerMap;
+    private readonly pubSubSubjectMap;
+    private readonly testMessageObs;
+    constructor(connectionConfig: KafkaConnectionConfig);
+    publishMessage<T>(topicId: TopicId, message: T, key: string): Promise<void>;
+    observePubSubTopic<T>(topicId: TopicId): Observable<T>;
+    observeTopic<T>(topicId: TopicId): Observable<MessageEnvelope<T>>;
+    observeBatchTopic<T>(topicId: TopicId): Observable<Array<MessageEnvelope<T>>>;
+    sendMessage<T>(topicId: TopicId, message: T, key: string): Promise<void>;
+    sendMessages<T>(topicId: TopicId, messages: Array<{
+        message: T;
+        key: string;
+    }>): Promise<void>;
+    private subscribeBatchAndRun;
+    checkHeartbeat(): Promise<void>;
+}

package/dist/internal-common/src/kafka/kafka.manager.js

@@ -1,130 +1,130 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.KafkaManager = void 0;
-const common_1 = require("@squidcloud/common");
-const kafkajs_1 = require("kafkajs");
-const rxjs_1 = require("rxjs");
-class KafkaManager {
-    constructor(connectionConfig) {
-        this.connectionConfig = connectionConfig;
-        this.producerMap = {};
-        this.pubSubSubjectMap = {};
-        this.testMessageObs = new rxjs_1.Subject();
-        const config = {
-            clientId: (0, common_1.generateId)(),
-            brokers: connectionConfig.clusterUrls,
-            logCreator: connectionConfig.logAdapter,
-            sasl: connectionConfig.sasl,
-            ssl: connectionConfig.ssl,
-            logLevel: kafkajs_1.logLevel.WARN,
-        };
-        this.kafka = new kafkajs_1.Kafka(config);
-        this.observePubSubTopic('test_messages').subscribe((message) => {
-            this.testMessageObs.next(message);
-        });
-    }
-    async publishMessage(topicId, message, key) {
-        await this.sendMessage(topicId, message, key);
-    }
-    observePubSubTopic(topicId) {
-        if (this.pubSubSubjectMap[topicId]) {
-            return this.pubSubSubjectMap[topicId];
-        }
-        const subject = new rxjs_1.Subject();
-        const consumer = this.kafka.consumer({
-            groupId: `${topicId}-group - ${(0, common_1.generateId)()}`,
-            sessionTimeout: this.connectionConfig.consumerSessionTimeoutMs,
-        });
-        const obs = (0, rxjs_1.from)(this.subscribeBatchAndRun(consumer, topicId, subject)).pipe((0, rxjs_1.switchMap)(() => subject), (0, rxjs_1.finalize)(async () => {
-            delete this.pubSubSubjectMap[topicId];
-            await consumer.disconnect();
-        }), (0, rxjs_1.mergeMap)((a) => a), (0, rxjs_1.tap)((a) => a.commit()), (0, rxjs_1.map)((a) => a.data), (0, rxjs_1.share)());
-        this.pubSubSubjectMap[topicId] = obs;
-        return obs;
-    }
-    observeTopic(topicId) {
-        return this.observeBatchTopic(topicId).pipe((0, rxjs_1.mergeMap)((a) => a));
-    }
-    observeBatchTopic(topicId) {
-        const consumer = this.kafka.consumer({
-            groupId: `${topicId}-group`,
-            sessionTimeout: this.connectionConfig.consumerSessionTimeoutMs,
-            heartbeatInterval: 1000,
-        });
-        const subject = new rxjs_1.Subject();
-        return (0, rxjs_1.from)(this.subscribeBatchAndRun(consumer, topicId, subject)).pipe((0, rxjs_1.switchMap)(() => subject), (0, rxjs_1.finalize)(async () => {
-            await consumer.disconnect();
-        }), (0, rxjs_1.share)());
-    }
-    async sendMessage(topicId, message, key) {
-        await this.sendMessages(topicId, [{ message, key }]);
-    }
-    async sendMessages(topicId, messages) {
-        if (!this.producerMap[topicId]) {
-            const producer = this.kafka.producer({
-                createPartitioner: kafkajs_1.Partitioners.DefaultPartitioner,
-                retry: {
-                    retries: 5,
-                    initialRetryTime: 300,
-                    maxRetryTime: 5000,
-                },
-                idempotent: false,
-                transactionTimeout: 30000,
-            });
-            this.producerMap[topicId] = producer.connect().then(() => producer);
-        }
-        const producer = await this.producerMap[topicId];
-        const messagesWithValue = messages.map(({ message, key }) => {
-            return { value: (0, common_1.serializeObj)(message), key };
-        });
-        try {
-            await producer.send({ topic: topicId, messages: messagesWithValue });
-        }
-        catch (e) {
-            console.error('Error sending messages to Kafka', e, messagesWithValue);
-            throw e;
-        }
-    }
-    async subscribeBatchAndRun(consumer, topicId, subject) {
-        await consumer.connect();
-        await consumer.subscribe({ topic: topicId, fromBeginning: false });
-        await consumer.run({
-            eachBatchAutoResolve: true,
-            eachBatch: async ({ batch, resolveOffset }) => {
-                const messageEnvelopes = batch.messages
-                    .filter((message) => message.value !== null)
-                    .map((message) => {
-                    const data = (0, common_1.deserializeObj)((0, common_1.truthy)(message.value).toString());
-                    return {
-                        data,
-                        timestamp: parseInt(message.timestamp),
-                        offset: message.offset,
-                        partition: batch.partition,
-                        topicId,
-                        commit: () => {
-                            resolveOffset(message.offset);
-                        },
-                    };
-                });
-                if (messageEnvelopes.length) {
-                    subject.next(messageEnvelopes);
-                }
-            },
-        });
-    }
-    async checkHeartbeat() {
-        const randomMessage = { id: (0, common_1.generateId)() };
-        const messageObs = this.testMessageObs.pipe((0, rxjs_1.filter)((message) => {
-            return message.id === randomMessage.id;
-        }));
-        await (0, rxjs_1.firstValueFrom)((0, rxjs_1.timer)(0));
-        this.sendMessage('test_messages', randomMessage, 'test_key')
-            .then()
-            .catch((err) => {
-            console.log('Error while sending test message to Kafka', err);
-        });
-        await (0, rxjs_1.firstValueFrom)(messageObs);
-    }
-}
-exports.KafkaManager = KafkaManager;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KafkaManager = void 0;
+const common_1 = require("@squidcloud/common");
+const kafkajs_1 = require("kafkajs");
+const rxjs_1 = require("rxjs");
+class KafkaManager {
+    constructor(connectionConfig) {
+        this.connectionConfig = connectionConfig;
+        this.producerMap = {};
+        this.pubSubSubjectMap = {};
+        this.testMessageObs = new rxjs_1.Subject();
+        const config = {
+            clientId: (0, common_1.generateId)(),
+            brokers: connectionConfig.clusterUrls,
+            logCreator: connectionConfig.logAdapter,
+            sasl: connectionConfig.sasl,
+            ssl: connectionConfig.ssl,
+            logLevel: kafkajs_1.logLevel.WARN,
+        };
+        this.kafka = new kafkajs_1.Kafka(config);
+        this.observePubSubTopic('test_messages').subscribe((message) => {
+            this.testMessageObs.next(message);
+        });
+    }
+    async publishMessage(topicId, message, key) {
+        await this.sendMessage(topicId, message, key);
+    }
+    observePubSubTopic(topicId) {
+        if (this.pubSubSubjectMap[topicId]) {
+            return this.pubSubSubjectMap[topicId];
+        }
+        const subject = new rxjs_1.Subject();
+        const consumer = this.kafka.consumer({
+            groupId: `${topicId}-group - ${(0, common_1.generateId)()}`,
+            sessionTimeout: this.connectionConfig.consumerSessionTimeoutMs,
+        });
+        const obs = (0, rxjs_1.from)(this.subscribeBatchAndRun(consumer, topicId, subject)).pipe((0, rxjs_1.switchMap)(() => subject), (0, rxjs_1.finalize)(async () => {
+            delete this.pubSubSubjectMap[topicId];
+            await consumer.disconnect();
+        }), (0, rxjs_1.mergeMap)((a) => a), (0, rxjs_1.tap)((a) => a.commit()), (0, rxjs_1.map)((a) => a.data), (0, rxjs_1.share)());
+        this.pubSubSubjectMap[topicId] = obs;
+        return obs;
+    }
+    observeTopic(topicId) {
+        return this.observeBatchTopic(topicId).pipe((0, rxjs_1.mergeMap)((a) => a));
+    }
+    observeBatchTopic(topicId) {
+        const consumer = this.kafka.consumer({
+            groupId: `${topicId}-group`,
+            sessionTimeout: this.connectionConfig.consumerSessionTimeoutMs,
+            heartbeatInterval: 1000,
+        });
+        const subject = new rxjs_1.Subject();
+        return (0, rxjs_1.from)(this.subscribeBatchAndRun(consumer, topicId, subject)).pipe((0, rxjs_1.switchMap)(() => subject), (0, rxjs_1.finalize)(async () => {
+            await consumer.disconnect();
+        }), (0, rxjs_1.share)());
+    }
+    async sendMessage(topicId, message, key) {
+        await this.sendMessages(topicId, [{ message, key }]);
+    }
+    async sendMessages(topicId, messages) {
+        if (!this.producerMap[topicId]) {
+            const producer = this.kafka.producer({
+                createPartitioner: kafkajs_1.Partitioners.DefaultPartitioner,
+                retry: {
+                    retries: 5,
+                    initialRetryTime: 300,
+                    maxRetryTime: 5000,
+                },
+                idempotent: false,
+                transactionTimeout: 30000,
+            });
+            this.producerMap[topicId] = producer.connect().then(() => producer);
+        }
+        const producer = await this.producerMap[topicId];
+        const messagesWithValue = messages.map(({ message, key }) => {
+            return { value: (0, common_1.serializeObj)(message), key };
+        });
+        try {
+            await producer.send({ topic: topicId, messages: messagesWithValue });
+        }
+        catch (e) {
+            console.error('Error sending messages to Kafka', e, messagesWithValue);
+            throw e;
+        }
+    }
+    async subscribeBatchAndRun(consumer, topicId, subject) {
+        await consumer.connect();
+        await consumer.subscribe({ topic: topicId, fromBeginning: false });
+        await consumer.run({
+            eachBatchAutoResolve: true,
+            eachBatch: async ({ batch, resolveOffset }) => {
+                const messageEnvelopes = batch.messages
+                    .filter((message) => message.value !== null)
+                    .map((message) => {
+                    const data = (0, common_1.deserializeObj)((0, common_1.truthy)(message.value).toString());
+                    return {
+                        data,
+                        timestamp: parseInt(message.timestamp),
+                        offset: message.offset,
+                        partition: batch.partition,
+                        topicId,
+                        commit: () => {
+                            resolveOffset(message.offset);
+                        },
+                    };
+                });
+                if (messageEnvelopes.length) {
+                    subject.next(messageEnvelopes);
+                }
+            },
+        });
+    }
+    async checkHeartbeat() {
+        const randomMessage = { id: (0, common_1.generateId)() };
+        const messageObs = this.testMessageObs.pipe((0, rxjs_1.filter)((message) => {
+            return message.id === randomMessage.id;
+        }));
+        await (0, rxjs_1.firstValueFrom)((0, rxjs_1.timer)(0));
+        this.sendMessage('test_messages', randomMessage, 'test_key')
+            .then()
+            .catch((err) => {
+            console.log('Error while sending test message to Kafka', err);
+        });
+        await (0, rxjs_1.firstValueFrom)(messageObs);
+    }
+}
+exports.KafkaManager = KafkaManager;
 //# sourceMappingURL=kafka.manager.js.map

package/dist/internal-common/src/kafka/kafka.types.d.ts

@@ -1,17 +1,17 @@
-import { LogEntry, logLevel, Mechanism, SASLOptions } from 'kafkajs';
-export type TopicId = string;
-export interface MessageEnvelope<T> {
-    data: T;
-    offset: string;
-    timestamp: number;
-    partition: number;
-    topicId: TopicId;
-    commit: () => void;
-}
-export interface KafkaConnectionConfig {
-    clusterUrls: string[];
-    logAdapter?: (logLevel: logLevel) => (entry: LogEntry) => void;
-    ssl: boolean;
-    sasl?: SASLOptions | Mechanism;
-    consumerSessionTimeoutMs: number;
-}
+import { LogEntry, logLevel, Mechanism, SASLOptions } from 'kafkajs';
+export type TopicId = string;
+export interface MessageEnvelope<T> {
+    data: T;
+    offset: string;
+    timestamp: number;
+    partition: number;
+    topicId: TopicId;
+    commit: () => void;
+}
+export interface KafkaConnectionConfig {
+    clusterUrls: string[];
+    logAdapter?: (logLevel: logLevel) => (entry: LogEntry) => void;
+    ssl: boolean;
+    sasl?: SASLOptions | Mechanism;
+    consumerSessionTimeoutMs: number;
+}

package/dist/internal-common/src/kafka/kafka.types.js

@@ -1,3 +1,3 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
 //# sourceMappingURL=kafka.types.js.map

package/dist/internal-common/src/kafka-observability-publisher.d.ts

@@ -1,11 +1,11 @@
-import { LogEvent, MetricEvent } from './observability.types';
-import { KafkaManager } from './kafka/kafka.manager';
-import { ObservabilityPublisher } from './observability-publisher';
-export declare class KafkaObservabilityPublisher extends ObservabilityPublisher {
-    private readonly kafkaManager;
-    private readonly metricsTopicId;
-    private readonly logsTopicId;
-    constructor(kafkaManager: KafkaManager, metricsTopicId: string, logsTopicId: string);
-    reportMetric(event: Omit<MetricEvent, 'timestamp'>): Promise<void>;
-    reportLog(event: LogEvent): Promise<void>;
-}
+import { LogEvent, MetricEvent } from './observability.types';
+import { KafkaManager } from './kafka/kafka.manager';
+import { ObservabilityPublisher } from './observability-publisher';
+export declare class KafkaObservabilityPublisher extends ObservabilityPublisher {
+    private readonly kafkaManager;
+    private readonly metricsTopicId;
+    private readonly logsTopicId;
+    constructor(kafkaManager: KafkaManager, metricsTopicId: string, logsTopicId: string);
+    reportMetric(event: Omit<MetricEvent, 'timestamp'>): Promise<void>;
+    reportLog(event: LogEvent): Promise<void>;
+}

package/dist/internal-common/src/kafka-observability-publisher.js

@@ -1,21 +1,21 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.KafkaObservabilityPublisher = void 0;
-const observability_publisher_1 = require("./observability-publisher");
-class KafkaObservabilityPublisher extends observability_publisher_1.ObservabilityPublisher {
-    constructor(kafkaManager, metricsTopicId, logsTopicId) {
-        super();
-        this.kafkaManager = kafkaManager;
-        this.metricsTopicId = metricsTopicId;
-        this.logsTopicId = logsTopicId;
-    }
-    async reportMetric(event) {
-        const modifiedEvent = Object.assign(Object.assign({}, event), { timestamp: new Date(), name: `squid_${event.name}` });
-        await this.kafkaManager.sendMessage(this.metricsTopicId, modifiedEvent, 'report_metric');
-    }
-    async reportLog(event) {
-        await this.kafkaManager.sendMessage(this.logsTopicId, event, 'report_log');
-    }
-}
-exports.KafkaObservabilityPublisher = KafkaObservabilityPublisher;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KafkaObservabilityPublisher = void 0;
+const observability_publisher_1 = require("./observability-publisher");
+class KafkaObservabilityPublisher extends observability_publisher_1.ObservabilityPublisher {
+    constructor(kafkaManager, metricsTopicId, logsTopicId) {
+        super();
+        this.kafkaManager = kafkaManager;
+        this.metricsTopicId = metricsTopicId;
+        this.logsTopicId = logsTopicId;
+    }
+    async reportMetric(event) {
+        const modifiedEvent = Object.assign(Object.assign({}, event), { timestamp: new Date(), name: `squid_${event.name}` });
+        await this.kafkaManager.sendMessage(this.metricsTopicId, modifiedEvent, 'report_metric');
+    }
+    async reportLog(event) {
+        await this.kafkaManager.sendMessage(this.logsTopicId, event, 'report_log');
+    }
+}
+exports.KafkaObservabilityPublisher = KafkaObservabilityPublisher;
 //# sourceMappingURL=kafka-observability-publisher.js.map

package/dist/internal-common/src/local-dev.types.d.ts

@@ -1,31 +1,31 @@
-import { RunSecrets } from '@squidcloud/common';
-import { BatchRunConfigurationRequest } from './code-executor.types';
-export type MessageFromDeveloperType = 'bundleData' | 'response';
-interface BaseMessageFromDeveloper {
-    type: MessageFromDeveloperType;
-    payload?: unknown;
-    requestId: string;
-}
-export type MessageFromDeveloper = ResponseMessageFromDeveloper;
-export interface ResponseMessageFromDeveloper extends BaseMessageFromDeveloper {
-    type: 'response';
-    payload: any;
-}
-export type MessageToDeveloperType = 'executeFunction' | 'localBackendData';
-interface BaseMessageToDeveloper {
-    type: MessageToDeveloperType;
-    payload: unknown;
-    requestId: string;
-}
-export type MessageToDeveloper = ExecuteFunctionMessageToDeveloper | LocalBackendDataMessageToDeveloper;
-export interface ExecuteFunctionMessageToDeveloper extends BaseMessageToDeveloper {
-    type: 'executeFunction';
-    payload: BatchRunConfigurationRequest;
-}
-export interface LocalBackendDataMessageToDeveloper extends BaseMessageToDeveloper {
-    type: 'localBackendData';
-    payload: {
-        secrets: RunSecrets;
-    };
-}
-export {};
+import { RunSecrets } from '@squidcloud/common';
+import { BatchRunConfigurationRequest } from './code-executor.types';
+export type MessageFromDeveloperType = 'bundleData' | 'response';
+interface BaseMessageFromDeveloper {
+    type: MessageFromDeveloperType;
+    payload?: unknown;
+    requestId: string;
+}
+export type MessageFromDeveloper = ResponseMessageFromDeveloper;
+export interface ResponseMessageFromDeveloper extends BaseMessageFromDeveloper {
+    type: 'response';
+    payload: any;
+}
+export type MessageToDeveloperType = 'executeFunction' | 'localBackendData';
+interface BaseMessageToDeveloper {
+    type: MessageToDeveloperType;
+    payload: unknown;
+    requestId: string;
+}
+export type MessageToDeveloper = ExecuteFunctionMessageToDeveloper | LocalBackendDataMessageToDeveloper;
+export interface ExecuteFunctionMessageToDeveloper extends BaseMessageToDeveloper {
+    type: 'executeFunction';
+    payload: BatchRunConfigurationRequest;
+}
+export interface LocalBackendDataMessageToDeveloper extends BaseMessageToDeveloper {
+    type: 'localBackendData';
+    payload: {
+        secrets: RunSecrets;
+    };
+}
+export {};

package/dist/internal-common/src/local-dev.types.js

@@ -1,3 +1,3 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
 //# sourceMappingURL=local-dev.types.js.map

package/dist/internal-common/src/observability-publisher.d.ts

@@ -1,9 +1,9 @@
-import { LogEvent, MetricEvent } from './observability.types';
-export declare abstract class ObservabilityPublisher {
-    abstract reportMetric(event: Omit<MetricEvent, 'timestamp'>): Promise<void>;
-    abstract reportLog(event: LogEvent): Promise<void>;
-    timeFunction(event: Omit<MetricEvent, 'type' | 'timestamp' | 'value'>, timedFunction: () => Promise<any>): Promise<any>;
-    measureMetric<T extends (...args: any[]) => any>(func: T, name: string, tags: Record<string, string> | undefined, isExposedToUser: boolean, reportTime: boolean): ReturnType<T>;
-    reportTimeMetric(name: string, isExposedToUser: boolean, startTime: number, tags: Record<string, string>): void;
-    reportCountMetric(name: string, isExposedToUser: boolean, tags: Record<string, string>, success: boolean): void;
-}
+import { LogEvent, MetricEvent } from './observability.types';
+export declare abstract class ObservabilityPublisher {
+    abstract reportMetric(event: Omit<MetricEvent, 'timestamp'>): Promise<void>;
+    abstract reportLog(event: LogEvent): Promise<void>;
+    timeFunction(event: Omit<MetricEvent, 'type' | 'timestamp' | 'value'>, timedFunction: () => Promise<any>): Promise<any>;
+    measureMetric<T extends (...args: any[]) => any>(func: T, name: string, tags: Record<string, string> | undefined, isExposedToUser: boolean, reportTime: boolean): ReturnType<T>;
+    reportTimeMetric(name: string, isExposedToUser: boolean, startTime: number, tags: Record<string, string>): void;
+    reportCountMetric(name: string, isExposedToUser: boolean, tags: Record<string, string>, success: boolean): void;
+}

package/dist/internal-common/src/observability-publisher.js

@@ -1,59 +1,59 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ObservabilityPublisher = void 0;
-class ObservabilityPublisher {
-    async timeFunction(event, timedFunction) {
-        const startTime = Date.now();
-        const result = await timedFunction();
-        this.reportMetric(Object.assign(Object.assign({}, event), { type: 'gauge', value: Date.now() - startTime })).then();
-        return result;
-    }
-    measureMetric(func, name, tags = {}, isExposedToUser, reportTime) {
-        try {
-            const startTime = Date.now();
-            const result = func();
-            if (result instanceof Promise) {
-                return result
-                    .then((r) => {
-                    if (reportTime)
-                        this.reportTimeMetric(name, isExposedToUser, startTime, tags);
-                    this.reportCountMetric(name, isExposedToUser, tags, true);
-                    return r;
-                })
-                    .catch((e) => {
-                    this.reportCountMetric(name, isExposedToUser, tags, false);
-                    throw e;
-                });
-            }
-            if (reportTime)
-                this.reportTimeMetric(name, isExposedToUser, startTime, tags);
-            this.reportCountMetric(name, isExposedToUser, tags, true);
-            return result;
-        }
-        catch (e) {
-            this.reportCountMetric(name, isExposedToUser, tags, false);
-            throw e;
-        }
-    }
-    reportTimeMetric(name, isExposedToUser, startTime, tags) {
-        this.reportMetric({
-            name: `${name}_time`,
-            isExposedToUser,
-            tags,
-            value: Date.now() - startTime,
-            type: 'gauge',
-        }).then();
-    }
-    reportCountMetric(name, isExposedToUser, tags, success) {
-        tags = Object.assign(Object.assign({}, tags), { success: success ? 'true' : 'false' });
-        this.reportMetric({
-            name: `${name}_count`,
-            isExposedToUser,
-            tags,
-            value: 1,
-            type: 'count',
-        }).then();
-    }
-}
-exports.ObservabilityPublisher = ObservabilityPublisher;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ObservabilityPublisher = void 0;
+class ObservabilityPublisher {
+    async timeFunction(event, timedFunction) {
+        const startTime = Date.now();
+        const result = await timedFunction();
+        this.reportMetric(Object.assign(Object.assign({}, event), { type: 'gauge', value: Date.now() - startTime })).then();
+        return result;
+    }
+    measureMetric(func, name, tags = {}, isExposedToUser, reportTime) {
+        try {
+            const startTime = Date.now();
+            const result = func();
+            if (result instanceof Promise) {
+                return result
+                    .then((r) => {
+                    if (reportTime)
+                        this.reportTimeMetric(name, isExposedToUser, startTime, tags);
+                    this.reportCountMetric(name, isExposedToUser, tags, true);
+                    return r;
+                })
+                    .catch((e) => {
+                    this.reportCountMetric(name, isExposedToUser, tags, false);
+                    throw e;
+                });
+            }
+            if (reportTime)
+                this.reportTimeMetric(name, isExposedToUser, startTime, tags);
+            this.reportCountMetric(name, isExposedToUser, tags, true);
+            return result;
+        }
+        catch (e) {
+            this.reportCountMetric(name, isExposedToUser, tags, false);
+            throw e;
+        }
+    }
+    reportTimeMetric(name, isExposedToUser, startTime, tags) {
+        this.reportMetric({
+            name: `${name}_time`,
+            isExposedToUser,
+            tags,
+            value: Date.now() - startTime,
+            type: 'gauge',
+        }).then();
+    }
+    reportCountMetric(name, isExposedToUser, tags, success) {
+        tags = Object.assign(Object.assign({}, tags), { success: success ? 'true' : 'false' });
+        this.reportMetric({
+            name: `${name}_count`,
+            isExposedToUser,
+            tags,
+            value: 1,
+            type: 'count',
+        }).then();
+    }
+}
+exports.ObservabilityPublisher = ObservabilityPublisher;
 //# sourceMappingURL=observability-publisher.js.map