@or-sdk/druid 1.1.0-beta.1311.0

package/dist/cjs/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":""}

package/dist/esm/Druid.js

@@ -0,0 +1,49 @@
+import { Base } from '@or-sdk/base';
+import { SERVICE_KEY, V3_ROUTE_PATH } from './constants';
+import { convertQueryResultsToPreview } from './result-converter';
+export class Druid extends Base {
+    constructor(params) {
+        super({
+            token: params.token,
+            serviceUrl: `${params.gatewayUrl}/${params.providersAccountId}`,
+            serviceKey: SERVICE_KEY,
+        });
+    }
+    fetchMetadata() {
+        return this.callApiV2({
+            method: 'GET',
+            route: V3_ROUTE_PATH,
+            params: {
+                includeLookups: true,
+            },
+        });
+    }
+    convertSQLQuery(query, isRaw = false) {
+        const data = {
+            query,
+            type: 'ExplainSQLPlan',
+            isRaw,
+        };
+        return this.callApiV2({
+            method: 'POST',
+            data,
+            route: V3_ROUTE_PATH,
+        });
+    }
+    runNativeQuery(query, options = {}) {
+        const data = {
+            query,
+        };
+        const promise = this.callApiV2({
+            method: 'POST',
+            data,
+            route: V3_ROUTE_PATH,
+        });
+        if (typeof options === 'boolean' && options) {
+            return promise;
+        }
+        const includeTimestamp = typeof options.includeTimestamp === 'boolean' ? options.includeTimestamp : false;
+        return promise.then((r) => convertQueryResultsToPreview(r, includeTimestamp));
+    }
+}
+//# sourceMappingURL=Druid.js.map

package/dist/esm/Druid.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Druid.js","sourceRoot":"","sources":["../../src/Druid.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,cAAc,CAAC;AAEpC,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,4BAA4B,EAAE,MAAM,oBAAoB,CAAC;AAgBlE,MAAM,OAAO,KAAM,SAAQ,IAAI;IAU7B,YAAY,MAAmB;QAC7B,KAAK,CAAC;YACJ,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,UAAU,EAAE,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,kBAAkB,EAAE;YAC/D,UAAU,EAAE,WAAW;SACxB,CAAC,CAAC;IACL,CAAC;IASM,aAAa;QAClB,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,MAAM,EAAE,KAAK;YACb,KAAK,EAAE,aAAa;YACpB,MAAM,EAAE;gBACN,cAAc,EAAE,IAAI;aACrB;SACF,CAAC,CAAC;IACL,CAAC;IAaM,eAAe,CAAI,KAAa,EAAE,KAAK,GAAG,KAAK;QACpD,MAAM,IAAI,GAAG;YACX,KAAK;YACL,IAAI,EAAE,gBAAgB;YACtB,KAAK;SACN,CAAC;QAEF,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,MAAM,EAAE,MAAM;YACd,IAAI;YACJ,KAAK,EAAE,aAAa;SACrB,CAAC,CAAC;IACL,CAAC;IAQM,cAAc,CAAC,KAA8B,EAAE,UAAiC,EAAG;QACxF,MAAM,IAAI,GAAG;YACX,KAAK;SACN,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;YAC7B,MAAM,EAAE,MAAM;YACd,IAAI;YACJ,KAAK,EAAE,aAAa;SACrB,CAAC,CAAC;QACH,IAAI,OAAO,OAAO,KAAK,SAAS,IAAI,OAAO,EAAE;YAC3C,OAAO,OAAO,CAAC;SAChB;QACD,MAAM,gBAAgB,GAAG,OAAO,OAAO,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC;QAC1G,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,4BAA4B,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAChF,CAAC;CACF"}

package/dist/esm/__tests__/Druid.spec.js

