import _ from 'lodash'; import ResponseParser from './response_parser'; export class MssqlDatasource { id: any; name: any; responseParser: ResponseParser; interval: string; /** @ngInject */ constructor(instanceSettings, private backendSrv, private $q, private templateSrv, private timeSrv) { this.name = instanceSettings.name; this.id = instanceSettings.id; this.responseParser = new ResponseParser(this.$q); this.interval = (instanceSettings.jsonData || {}).timeInterval || '1m'; } interpolateVariable(value, variable) { if (typeof value === 'string') { if (variable.multi || variable.includeAll) { return "'" + value.replace(/'/g, `''`) + "'"; } else { return value; } } if (typeof value === 'number') { return value; } const quotedValues = _.map(value, val => { if (typeof value === 'number') { return value; } return "'" + val.replace(/'/g, `''`) + "'"; }); return quotedValues.join(','); } query(options) { const queries = _.filter(options.targets, item => { return item.hide !== true; }).map(item => { return { refId: item.refId, intervalMs: options.intervalMs, maxDataPoints: options.maxDataPoints, datasourceId: this.id, rawSql: this.templateSrv.replace(item.rawSql, options.scopedVars, this.interpolateVariable), format: item.format, }; }); if (queries.length === 0) { return this.$q.when({ data: [] }); } return this.backendSrv .datasourceRequest({ url: '/api/tsdb/query', method: 'POST', data: { from: options.range.from.valueOf().toString(), to: options.range.to.valueOf().toString(), queries: queries, }, }) .then(this.responseParser.processQueryResult); } annotationQuery(options) { if (!options.annotation.rawQuery) { return this.$q.reject({ message: 'Query missing in annotation definition' }); } const query = { refId: options.annotation.name, datasourceId: this.id, rawSql: this.templateSrv.replace(options.annotation.rawQuery, options.scopedVars, this.interpolateVariable), format: 'table', }; return this.backendSrv .datasourceRequest({ url: '/api/tsdb/query', method: 'POST', data: { from: options.range.from.valueOf().toString(), to: options.range.to.valueOf().toString(), queries: [query], }, }) .then(data => this.responseParser.transformAnnotationResponse(options, data)); } metricFindQuery(query, optionalOptions) { let refId = 'tempvar'; if (optionalOptions && optionalOptions.variable && optionalOptions.variable.name) { refId = optionalOptions.variable.name; } const interpolatedQuery = { refId: refId, datasourceId: this.id, rawSql: this.templateSrv.replace(query, {}, this.interpolateVariable), format: 'table', }; const range = this.timeSrv.timeRange(); const data = { queries: [interpolatedQuery], from: range.from.valueOf().toString(), to: range.to.valueOf().toString(), }; return this.backendSrv .datasourceRequest({ url: '/api/tsdb/query', method: 'POST', data: data, }) .then(data => this.responseParser.parseMetricFindQueryResult(refId, data)); } testDatasource() { return this.backendSrv .datasourceRequest({ url: '/api/tsdb/query', method: 'POST', data: { from: '5m', to: 'now', queries: [ { refId: 'A', intervalMs: 1, maxDataPoints: 1, datasourceId: this.id, rawSql: 'SELECT 1', format: 'table', }, ], }, }) .then(res => { return { status: 'success', message: 'Database Connection OK' }; }) .catch(err => { console.log(err); if (err.data && err.data.message) { return { status: 'error', message: err.data.message }; } else { return { status: 'error', message: err.status }; } }); } }