grafana/public/app/plugins/datasource/tempo/datasource.ts

import {
  ArrayVector,
  DataFrame,
  DataQuery,
  DataQueryRequest,
  DataQueryResponse,
  DataSourceInstanceSettings,
  Field,
  FieldType,
} from '@grafana/data';
import { DataSourceWithBackend } from '@grafana/runtime';
import { Observable } from 'rxjs';
import { map } from 'rxjs/operators';

export type TempoQuery = {
  query: string;
} & DataQuery;

export class TempoDatasource extends DataSourceWithBackend<TempoQuery> {
  constructor(instanceSettings: DataSourceInstanceSettings) {
    super(instanceSettings);
  }

  query(options: DataQueryRequest<TempoQuery>): Observable<DataQueryResponse> {
    return super.query(options).pipe(
      map((response) => {
        if (response.error) {
          return response;
        }

        // We need to parse some of the fields which contain stringified json.
        // Seems like we can't just map the values as the frame we got from backend has some default processing
        // and will stringify the json back when we try to set it. So we create a new field and swap it instead.
        const frame: DataFrame = response.data[0];
        for (const fieldName of ['serviceTags', 'logs', 'tags']) {
          const field = frame.fields.find((f) => f.name === fieldName);
          if (field) {
            const fieldIndex = frame.fields.indexOf(field);
            const values = new ArrayVector();
            const newField: Field = {
              ...field,
              values,
              type: FieldType.other,
            };

            for (let i = 0; i < field.values.length; i++) {
              const value = field.values.get(i);
              values.set(i, value === '' ? undefined : JSON.parse(value));
            }
            frame.fields[fieldIndex] = newField;
          }
        }

        return response;
      })
    );
  }

  async testDatasource(): Promise<any> {
    const response = await super.query({ targets: [{ query: '', refId: 'A' }] } as any).toPromise();

    if (!response.error?.message?.startsWith('failed to get trace')) {
      return { status: 'error', message: 'Data source is not working' };
    }

    return { status: 'success', message: 'Data source is working' };
  }

  getQueryDisplayText(query: TempoQuery) {
    return query.query;
  }
}