@@ -0,0 +1,62 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { afterAll, afterEach, beforeAll, describe, expect, it } from 'vitest';
+import { rest } from 'msw';
+import { setupServer } from 'msw/node';
+import { Druid } from '../';
+import metadata from './fixtures/metadata.json';
+import convertResponse from './fixtures/convert_response.json';
+import runQueryResponse from './fixtures/scan_table_response.json';
+const server = setupServer(rest.get('https://gateway.url/test-account-id/druid/query/v3', (req, res, ctx) => {
+    if (req.headers.get('authorization') !== 'valid-auth-token') {
+        return res(ctx.status(401));
+    }
+    if (req.url.searchParams.get('includeLookups') !== 'true') {
+        return res(ctx.status(500));
+    }
+    return res(ctx.json(metadata));
+}), rest.post('https://gateway.url/test-account-id/druid/query/v3', (req, res, ctx) => __awaiter(void 0, void 0, void 0, function* () {
+    if (req.headers.get('authorization') !== 'valid-auth-token') {
+        return res(ctx.status(401));
+    }
+    const body = yield req.json();
+    if (body.type === 'ExplainSQLPlan') {
+        return res(ctx.json(convertResponse));
+    }
+    return res(ctx.json(runQueryResponse));
+})));
+const createDruidInstance = (opts = {}) => {
+    return new Druid(Object.assign({ gatewayUrl: 'https://gateway.url', providersAccountId: 'test-account-id', token: 'valid-auth-token' }, opts));
+};
+describe('Druid', () => {
+    beforeAll(() => server.listen());
+    afterEach(() => server.resetHandlers());
+    afterAll(() => server.close());
+    it('should create an instance successfully', () => {
+        const druid = createDruidInstance();
+        expect(druid).toBeInstanceOf(Druid);
+    });
+    it('should fetch metadata properly', () => __awaiter(void 0, void 0, void 0, function* () {
+        const druid = createDruidInstance();
+        const resp = yield druid.fetchMetadata();
+        expect(resp).toEqual(metadata);
+    }));
+    it('should convert sql query to native one properly', () => __awaiter(void 0, void 0, void 0, function* () {
+        const druid = createDruidInstance();
+        const resp = yield druid.convertSQLQuery('select "BotId" from _table');
+        expect(resp).toEqual(convertResponse);
+    }));
+    it('should run native query', () => __awaiter(void 0, void 0, void 0, function* () {
+        const druid = createDruidInstance();
+        const resp = yield druid.runNativeQuery({ queryType: 'scan' }, true);
+        expect(resp).toEqual(runQueryResponse);
+    }));
+});
+//# sourceMappingURL=Druid.spec.js.map

package/dist/esm/__tests__/Druid.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Druid.spec.js","sourceRoot":"","sources":["../../../src/__tests__/Druid.spec.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9E,OAAO,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAC3B,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAEvC,OAAO,EAAE,KAAK,EAAE,MAAM,KAAK,CAAC;AAE5B,OAAO,QAAQ,MAAM,0BAA0B,CAAC;AAChD,OAAO,eAAe,MAAM,kCAAkC,CAAC;AAC/D,OAAO,gBAAgB,MAAM,qCAAqC,CAAC;AAEnE,MAAM,MAAM,GAAG,WAAW,CACxB,IAAI,CAAC,GAAG,CAAC,oDAAoD,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;IAC/E,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,kBAAkB,EAAE;QAC3D,OAAO,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;KAC7B;IACD,IAAI,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,gBAAgB,CAAC,KAAK,MAAM,EAAE;QACzD,OAAO,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;KAC7B;IACD,OAAO,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;AACjC,CAAC,CAAC,EACF,IAAI,CAAC,IAAI,CAAC,oDAAoD,EAAE,CAAO,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;IACtF,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,kBAAkB,EAAE;QAC3D,OAAO,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;KAC7B;IACD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;IAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,gBAAgB,EAAE;QAClC,OAAO,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;KACvC;IACD,OAAO,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;AACzC,CAAC,CAAA,CAAC,CACH,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAAC,OAA6B,EAAE,EAAE,EAAE;IAC9D,OAAO,IAAI,KAAK,iBACd,UAAU,EAAE,qBAAqB,EACjC,kBAAkB,EAAE,iBAAiB,EACrC,KAAK,EAAE,kBAAkB,IACtB,IAAI,EACP,CAAC;AACL,CAAC,CAAC;AAEF,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;IACrB,SAAS,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACjC,SAAS,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;IACxC,QAAQ,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;IAE/B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,mBAAmB,EAAE,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAS,EAAE;QAC9C,MAAM,KAAK,GAAG,mBAAmB,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,aAAa,EAAE,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC,CAAA,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAS,EAAE;QAC/D,MAAM,KAAK,GAAG,mBAAmB,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,4BAA4B,CAAC,CAAC;QACvE,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACxC,CAAC,CAAA,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAS,EAAE;QACvC,MAAM,KAAK,GAAG,mBAAmB,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;QACrE,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACzC,CAAC,CAAA,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/esm/__tests__/fixtures/convert_response.json

