Function onQueryEvent

extensions/azdata-sanddance/src/extension.ts:37–75

(type: azdata.queryeditor.QueryEvent, document: azdata.queryeditor.QueryDocument, args: any)

    // Issue #6374 created in ADS repository to track this ask
    azdata.queryeditor.registerQueryEventListener({
        async onQueryEvent(type: azdata.queryeditor.QueryEvent, document: azdata.queryeditor.QueryDocument, args: any) {
            if (type === 'visualize') {
                const providerid = document.providerId;
                const provider: azdata.QueryProvider = azdata.dataprotocol.getProvider(providerid, azdata.DataProviderType.QueryProvider);
                const data = await provider.getQueryRows({
                    ownerUri: document.uri,
                    batchIndex: args.batchId,
                    resultSetIndex: args.id,
                    rowsStartIndex: 0,
                    rowsCount: args.rowCount,
                });

                const rows = data.resultSubset.rows;
                const columns = args.columnInfo;
                const rowsCount = args.rowCount;

                // Create Json
                const jsonArray: jsonType[] = [];

                interface jsonType {
                    [key: string]: any
                }

                for (let row = 0; row < rowsCount; row++) {
                    const jsonObject: jsonType = {};
                    for (let col = 0; col < columns.length; col++) {
                        if (!rows[row][col].isNull) {
                            jsonObject[columns[col].columnName] = rows[row][col].displayValue;
                        }
                        // If display value is null, don't do anything for now
                    }
                    jsonArray.push(jsonObject);
                }

                viewInSandDance(() => {
                    return new Promise<string>(resolve => resolve(JSON.stringify(jsonArray)));
                }, document.uri, 'json', context);
            }
        },
    });
}