321 lines
9.9 KiB
Go
321 lines
9.9 KiB
Go
package query_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/v2/influxql/query"
|
|
"github.com/influxdata/influxql"
|
|
)
|
|
|
|
func TestRewriteStatement(t *testing.T) {
|
|
tests := []struct {
|
|
stmt string
|
|
s string
|
|
}{
|
|
{
|
|
stmt: `SHOW FIELD KEYS`,
|
|
s: `SELECT fieldKey, fieldType FROM _fieldKeys`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS ON db0`,
|
|
s: `SELECT fieldKey, fieldType FROM db0.._fieldKeys`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS FROM cpu`,
|
|
s: `SELECT fieldKey, fieldType FROM _fieldKeys WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS ON db0 FROM cpu`,
|
|
s: `SELECT fieldKey, fieldType FROM db0.._fieldKeys WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS FROM /c.*/`,
|
|
s: `SELECT fieldKey, fieldType FROM _fieldKeys WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS ON db0 FROM /c.*/`,
|
|
s: `SELECT fieldKey, fieldType FROM db0.._fieldKeys WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS FROM mydb.myrp2.cpu`,
|
|
s: `SELECT fieldKey, fieldType FROM mydb.myrp2._fieldKeys WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS ON db0 FROM mydb.myrp2.cpu`,
|
|
s: `SELECT fieldKey, fieldType FROM mydb.myrp2._fieldKeys WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS FROM mydb.myrp2./c.*/`,
|
|
s: `SELECT fieldKey, fieldType FROM mydb.myrp2._fieldKeys WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW FIELD KEYS ON db0 FROM mydb.myrp2./c.*/`,
|
|
s: `SELECT fieldKey, fieldType FROM mydb.myrp2._fieldKeys WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES`,
|
|
s: `SELECT "key" FROM _series`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0`,
|
|
s: `SELECT "key" FROM db0.._series`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM cpu`,
|
|
s: `SELECT "key" FROM _series WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 FROM cpu`,
|
|
s: `SELECT "key" FROM db0.._series WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM mydb.myrp1.cpu`,
|
|
s: `SELECT "key" FROM mydb.myrp1._series WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 FROM mydb.myrp1.cpu`,
|
|
s: `SELECT "key" FROM mydb.myrp1._series WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM mydb.myrp1./c.*/`,
|
|
s: `SELECT "key" FROM mydb.myrp1._series WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM mydb.myrp1./c.*/ WHERE region = 'uswest'`,
|
|
s: `SELECT "key" FROM mydb.myrp1._series WHERE (_name =~ /c.*/) AND (region = 'uswest')`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 FROM mydb.myrp1./c.*/`,
|
|
s: `SELECT "key" FROM mydb.myrp1._series WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM /.+/ WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM db0../.+/ WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM cpu WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM cpu WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 FROM cpu WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM db0..cpu WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM mydb.myrp1.cpu WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM mydb.myrp1.cpu WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 FROM mydb.myrp1.cpu WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM mydb.myrp1.cpu WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM mydb.myrp1./c.*/ WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM mydb.myrp1./c.*/ WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES FROM mydb.myrp1./c.*/ WHERE region = 'uswest' AND time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM mydb.myrp1./c.*/ WHERE region = 'uswest' AND time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES ON db0 FROM mydb.myrp1./c.*/ WHERE time > 0`,
|
|
s: `SELECT _seriesKey AS "key" FROM mydb.myrp1./c.*/ WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES CARDINALITY FROM m`,
|
|
s: `SELECT count(distinct(_seriesKey)) AS count FROM m`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES EXACT CARDINALITY`,
|
|
s: `SELECT count(distinct(_seriesKey)) AS count FROM /.+/`,
|
|
},
|
|
{
|
|
stmt: `SHOW SERIES EXACT CARDINALITY FROM m`,
|
|
s: `SELECT count(distinct(_seriesKey)) AS count FROM m`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS`,
|
|
s: `SHOW TAG KEYS`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0`,
|
|
s: `SHOW TAG KEYS ON db0`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM cpu`,
|
|
s: `SHOW TAG KEYS WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM cpu`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM /c.*/`,
|
|
s: `SHOW TAG KEYS WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM /c.*/`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM cpu WHERE region = 'uswest'`,
|
|
s: `SHOW TAG KEYS WHERE (_name = 'cpu') AND (region = 'uswest')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM cpu WHERE region = 'uswest'`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name = 'cpu') AND (region = 'uswest')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM mydb.myrp1.cpu`,
|
|
s: `SHOW TAG KEYS WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM mydb.myrp1.cpu`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE _name = 'cpu'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM mydb.myrp1./c.*/`,
|
|
s: `SHOW TAG KEYS WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM mydb.myrp1./c.*/`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE _name =~ /c.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM mydb.myrp1.cpu WHERE region = 'uswest'`,
|
|
s: `SHOW TAG KEYS WHERE (_name = 'cpu') AND (region = 'uswest')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM mydb.myrp1.cpu WHERE region = 'uswest'`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name = 'cpu') AND (region = 'uswest')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS WHERE time > 0`,
|
|
s: `SHOW TAG KEYS WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 WHERE time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE time > 0`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM cpu WHERE time > 0`,
|
|
s: `SHOW TAG KEYS WHERE (_name = 'cpu') AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM cpu WHERE time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name = 'cpu') AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM /c.*/ WHERE time > 0`,
|
|
s: `SHOW TAG KEYS WHERE (_name =~ /c.*/) AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM /c.*/ WHERE time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name =~ /c.*/) AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM cpu WHERE region = 'uswest' AND time > 0`,
|
|
s: `SHOW TAG KEYS WHERE (_name = 'cpu') AND (region = 'uswest' AND time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM cpu WHERE region = 'uswest' AND time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name = 'cpu') AND (region = 'uswest' AND time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM mydb.myrp1.cpu WHERE time > 0`,
|
|
s: `SHOW TAG KEYS WHERE (_name = 'cpu') AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM mydb.myrp1.cpu WHERE time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name = 'cpu') AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM mydb.myrp1./c.*/ WHERE time > 0`,
|
|
s: `SHOW TAG KEYS WHERE (_name =~ /c.*/) AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM mydb.myrp1./c.*/ WHERE time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name =~ /c.*/) AND (time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS FROM mydb.myrp1.cpu WHERE region = 'uswest' AND time > 0`,
|
|
s: `SHOW TAG KEYS WHERE (_name = 'cpu') AND (region = 'uswest' AND time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG KEYS ON db0 FROM mydb.myrp1.cpu WHERE region = 'uswest' AND time > 0`,
|
|
s: `SHOW TAG KEYS ON db0 WHERE (_name = 'cpu') AND (region = 'uswest' AND time > 0)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY = "region"`,
|
|
s: `SHOW TAG VALUES WITH KEY = region WHERE _tagKey = 'region'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY = "region" WHERE "region" = 'uswest'`,
|
|
s: `SHOW TAG VALUES WITH KEY = region WHERE (region = 'uswest') AND (_tagKey = 'region')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY IN ("region", "server") WHERE "platform" = 'cloud'`,
|
|
s: `SHOW TAG VALUES WITH KEY IN (region, server) WHERE (platform = 'cloud') AND (_tagKey = 'region' OR _tagKey = 'server')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY = "region" WHERE "region" = 'uswest' AND time > 0`,
|
|
s: `SHOW TAG VALUES WITH KEY = region WHERE (region = 'uswest' AND time > 0) AND (_tagKey = 'region')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY = "region" ON db0`,
|
|
s: `SHOW TAG VALUES WITH KEY = region WHERE _tagKey = 'region'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES FROM cpu WITH KEY = "region"`,
|
|
s: `SHOW TAG VALUES WITH KEY = region WHERE (_name = 'cpu') AND (_tagKey = 'region')`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY != "region"`,
|
|
s: `SHOW TAG VALUES WITH KEY != region WHERE _tagKey != 'region'`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY =~ /re.*/`,
|
|
s: `SHOW TAG VALUES WITH KEY =~ /re.*/ WHERE _tagKey =~ /re.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY =~ /re.*/ WHERE time > 0`,
|
|
s: `SHOW TAG VALUES WITH KEY =~ /re.*/ WHERE (time > 0) AND (_tagKey =~ /re.*/)`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY !~ /re.*/`,
|
|
s: `SHOW TAG VALUES WITH KEY !~ /re.*/ WHERE _tagKey !~ /re.*/`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY !~ /re.*/ LIMIT 1`,
|
|
s: `SHOW TAG VALUES WITH KEY !~ /re.*/ WHERE _tagKey !~ /re.*/ LIMIT 1`,
|
|
},
|
|
{
|
|
stmt: `SHOW TAG VALUES WITH KEY !~ /re.*/ OFFSET 2`,
|
|
s: `SHOW TAG VALUES WITH KEY !~ /re.*/ WHERE _tagKey !~ /re.*/ OFFSET 2`,
|
|
},
|
|
{
|
|
stmt: `SELECT value FROM cpu`,
|
|
s: `SELECT value FROM cpu`,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.stmt, func(t *testing.T) {
|
|
stmt, err := influxql.ParseStatement(test.stmt)
|
|
if err != nil {
|
|
t.Errorf("error parsing statement: %s", err)
|
|
} else {
|
|
stmt, err = query.RewriteStatement(stmt)
|
|
if err != nil {
|
|
t.Errorf("error rewriting statement: %s", err)
|
|
} else if s := stmt.String(); s != test.s {
|
|
t.Errorf("error rendering string. expected %s, actual: %s", test.s, s)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|