@@ -0,0 +1,105 @@
+{
+    "query": {
+        "queryType": "scan",
+        "dataSource": {
+            "type": "table",
+            "name": "staging_re_rollup_provider_ExecutionTime"
+        },
+        "intervals": {
+            "type": "intervals",
+            "intervals": [
+                "-146136543-09-08T08:23:32.096Z/146140482-04-24T15:36:27.903Z"
+            ]
+        },
+        "resultFormat": "compactedList",
+        "limit": 1000,
+        "columns": [
+            "AccountId",
+            "BotId",
+            "FlowId",
+            "Type",
+            "__time",
+            "count",
+            "sum_AudioDuration15Sec",
+            "sum_AudioDurationMin",
+            "sum_AudioDurationSec",
+            "sum_CharacterCount",
+            "sum_ExecutionDurationMs",
+            "sum_MemoryMb",
+            "sum_Millisecond100Seg",
+            "sum_Millisecond200Seg",
+            "sum_MillisecondTotal"
+        ],
+        "legacy": false,
+        "context": {
+            "queryId": "3ab8c67f-b348-4d0f-9210-ad3ccfa74b22",
+            "sqlOuterLimit": 1000,
+            "sqlQueryId": "3ab8c67f-b348-4d0f-9210-ad3ccfa74b22"
+        },
+        "granularity": {
+            "type": "all"
+        }
+    },
+    "signature": [
+        {
+            "name": "__time",
+            "type": "LONG"
+        },
+        {
+            "name": "AccountId",
+            "type": "STRING"
+        },
+        {
+            "name": "BotId",
+            "type": "STRING"
+        },
+        {
+            "name": "FlowId",
+            "type": "STRING"
+        },
+        {
+            "name": "Type",
+            "type": "STRING"
+        },
+        {
+            "name": "count",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_AudioDuration15Sec",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_AudioDurationMin",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_AudioDurationSec",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_CharacterCount",
+            "type": "DOUBLE"
+        },
+        {
+            "name": "sum_ExecutionDurationMs",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_MemoryMb",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_Millisecond100Seg",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_Millisecond200Seg",
+            "type": "LONG"
+        },
+        {
+            "name": "sum_MillisecondTotal",
+            "type": "LONG"
+        }
+    ]
+}

package/dist/esm/__tests__/fixtures/group_by_table_response.json

@@ -0,0 +1,146 @@
+[
+    {
+        "version": "v1",
+        "timestamp": "2023-05-15T12:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 21
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-15T12:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 21
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-15T13:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 4
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-15T13:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 4
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-15T14:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 7
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-15T14:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 7
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T05:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 5
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T05:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 5
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T07:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 25
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T07:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 25
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T08:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 52
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T08:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 52
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T09:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 21
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T09:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 20
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T10:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 7
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T10:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 7
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T11:00:00.000Z",
+        "event": {
+            "Type": "ProviderLambda",
+            "total_usage": 29
+        }
+    },
+    {
+        "version": "v1",
+        "timestamp": "2023-05-16T11:00:00.000Z",
+        "event": {
+            "Type": "Read Druid",
+            "total_usage": 29
+        }
+    }
+]