@ -1,6 +1,8 @@
package mysql
import (
"fmt"
"math/rand"
"testing"
"time"
@ -14,8 +16,12 @@ import (
// To run this test, remove the Skip from SkipConvey
// and set up a MySQL db named grafana_tests and a user/password grafana/password
// Use the docker/blocks/mysql_tests/docker-compose.yaml to spin up a
// preconfigured MySQL server suitable for running these tests.
// Thers's also a dashboard.json in same directory that you can import to Grafana
// once you've created a datasource for the test server/database.
func TestMySQL ( t * testing . T ) {
SkipConvey ( "MySQL" , t , func ( ) {
Convey ( "MySQL" , t , func ( ) {
x := InitMySQLTestDB ( t )
endpoint := & MysqlQueryEndpoint {
@ -29,110 +35,621 @@ func TestMySQL(t *testing.T) {
sess := x . NewSession ( )
defer sess . Close ( )
sql := "CREATE TABLE `mysql_types` ("
sql += "`atinyint` tinyint(1) NOT NULL,"
sql += "`avarchar` varchar(3) NOT NULL,"
sql += "`achar` char(3),"
sql += "`amediumint` mediumint NOT NULL,"
sql += "`asmallint` smallint NOT NULL,"
sql += "`abigint` bigint NOT NULL,"
sql += "`aint` int(11) NOT NULL,"
sql += "`adouble` double(10,2),"
sql += "`anewdecimal` decimal(10,2),"
sql += "`afloat` float(10,2) NOT NULL,"
sql += "`atimestamp` timestamp NOT NULL,"
sql += "`adatetime` datetime NOT NULL,"
sql += "`atime` time NOT NULL,"
// sql += "`ayear` year," // Crashes xorm when running cleandb
sql += "`abit` bit(1),"
sql += "`atinytext` tinytext,"
sql += "`atinyblob` tinyblob,"
sql += "`atext` text,"
sql += "`ablob` blob,"
sql += "`amediumtext` mediumtext,"
sql += "`amediumblob` mediumblob,"
sql += "`alongtext` longtext,"
sql += "`alongblob` longblob,"
sql += "`aenum` enum('val1', 'val2'),"
sql += "`aset` set('a', 'b', 'c', 'd'),"
sql += "`adate` date,"
sql += "`time_sec` datetime(6),"
sql += "`aintnull` int(11),"
sql += "`afloatnull` float(10,2),"
sql += "`avarcharnull` varchar(3),"
sql += "`adecimalnull` decimal(10,2)"
sql += ") ENGINE=InnoDB DEFAULT CHARSET=latin1;"
_ , err := sess . Exec ( sql )
So ( err , ShouldBeNil )
sql = "INSERT INTO `mysql_types` "
sql += "(`atinyint`, `avarchar`, `achar`, `amediumint`, `asmallint`, `abigint`, `aint`, `adouble`, "
sql += "`anewdecimal`, `afloat`, `adatetime`, `atimestamp`, `atime`, `abit`, `atinytext`, "
sql += "`atinyblob`, `atext`, `ablob`, `amediumtext`, `amediumblob`, `alongtext`, `alongblob`, "
sql += "`aenum`, `aset`, `adate`, `time_sec`) "
sql += "VALUES(1, 'abc', 'def', 1, 10, 100, 1420070400, 1.11, "
sql += "2.22, 3.33, now(), current_timestamp(), '11:11:11', 1, 'tinytext', "
sql += "'tinyblob', 'text', 'blob', 'mediumtext', 'mediumblob', 'longtext', 'longblob', "
sql += "'val2', 'a,b', curdate(), '2018-01-01 00:01:01.123456');"
_ , err = sess . Exec ( sql )
So ( err , ShouldBeNil )
Convey ( "Query with Table format should map MySQL column types to Go types" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : "SELECT * FROM mysql_types" ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
fromStart := time . Date ( 2018 , 3 , 15 , 13 , 0 , 0 , 0 , time . Local )
Convey ( "Given a table with different native data types" , func ( ) {
if exists , err := sess . IsTableExist ( "mysql_types" ) ; err != nil || exists {
So ( err , ShouldBeNil )
sess . DropTable ( "mysql_types" )
}
resp , err := endpoint . Query ( nil , nil , query )
queryResult := resp . Results [ "A" ]
sql := "CREATE TABLE `mysql_types` ("
sql += "`atinyint` tinyint(1) NOT NULL,"
sql += "`avarchar` varchar(3) NOT NULL,"
sql += "`achar` char(3),"
sql += "`amediumint` mediumint NOT NULL,"
sql += "`asmallint` smallint NOT NULL,"
sql += "`abigint` bigint NOT NULL,"
sql += "`aint` int(11) NOT NULL,"
sql += "`adouble` double(10,2),"
sql += "`anewdecimal` decimal(10,2),"
sql += "`afloat` float(10,2) NOT NULL,"
sql += "`atimestamp` timestamp NOT NULL,"
sql += "`adatetime` datetime NOT NULL,"
sql += "`atime` time NOT NULL,"
sql += "`ayear` year," // Crashes xorm when running cleandb
sql += "`abit` bit(1),"
sql += "`atinytext` tinytext,"
sql += "`atinyblob` tinyblob,"
sql += "`atext` text,"
sql += "`ablob` blob,"
sql += "`amediumtext` mediumtext,"
sql += "`amediumblob` mediumblob,"
sql += "`alongtext` longtext,"
sql += "`alongblob` longblob,"
sql += "`aenum` enum('val1', 'val2'),"
sql += "`aset` set('a', 'b', 'c', 'd'),"
sql += "`adate` date,"
sql += "`time_sec` datetime(6),"
sql += "`aintnull` int(11),"
sql += "`afloatnull` float(10,2),"
sql += "`avarcharnull` varchar(3),"
sql += "`adecimalnull` decimal(10,2)"
sql += ") ENGINE=InnoDB DEFAULT CHARSET=latin1;"
_ , err := sess . Exec ( sql )
So ( err , ShouldBeNil )
sql = "INSERT INTO `mysql_types` "
sql += "(`atinyint`, `avarchar`, `achar`, `amediumint`, `asmallint`, `abigint`, `aint`, `adouble`, "
sql += "`anewdecimal`, `afloat`, `adatetime`, `atimestamp`, `atime`, `ayear`, `abit`, `atinytext`, "
sql += "`atinyblob`, `atext`, `ablob`, `amediumtext`, `amediumblob`, `alongtext`, `alongblob`, "
sql += "`aenum`, `aset`, `adate`, `time_sec`) "
sql += "VALUES(1, 'abc', 'def', 1, 10, 100, 1420070400, 1.11, "
sql += "2.22, 3.33, now(), current_timestamp(), '11:11:11', '2018', 1, 'tinytext', "
sql += "'tinyblob', 'text', 'blob', 'mediumtext', 'mediumblob', 'longtext', 'longblob', "
sql += "'val2', 'a,b', curdate(), '2018-01-01 00:01:01.123456');"
_ , err = sess . Exec ( sql )
So ( err , ShouldBeNil )
column := queryResult . Tables [ 0 ] . Rows [ 0 ]
So ( * column [ 0 ] . ( * int8 ) , ShouldEqual , 1 )
So ( column [ 1 ] . ( string ) , ShouldEqual , "abc" )
So ( column [ 2 ] . ( string ) , ShouldEqual , "def" )
So ( * column [ 3 ] . ( * int32 ) , ShouldEqual , 1 )
So ( * column [ 4 ] . ( * int16 ) , ShouldEqual , 10 )
So ( * column [ 5 ] . ( * int64 ) , ShouldEqual , 100 )
So ( * column [ 6 ] . ( * int32 ) , ShouldEqual , 1420070400 )
So ( column [ 7 ] . ( float64 ) , ShouldEqual , 1.11 )
So ( column [ 8 ] . ( float64 ) , ShouldEqual , 2.22 )
So ( * column [ 9 ] . ( * float32 ) , ShouldEqual , 3.33 )
_ , offset := time . Now ( ) . Zone ( )
So ( column [ 10 ] . ( time . Time ) , ShouldHappenWithin , time . Duration ( 10 * time . Second ) , time . Now ( ) . Add ( time . Duration ( offset ) * time . Second ) )
So ( column [ 11 ] . ( time . Time ) , ShouldHappenWithin , time . Duration ( 10 * time . Second ) , time . Now ( ) . Add ( time . Duration ( offset ) * time . Second ) )
So ( column [ 12 ] . ( string ) , ShouldEqual , "11:11:11" )
So ( * column [ 13 ] . ( * [ ] byte ) , ShouldHaveSameTypeAs , [ ] byte { 1 } )
So ( column [ 14 ] . ( string ) , ShouldEqual , "tinytext" )
So ( column [ 15 ] . ( string ) , ShouldEqual , "tinyblob" )
So ( column [ 16 ] . ( string ) , ShouldEqual , "text" )
So ( column [ 17 ] . ( string ) , ShouldEqual , "blob" )
So ( column [ 18 ] . ( string ) , ShouldEqual , "mediumtext" )
So ( column [ 19 ] . ( string ) , ShouldEqual , "mediumblob" )
So ( column [ 20 ] . ( string ) , ShouldEqual , "longtext" )
So ( column [ 21 ] . ( string ) , ShouldEqual , "longblob" )
So ( column [ 22 ] . ( string ) , ShouldEqual , "val2" )
So ( column [ 23 ] . ( string ) , ShouldEqual , "a,b" )
So ( column [ 24 ] . ( time . Time ) . Format ( "2006-01-02T00:00:00Z" ) , ShouldEqual , time . Now ( ) . Format ( "2006-01-02T00:00:00Z" ) )
So ( column [ 25 ] . ( float64 ) , ShouldEqual , 1514764861 )
So ( column [ 26 ] , ShouldEqual , nil )
So ( column [ 27 ] , ShouldEqual , nil )
So ( column [ 28 ] , ShouldEqual , "" )
So ( column [ 29 ] , ShouldEqual , nil )
Convey ( "Query with Table format should map MySQL column types to Go types" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : "SELECT * FROM mysql_types" ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
column := queryResult . Tables [ 0 ] . Rows [ 0 ]
So ( * column [ 0 ] . ( * int8 ) , ShouldEqual , 1 )
So ( column [ 1 ] . ( string ) , ShouldEqual , "abc" )
So ( column [ 2 ] . ( string ) , ShouldEqual , "def" )
So ( * column [ 3 ] . ( * int32 ) , ShouldEqual , 1 )
So ( * column [ 4 ] . ( * int16 ) , ShouldEqual , 10 )
So ( * column [ 5 ] . ( * int64 ) , ShouldEqual , 100 )
So ( * column [ 6 ] . ( * int32 ) , ShouldEqual , 1420070400 )
So ( column [ 7 ] . ( float64 ) , ShouldEqual , 1.11 )
So ( column [ 8 ] . ( float64 ) , ShouldEqual , 2.22 )
So ( * column [ 9 ] . ( * float32 ) , ShouldEqual , 3.33 )
_ , offset := time . Now ( ) . Zone ( )
So ( column [ 10 ] . ( time . Time ) , ShouldHappenWithin , time . Duration ( 10 * time . Second ) , time . Now ( ) . Add ( time . Duration ( offset ) * time . Second ) )
So ( column [ 11 ] . ( time . Time ) , ShouldHappenWithin , time . Duration ( 10 * time . Second ) , time . Now ( ) . Add ( time . Duration ( offset ) * time . Second ) )
So ( column [ 12 ] . ( string ) , ShouldEqual , "11:11:11" )
So ( column [ 13 ] . ( int64 ) , ShouldEqual , 2018 )
So ( * column [ 14 ] . ( * [ ] byte ) , ShouldHaveSameTypeAs , [ ] byte { 1 } )
So ( column [ 15 ] . ( string ) , ShouldEqual , "tinytext" )
So ( column [ 16 ] . ( string ) , ShouldEqual , "tinyblob" )
So ( column [ 17 ] . ( string ) , ShouldEqual , "text" )
So ( column [ 18 ] . ( string ) , ShouldEqual , "blob" )
So ( column [ 19 ] . ( string ) , ShouldEqual , "mediumtext" )
So ( column [ 20 ] . ( string ) , ShouldEqual , "mediumblob" )
So ( column [ 21 ] . ( string ) , ShouldEqual , "longtext" )
So ( column [ 22 ] . ( string ) , ShouldEqual , "longblob" )
So ( column [ 23 ] . ( string ) , ShouldEqual , "val2" )
So ( column [ 24 ] . ( string ) , ShouldEqual , "a,b" )
So ( column [ 25 ] . ( time . Time ) . Format ( "2006-01-02T00:00:00Z" ) , ShouldEqual , time . Now ( ) . Format ( "2006-01-02T00:00:00Z" ) )
So ( column [ 26 ] . ( float64 ) , ShouldEqual , float64 ( 1514764861000 ) )
So ( column [ 27 ] , ShouldEqual , nil )
So ( column [ 28 ] , ShouldEqual , nil )
So ( column [ 29 ] , ShouldEqual , "" )
So ( column [ 30 ] , ShouldEqual , nil )
} )
} )
Convey ( "Given a table with metrics that lacks data for some series " , func ( ) {
type metric struct {
Time time . Time
Value int64
}
if exist , err := sess . IsTableExist ( metric { } ) ; err != nil || exist {
So ( err , ShouldBeNil )
sess . DropTable ( metric { } )
}
err := sess . CreateTable ( metric { } )
So ( err , ShouldBeNil )
series := [ ] * metric { }
firstRange := genTimeRangeByInterval ( fromStart , 10 * time . Minute , 10 * time . Second )
secondRange := genTimeRangeByInterval ( fromStart . Add ( 20 * time . Minute ) , 10 * time . Minute , 10 * time . Second )
for _ , t := range firstRange {
series = append ( series , & metric {
Time : t ,
Value : 15 ,
} )
}
for _ , t := range secondRange {
series = append ( series , & metric {
Time : t ,
Value : 20 ,
} )
}
for _ , s := range series {
_ , err = sess . Insert ( s )
So ( err , ShouldBeNil )
}
Convey ( "When doing a metric query using timeGroup" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : "SELECT $__timeGroup(time, '5m') as time_sec, avg(value) as value FROM metric GROUP BY 1 ORDER BY 1" ,
"format" : "time_series" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
points := queryResult . Series [ 0 ] . Points
So ( len ( points ) , ShouldEqual , 6 )
dt := fromStart
for i := 0 ; i < 3 ; i ++ {
aValue := points [ i ] [ 0 ] . Float64
aTime := time . Unix ( int64 ( points [ i ] [ 1 ] . Float64 ) / 1000 , 0 )
So ( aValue , ShouldEqual , 15 )
So ( aTime , ShouldEqual , dt )
dt = dt . Add ( 5 * time . Minute )
}
// adjust for 5 minute gap
dt = dt . Add ( 5 * time . Minute )
for i := 3 ; i < 6 ; i ++ {
aValue := points [ i ] [ 0 ] . Float64
aTime := time . Unix ( int64 ( points [ i ] [ 1 ] . Float64 ) / 1000 , 0 )
So ( aValue , ShouldEqual , 20 )
So ( aTime , ShouldEqual , dt )
dt = dt . Add ( 5 * time . Minute )
}
} )
Convey ( "When doing a metric query using timeGroup with NULL fill enabled" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : "SELECT $__timeGroup(time, '5m', NULL) as time_sec, avg(value) as value FROM metric GROUP BY 1 ORDER BY 1" ,
"format" : "time_series" ,
} ) ,
RefId : "A" ,
} ,
} ,
TimeRange : & tsdb . TimeRange {
From : fmt . Sprintf ( "%v" , fromStart . Unix ( ) * 1000 ) ,
To : fmt . Sprintf ( "%v" , fromStart . Add ( 34 * time . Minute ) . Unix ( ) * 1000 ) ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
points := queryResult . Series [ 0 ] . Points
So ( len ( points ) , ShouldEqual , 7 )
dt := fromStart
for i := 0 ; i < 3 ; i ++ {
aValue := points [ i ] [ 0 ] . Float64
aTime := time . Unix ( int64 ( points [ i ] [ 1 ] . Float64 ) / 1000 , 0 )
So ( aValue , ShouldEqual , 15 )
So ( aTime , ShouldEqual , dt )
dt = dt . Add ( 5 * time . Minute )
}
So ( points [ 3 ] [ 0 ] . Valid , ShouldBeFalse )
// adjust for 5 minute gap
dt = dt . Add ( 5 * time . Minute )
for i := 4 ; i < 7 ; i ++ {
aValue := points [ i ] [ 0 ] . Float64
aTime := time . Unix ( int64 ( points [ i ] [ 1 ] . Float64 ) / 1000 , 0 )
So ( aValue , ShouldEqual , 20 )
So ( aTime , ShouldEqual , dt )
dt = dt . Add ( 5 * time . Minute )
}
} )
Convey ( "When doing a metric query using timeGroup with float fill enabled" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : "SELECT $__timeGroup(time, '5m', 1.5) as time_sec, avg(value) as value FROM metric GROUP BY 1 ORDER BY 1" ,
"format" : "time_series" ,
} ) ,
RefId : "A" ,
} ,
} ,
TimeRange : & tsdb . TimeRange {
From : fmt . Sprintf ( "%v" , fromStart . Unix ( ) * 1000 ) ,
To : fmt . Sprintf ( "%v" , fromStart . Add ( 34 * time . Minute ) . Unix ( ) * 1000 ) ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
points := queryResult . Series [ 0 ] . Points
So ( points [ 3 ] [ 0 ] . Float64 , ShouldEqual , 1.5 )
} )
} )
Convey ( "Given a table with metrics having multiple values and measurements" , func ( ) {
type metric_values struct {
Time time . Time
Measurement string
ValueOne int64 ` xorm:"integer 'valueOne'" `
ValueTwo int64 ` xorm:"integer 'valueTwo'" `
}
if exist , err := sess . IsTableExist ( metric_values { } ) ; err != nil || exist {
So ( err , ShouldBeNil )
sess . DropTable ( metric_values { } )
}
err := sess . CreateTable ( metric_values { } )
So ( err , ShouldBeNil )
rand . Seed ( time . Now ( ) . Unix ( ) )
rnd := func ( min , max int64 ) int64 {
return rand . Int63n ( max - min ) + min
}
series := [ ] * metric_values { }
for _ , t := range genTimeRangeByInterval ( fromStart . Add ( - 30 * time . Minute ) , 90 * time . Minute , 5 * time . Minute ) {
series = append ( series , & metric_values {
Time : t ,
Measurement : "Metric A" ,
ValueOne : rnd ( 0 , 100 ) ,
ValueTwo : rnd ( 0 , 100 ) ,
} )
series = append ( series , & metric_values {
Time : t ,
Measurement : "Metric B" ,
ValueOne : rnd ( 0 , 100 ) ,
ValueTwo : rnd ( 0 , 100 ) ,
} )
}
for _ , s := range series {
_ , err := sess . Insert ( s )
So ( err , ShouldBeNil )
}
Convey ( "When doing a metric query grouping by time and select metric column should return correct series" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : ` SELECT $__time(time), CONCAT(measurement, ' - value one') as metric, valueOne FROM metric_values ORDER BY 1 ` ,
"format" : "time_series" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Series ) , ShouldEqual , 2 )
So ( queryResult . Series [ 0 ] . Name , ShouldEqual , "Metric B - value one" )
So ( queryResult . Series [ 1 ] . Name , ShouldEqual , "Metric A - value one" )
} )
Convey ( "When doing a metric query grouping by time should return correct series" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : ` SELECT $__time(time), valueOne, valueTwo FROM metric_values ORDER BY 1 ` ,
"format" : "time_series" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Series ) , ShouldEqual , 2 )
So ( queryResult . Series [ 0 ] . Name , ShouldEqual , "valueOne" )
So ( queryResult . Series [ 1 ] . Name , ShouldEqual , "valueTwo" )
} )
} )
Convey ( "Given a table with event data" , func ( ) {
type event struct {
TimeSec int64
Description string
Tags string
}
if exist , err := sess . IsTableExist ( event { } ) ; err != nil || exist {
So ( err , ShouldBeNil )
sess . DropTable ( event { } )
}
err := sess . CreateTable ( event { } )
So ( err , ShouldBeNil )
events := [ ] * event { }
for _ , t := range genTimeRangeByInterval ( fromStart . Add ( - 20 * time . Minute ) , 60 * time . Minute , 25 * time . Minute ) {
events = append ( events , & event {
TimeSec : t . Unix ( ) ,
Description : "Someone deployed something" ,
Tags : "deploy" ,
} )
events = append ( events , & event {
TimeSec : t . Add ( 5 * time . Minute ) . Unix ( ) ,
Description : "New support ticket registered" ,
Tags : "ticket" ,
} )
}
for _ , e := range events {
_ , err = sess . Insert ( e )
So ( err , ShouldBeNil )
}
Convey ( "When doing an annotation query of deploy events should return expected result" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : ` SELECT time_sec, description as text, tags FROM event WHERE $__unixEpochFilter(time_sec) AND tags='deploy' ORDER BY 1 ASC ` ,
"format" : "table" ,
} ) ,
RefId : "Deploys" ,
} ,
} ,
TimeRange : & tsdb . TimeRange {
From : fmt . Sprintf ( "%v" , fromStart . Add ( - 20 * time . Minute ) . Unix ( ) * 1000 ) ,
To : fmt . Sprintf ( "%v" , fromStart . Add ( 40 * time . Minute ) . Unix ( ) * 1000 ) ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
queryResult := resp . Results [ "Deploys" ]
So ( err , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 3 )
} )
Convey ( "When doing an annotation query of ticket events should return expected result" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : ` SELECT time_sec, description as text, tags FROM event WHERE $__unixEpochFilter(time_sec) AND tags='ticket' ORDER BY 1 ASC ` ,
"format" : "table" ,
} ) ,
RefId : "Tickets" ,
} ,
} ,
TimeRange : & tsdb . TimeRange {
From : fmt . Sprintf ( "%v" , fromStart . Add ( - 20 * time . Minute ) . Unix ( ) * 1000 ) ,
To : fmt . Sprintf ( "%v" , fromStart . Add ( 40 * time . Minute ) . Unix ( ) * 1000 ) ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
queryResult := resp . Results [ "Tickets" ]
So ( err , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 3 )
} )
Convey ( "When doing an annotation query with a time column in datetime format" , func ( ) {
dt := time . Date ( 2018 , 3 , 14 , 21 , 20 , 6 , 0 , time . UTC )
dtFormat := "2006-01-02 15:04:05.999999999"
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : fmt . Sprintf ( ` SELECT
CAST ( ' % s ' as datetime ) as time_sec ,
' message ' as text ,
' tag1 , tag2 ' as tags
` , dt . Format ( dtFormat ) ) ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 1 )
columns := queryResult . Tables [ 0 ] . Rows [ 0 ]
//Should be in milliseconds
So ( columns [ 0 ] . ( float64 ) , ShouldEqual , float64 ( dt . Unix ( ) * 1000 ) )
} )
Convey ( "When doing an annotation query with a time column in epoch second format should return ms" , func ( ) {
dt := time . Date ( 2018 , 3 , 14 , 21 , 20 , 6 , 527e6 , time . UTC )
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : fmt . Sprintf ( ` SELECT
% d as time_sec ,
' message ' as text ,
' tag1 , tag2 ' as tags
` , dt . Unix ( ) ) ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 1 )
columns := queryResult . Tables [ 0 ] . Rows [ 0 ]
//Should be in milliseconds
So ( columns [ 0 ] . ( int64 ) , ShouldEqual , dt . Unix ( ) * 1000 )
} )
Convey ( "When doing an annotation query with a time column in epoch second format (signed integer) should return ms" , func ( ) {
dt := time . Date ( 2018 , 3 , 14 , 21 , 20 , 6 , 0 , time . Local )
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : fmt . Sprintf ( ` SELECT
CAST ( ' % d ' as signed integer ) as time_sec ,
' message ' as text ,
' tag1 , tag2 ' as tags
` , dt . Unix ( ) ) ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 1 )
columns := queryResult . Tables [ 0 ] . Rows [ 0 ]
//Should be in milliseconds
So ( columns [ 0 ] . ( int64 ) , ShouldEqual , int64 ( dt . Unix ( ) * 1000 ) )
} )
Convey ( "When doing an annotation query with a time column in epoch millisecond format should return ms" , func ( ) {
dt := time . Date ( 2018 , 3 , 14 , 21 , 20 , 6 , 527e6 , time . UTC )
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : fmt . Sprintf ( ` SELECT
% d as time_sec ,
' message ' as text ,
' tag1 , tag2 ' as tags
` , dt . Unix ( ) * 1000 ) ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 1 )
columns := queryResult . Tables [ 0 ] . Rows [ 0 ]
//Should be in milliseconds
So ( columns [ 0 ] . ( int64 ) , ShouldEqual , dt . Unix ( ) * 1000 )
} )
Convey ( "When doing an annotation query with a time column holding a unsigned integer null value should return nil" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : ` SELECT
cast ( null as unsigned integer ) as time_sec ,
' message ' as text ,
' tag1 , tag2 ' as tags
` ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 1 )
columns := queryResult . Tables [ 0 ] . Rows [ 0 ]
//Should be in milliseconds
So ( columns [ 0 ] , ShouldBeNil )
} )
Convey ( "When doing an annotation query with a time column holding a DATETIME null value should return nil" , func ( ) {
query := & tsdb . TsdbQuery {
Queries : [ ] * tsdb . Query {
{
Model : simplejson . NewFromAny ( map [ string ] interface { } {
"rawSql" : ` SELECT
cast ( null as DATETIME ) as time_sec ,
' message ' as text ,
' tag1 , tag2 ' as tags
` ,
"format" : "table" ,
} ) ,
RefId : "A" ,
} ,
} ,
}
resp , err := endpoint . Query ( nil , nil , query )
So ( err , ShouldBeNil )
queryResult := resp . Results [ "A" ]
So ( queryResult . Error , ShouldBeNil )
So ( len ( queryResult . Tables [ 0 ] . Rows ) , ShouldEqual , 1 )
columns := queryResult . Tables [ 0 ] . Rows [ 0 ]
//Should be in milliseconds
So ( columns [ 0 ] , ShouldBeNil )
} )
} )
} )
}
func InitMySQLTestDB ( t * testing . T ) * xorm . Engine {
x , err := xorm . NewEngine ( sqlutil . TestDB_Mysql . DriverName , sqlutil . TestDB_Mysql . ConnStr + "&parseTime=true" )
x . DatabaseTZ = time . Local
x . TZLocation = time . Local
// x.ShowSQL()
@ -140,7 +657,18 @@ func InitMySQLTestDB(t *testing.T) *xorm.Engine {
t . Fatalf ( "Failed to init mysql db %v" , err )
}
sqlutil . CleanDB ( x )
return x
}
func genTimeRangeByInterval ( from time . Time , duration time . Duration , interval time . Duration ) [ ] time . Time {
durationSec := int64 ( duration . Seconds ( ) )
intervalSec := int64 ( interval . Seconds ( ) )
timeRange := [ ] time . Time { }
for i := int64 ( 0 ) ; i < durationSec ; i += intervalSec {
timeRange = append ( timeRange , from )
from = from . Add ( time . Duration ( int64 ( time . Second ) * intervalSec ) )
}
return timeRange
}