import { AbstractLabelOperator, AbstractLabelMatcher, LanguageProvider, AbstractQuery } from '@grafana/data'; import { ElasticsearchDataQuery } from './dataquery.gen'; import { ElasticDatasource } from './datasource'; export default class ElasticsearchLanguageProvider extends LanguageProvider { declare request: (url: string, params?: any) => Promise<any>; declare start: () => Promise<any[]>; datasource: ElasticDatasource; constructor(datasource: ElasticDatasource) { super(); this.datasource = datasource; } /** * Queries are transformed to an ES Logs query since it's the behaviour most users expect. **/ importFromAbstractQuery(abstractQuery: AbstractQuery): ElasticsearchDataQuery { return { metrics: [ { id: '1', type: 'logs', }, ], query: this.getElasticsearchDataQuery(abstractQuery.labelMatchers), refId: abstractQuery.refId, }; } getElasticsearchDataQuery(labels: AbstractLabelMatcher[]): string { return labels .map((label) => { switch (label.operator) { case AbstractLabelOperator.Equal: { return label.name + ':"' + label.value + '"'; } case AbstractLabelOperator.NotEqual: { return '-' + label.name + ':"' + label.value + '"'; } case AbstractLabelOperator.EqualRegEx: { return label.name + ':/' + label.value + '/'; } case AbstractLabelOperator.NotEqualRegEx: { return '-' + label.name + ':/' + label.value + '/'; } } }) .join(' AND '); } }