IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-11 08:05:43 -06:00
Code Issues Packages Projects Releases Wiki Activity

Tempo: Cache autocomplete values for tags (#54622)

Browse Source
This commit is contained in:
Hamas Shafiq 2022-09-05 15:44:44 +01:00 committed by GitHub
parent 78833b4a47
commit 413c7c7d99
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
public/app/plugins/datasource/tempo/traceql/autocomplete.ts
View File

@ -1,3 +1,4 @@
import { SelectableValue } from '@grafana/data';
import type { Monaco, monacoTypes } from '@grafana/ui';
import TempoLanguageProvider from '../language_provider';
@ -29,6 +30,7 @@ export class CompletionProvider implements monacoTypes.languages.CompletionItemP
editor: monacoTypes.editor.IStandaloneCodeEditor | undefined;
private tags: { [tag: string]: Set<string> } = {};
private cachedValues: { [key: string]: Array<SelectableValue<string>> } = {};
provideCompletionItems(
model: monacoTypes.editor.ITextModel,
@ -81,6 +83,18 @@ export class CompletionProvider implements monacoTypes.languages.CompletionItemP
}
}
private async getTagValues(tagName: string): Promise<Array<SelectableValue<string>>> {
let tagValues: Array<SelectableValue<string>> = [];
if (this.cachedValues.hasOwnProperty(tagName)) {
tagValues = this.cachedValues[tagName];
} else {
tagValues = await this.languageProvider.getOptions(tagName);
this.cachedValues[tagName] = tagValues;
}
return tagValues;
}
/**
* Get suggestion based on the situation we are in like whether we should suggest tag names or values.
* @param situation
@ -114,19 +128,18 @@ export class CompletionProvider implements monacoTypes.languages.CompletionItemP
}));
case 'SPANSET_IN_VALUE':
const tagName = this.overrideTagName(situation.tagName);
return await this.languageProvider.getOptions(tagName).then((res) => {
const items: Completion[] = [];
res.forEach((val) => {
if (val?.label) {
items.push({
label: val.label,
insertText: situation.betweenQuotes ? val.label : `"${val.label}"`,
type: 'TAG_VALUE',
});
}
});
return items;
const tagValues = await this.getTagValues(tagName);
const items: Completion[] = [];
tagValues.forEach((val) => {
if (val?.label) {
items.push({
label: val.label,
insertText: situation.betweenQuotes ? val.label : `"${val.label}"`,
type: 'TAG_VALUE',
});
}
});
return items;
case 'SPANSET_AFTER_VALUE':
return CompletionProvider.logicalOps.concat('}').map((key) => ({
label: key,