mirror of https://github.com/grafana/grafana.git
109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package mysql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
|
|
"github.com/go-sql-driver/mysql"
|
|
"github.com/go-xorm/core"
|
|
"github.com/grafana/grafana/pkg/log"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/tsdb"
|
|
)
|
|
|
|
func init() {
|
|
tsdb.RegisterTsdbQueryEndpoint("mysql", newMysqlQueryEndpoint)
|
|
}
|
|
|
|
func newMysqlQueryEndpoint(datasource *models.DataSource) (tsdb.TsdbQueryEndpoint, error) {
|
|
logger := log.New("tsdb.mysql")
|
|
|
|
cnnstr := fmt.Sprintf("%s:%s@%s(%s)/%s?collation=utf8mb4_unicode_ci&parseTime=true&loc=UTC&allowNativePasswords=true",
|
|
datasource.User,
|
|
datasource.Password,
|
|
"tcp",
|
|
datasource.Url,
|
|
datasource.Database,
|
|
)
|
|
logger.Debug("getEngine", "connection", cnnstr)
|
|
|
|
config := tsdb.SqlQueryEndpointConfiguration{
|
|
DriverName: "mysql",
|
|
ConnectionString: cnnstr,
|
|
Datasource: datasource,
|
|
TimeColumnNames: []string{"time", "time_sec"},
|
|
MetricColumnTypes: []string{"CHAR", "VARCHAR", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT"},
|
|
}
|
|
|
|
rowTransformer := mysqlRowTransformer{
|
|
log: logger,
|
|
}
|
|
|
|
return tsdb.NewSqlQueryEndpoint(&config, &rowTransformer, newMysqlMacroEngine(), logger)
|
|
}
|
|
|
|
type mysqlRowTransformer struct {
|
|
log log.Logger
|
|
}
|
|
|
|
func (t *mysqlRowTransformer) Transform(columnTypes []*sql.ColumnType, rows *core.Rows) (tsdb.RowValues, error) {
|
|
values := make([]interface{}, len(columnTypes))
|
|
|
|
for i := range values {
|
|
scanType := columnTypes[i].ScanType()
|
|
values[i] = reflect.New(scanType).Interface()
|
|
|
|
if columnTypes[i].DatabaseTypeName() == "BIT" {
|
|
values[i] = new([]byte)
|
|
}
|
|
}
|
|
|
|
if err := rows.Scan(values...); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i := 0; i < len(columnTypes); i++ {
|
|
typeName := reflect.ValueOf(values[i]).Type().String()
|
|
|
|
switch typeName {
|
|
case "*sql.RawBytes":
|
|
values[i] = string(*values[i].(*sql.RawBytes))
|
|
case "*mysql.NullTime":
|
|
sqlTime := (*values[i].(*mysql.NullTime))
|
|
if sqlTime.Valid {
|
|
values[i] = sqlTime.Time
|
|
} else {
|
|
values[i] = nil
|
|
}
|
|
case "*sql.NullInt64":
|
|
nullInt64 := (*values[i].(*sql.NullInt64))
|
|
if nullInt64.Valid {
|
|
values[i] = nullInt64.Int64
|
|
} else {
|
|
values[i] = nil
|
|
}
|
|
case "*sql.NullFloat64":
|
|
nullFloat64 := (*values[i].(*sql.NullFloat64))
|
|
if nullFloat64.Valid {
|
|
values[i] = nullFloat64.Float64
|
|
} else {
|
|
values[i] = nil
|
|
}
|
|
}
|
|
|
|
if columnTypes[i].DatabaseTypeName() == "DECIMAL" {
|
|
f, err := strconv.ParseFloat(values[i].(string), 64)
|
|
|
|
if err == nil {
|
|
values[i] = f
|
|
} else {
|
|
values[i] = nil
|
|
}
|
|
}
|
|
}
|
|
|
|
return values, nil
|
|
}
|