mirror of https://gitee.com/bigwinds/arangodb
275 lines
13 KiB
JavaScript
275 lines
13 KiB
JavaScript
/*jshint globalstrict:false, strict:false, maxlen: 500 */
|
|
/*global assertEqual, assertNotEqual, AQL_EXPLAIN, AQL_EXECUTE */
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief tests for optimizer rules
|
|
///
|
|
/// @file
|
|
///
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2010-2012 triagens GmbH, Cologne, Germany
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// you may not use this file except in compliance with the License.
|
|
/// You may obtain a copy of the License at
|
|
///
|
|
/// http://www.apache.org/licenses/LICENSE-2.0
|
|
///
|
|
/// Unless required by applicable law or agreed to in writing, software
|
|
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
/// See the License for the specific language governing permissions and
|
|
/// limitations under the License.
|
|
///
|
|
/// Copyright holder is triAGENS GmbH, Cologne, Germany
|
|
///
|
|
/// @author Jan Steemann
|
|
/// @author Copyright 2012, triAGENS GmbH, Cologne, Germany
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var jsunity = require("jsunity");
|
|
var helper = require("@arangodb/aql-helper");
|
|
var db = require("@arangodb").db;
|
|
const isCluster = require("@arangodb/cluster").isCluster();
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test suite
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
function optimizerRuleTestSuite () {
|
|
var ruleName = "move-calculations-down";
|
|
|
|
// various choices to control the optimizer:
|
|
var paramNone = { optimizer: { rules: [ "-all" ] } };
|
|
var paramEnabled = { optimizer: { rules: [ "-all", "+" + ruleName ] } };
|
|
var paramDisabled = { optimizer: { rules: [ "+all", "-" + ruleName ] } };
|
|
|
|
var c;
|
|
var cn = "UnitTestsAhuacatlCalculation";
|
|
|
|
return {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief set up
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
setUp : function () {
|
|
db._drop(cn);
|
|
c = db._create(cn);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief tear down
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
tearDown : function () {
|
|
db._drop(cn);
|
|
c = null;
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test that rule has no effect when explicitly disabled
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testRuleDisabled : function () {
|
|
var queries = [
|
|
"FOR i IN 1..10 LET result = i + 1 FILTER i < 5 RETURN result",
|
|
"FOR i IN 1..10 LET a = i + 1 LET b = i + 2 FILTER i < 2 LET c = b FILTER i < 10 RETURN c",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 2 RETURN result",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) LIMIT 1 RETURN result",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) SORT i RETURN result",
|
|
"FOR i IN 1..10 LET result = PUSH(i, i + 1) LET x = (FOR j IN result RETURN j) RETURN x",
|
|
];
|
|
|
|
queries.forEach(function(query) {
|
|
var result = AQL_EXPLAIN(query, { }, paramNone);
|
|
assertEqual([ ], result.plan.rules, query);
|
|
});
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test that rule has no effect
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testRuleNoEffect : function () {
|
|
var queries = [
|
|
"LET a = PASSTHRU(39) FOR i IN 1..10 RETURN a",
|
|
"FOR i IN 1..10 RETURN i",
|
|
"FOR i IN 1..10 FILTER i < 10 RETURN i",
|
|
"FOR i IN 1..10 FILTER i < 10 FILTER i + 1 < 10 RETURN i",
|
|
"FOR i IN 1..10 FILTER i < 10 FILTER i + 1 < 10 LIMIT 2 RETURN i",
|
|
"FOR i IN 1..10 LET result = i < 5 FILTER result RETURN i",
|
|
"FOR i IN 1..10 LET result = i + 1 SORT result RETURN i",
|
|
"FOR i IN 1..10 LET result = i + 1 LET test = (FOR j IN 1..result RETURN j) RETURN test",
|
|
"FOR i IN 1..10 LIMIT 2 LET result = i < 5 RETURN i",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) FOR j IN 1..2 RETURN j",
|
|
"FOR i IN 1..10 FILTER i < 10 LET result = IS_STRING(i) FOR j IN 1..2 RETURN j",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) COLLECT r = result RETURN r",
|
|
"FOR i IN 1..10 LET result = MAX(i) FILTER result < 3 RETURN result",
|
|
"FOR i IN 1..10 LET result = RAND() FILTER i < 10 RETURN result"
|
|
];
|
|
|
|
queries.forEach(function(query) {
|
|
var result = AQL_EXPLAIN(query, { }, paramEnabled);
|
|
assertEqual([ ], result.plan.rules, query);
|
|
});
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test that rule has an effect
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testRuleHasEffect : function () {
|
|
var queries = [
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 10 RETURN result",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 10 FILTER i > 2 RETURN result",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 2 LIMIT 1 RETURN result",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) LIMIT 1 RETURN result",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) LET test = result + 1 LIMIT 1 RETURN test",
|
|
"FOR i IN 1..10 LET result = IS_STRING(i) SORT i RETURN result",
|
|
"FOR i IN 1..10 LET result = i + 1 LET test = (FOR j IN 1..2 RETURN j) RETURN result IN test",
|
|
"FOR i IN 1..10 LET v = i * 2 LIMIT 2 LET result = v < 5 RETURN v"
|
|
];
|
|
|
|
queries.forEach(function(query) {
|
|
var result = AQL_EXPLAIN(query, { }, paramEnabled);
|
|
assertEqual([ ruleName ], result.plan.rules, query);
|
|
});
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test generated plans
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPlans : function () {
|
|
var plans = [
|
|
[ "FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 10 RETURN result", [ "SingletonNode", "CalculationNode", "EnumerateListNode", "CalculationNode", "FilterNode", "CalculationNode", "ReturnNode" ] ],
|
|
[ "FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 10 FILTER i > 2 RETURN result", [ "SingletonNode","CalculationNode","EnumerateListNode","CalculationNode","FilterNode","CalculationNode","FilterNode","CalculationNode","ReturnNode" ] ],
|
|
[ "FOR i IN 1..10 LET result = IS_STRING(i) FILTER i < 2 LIMIT 1 RETURN result", [ "SingletonNode","CalculationNode","EnumerateListNode","CalculationNode","FilterNode","LimitNode","CalculationNode","ReturnNode" ] ],
|
|
[ "FOR i IN 1..10 LET result = IS_STRING(i) LIMIT 1 RETURN result", [ "SingletonNode","CalculationNode","EnumerateListNode","LimitNode","CalculationNode","ReturnNode" ] ],
|
|
[ "FOR i IN 1..10 LET result = IS_STRING(i) LET test = result + 1 LIMIT 1 RETURN test", [ "SingletonNode","CalculationNode","EnumerateListNode","LimitNode","CalculationNode","CalculationNode","ReturnNode" ] ] ,
|
|
[ "FOR i IN 1..10 LET result = IS_STRING(i) SORT i RETURN result", [ "SingletonNode","CalculationNode","EnumerateListNode","SortNode","CalculationNode","ReturnNode" ] ],
|
|
[ "FOR i IN 1..10 LET v = i * 2 LIMIT 2 LET result = v < 5 RETURN v", [ "SingletonNode","CalculationNode","EnumerateListNode","LimitNode","CalculationNode","CalculationNode","ReturnNode" ] ]
|
|
];
|
|
|
|
plans.forEach(function(plan) {
|
|
var result = AQL_EXPLAIN(plan[0], { }, paramEnabled);
|
|
assertEqual([ ruleName ], result.plan.rules, plan[0]);
|
|
assertEqual(plan[1], helper.getCompactPlan(result).map(function(node) { return node.type; }), plan[0]);
|
|
});
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test results
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testResults : function () {
|
|
var queries = [
|
|
[ "FOR i IN 1..10 LET a = i + 1 FILTER i < 4 RETURN a", [ 2, 3, 4 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 FILTER i < 7 FILTER i > 1 RETURN a", [ 3, 4, 5, 6, 7 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 LIMIT 4 RETURN a", [ 2, 3, 4, 5 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 LET b = a + 1 FILTER i < 3 RETURN b", [ 3, 4 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 LET b = a + 1 LIMIT 4 RETURN b", [ 3, 4, 5, 6 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 LET b = a + 1 FILTER i < 5 LIMIT 4 RETURN b", [ 3, 4, 5, 6 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 LET x = (FOR j IN 1..i RETURN j) RETURN a - 1 IN x ? 1 : 0", [ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 ] ],
|
|
[ "FOR i IN 1..10 LET a = i + 1 LET x = (FOR j IN 1..i RETURN j) RETURN a IN x ? 1 : 0", [ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ] ],
|
|
[ "FOR i IN 1..10 LET a = MAX(1..i) FILTER i > 4 RETURN a", [ 5, 6, 7, 8, 9, 10 ] ],
|
|
[ "FOR i IN 1..10 LET a = SUM(1..i) FILTER i == 3 RETURN a", [ 6 ] ],
|
|
[ "FOR i IN 1..10 LET a = SUM(1..i) LET b = MIN(1..i) FILTER i == 3 RETURN [ a, b ]", [ [ 6, 1 ] ] ],
|
|
[ "FOR i IN 1..10 LET a = SUM(1..i) LIMIT 3 RETURN a", [ 1, 3, 6 ] ],
|
|
[ "FOR i IN 1..10 LET a = SUM(1..i) LIMIT 3 FILTER a > 2 RETURN a", [ 3, 6 ] ],
|
|
[ "FOR i IN 1..10 LET a = SUM(1..i) LIMIT 5 FILTER i > 2 RETURN a", [ 6, 10, 15 ] ]
|
|
];
|
|
|
|
queries.forEach(function(query) {
|
|
var planDisabled = AQL_EXPLAIN(query[0], { }, paramDisabled);
|
|
var planEnabled = AQL_EXPLAIN(query[0], { }, paramEnabled);
|
|
|
|
var resultDisabled = AQL_EXECUTE(query[0], { }, paramDisabled);
|
|
var resultEnabled = AQL_EXECUTE(query[0], { }, paramEnabled);
|
|
|
|
assertEqual(-1, planDisabled.plan.rules.indexOf(ruleName), query[0]);
|
|
assertNotEqual(-1, planEnabled.plan.rules.indexOf(ruleName), query[0]);
|
|
|
|
assertEqual(resultDisabled.json, query[1], query[0]);
|
|
assertEqual(resultEnabled.json, query[1], query[0]);
|
|
});
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test results
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testCollection1 : function () {
|
|
var expected = [ ];
|
|
for (var i = 0; i < 100; ++i) {
|
|
c.save({ _key: "test" + i, value: i });
|
|
expected.push("test" + i + "-" + i);
|
|
}
|
|
|
|
var query = "FOR i IN " + cn + " LET result = CONCAT(i._key, '-', i.value) SORT i.value RETURN result";
|
|
var planDisabled = AQL_EXPLAIN(query, { }, paramDisabled);
|
|
var planEnabled = AQL_EXPLAIN(query, { }, paramEnabled);
|
|
|
|
var resultDisabled = AQL_EXECUTE(query, { }, paramDisabled);
|
|
var resultEnabled = AQL_EXECUTE(query, { }, paramEnabled);
|
|
|
|
assertEqual(-1, planDisabled.plan.rules.indexOf(ruleName), query[0]);
|
|
assertNotEqual(-1, planEnabled.plan.rules.indexOf(ruleName), query[0]);
|
|
|
|
assertEqual(resultDisabled.json, expected, query[0]);
|
|
assertEqual(resultEnabled.json, expected, query[0]);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test results
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testCollection2 : function () {
|
|
var expected = [ "test43-43", "test44-44" ];
|
|
for (var i = 0; i < 100; ++i) {
|
|
c.save({ _key: "test" + i, value: i });
|
|
}
|
|
|
|
var query = "FOR i IN " + cn + " LET result = CONCAT(i._key, '-', i.value) FILTER i.value > 42 SORT i.value LIMIT 2 RETURN result";
|
|
var planDisabled = AQL_EXPLAIN(query, { }, paramDisabled);
|
|
var planEnabled = AQL_EXPLAIN(query, { }, paramEnabled);
|
|
|
|
var resultDisabled = AQL_EXECUTE(query, { }, paramDisabled);
|
|
var resultEnabled = AQL_EXECUTE(query, { }, paramEnabled);
|
|
|
|
assertEqual(-1, planDisabled.plan.rules.indexOf(ruleName), query);
|
|
assertNotEqual(-1, planEnabled.plan.rules.indexOf(ruleName), query);
|
|
|
|
assertEqual(resultDisabled.json, expected, query);
|
|
assertEqual(resultEnabled.json, expected, query);
|
|
},
|
|
|
|
testCollection3 : function () {
|
|
var expected = [ "test0-0", "test1-1" ];
|
|
for (var i = 0; i < 100; ++i) {
|
|
c.save({ _key: "test" + i, value: i });
|
|
}
|
|
|
|
var query = "FOR i IN " + cn + " LET result = CONCAT(i._key, '-', i.value) SORT i.value LIMIT 2 RETURN result";
|
|
var planDisabled = AQL_EXPLAIN(query, { }, paramDisabled);
|
|
var planEnabled = AQL_EXPLAIN(query, { }, paramEnabled);
|
|
|
|
var resultDisabled = AQL_EXECUTE(query, { }, paramDisabled);
|
|
var resultEnabled = AQL_EXECUTE(query, { }, paramEnabled);
|
|
|
|
assertEqual(-1, planDisabled.plan.rules.indexOf(ruleName), query);
|
|
assertNotEqual(-1, planEnabled.plan.rules.indexOf(ruleName), query);
|
|
|
|
assertEqual(resultDisabled.json, expected, query);
|
|
assertEqual(resultEnabled.json, expected, query);
|
|
}
|
|
|
|
};
|
|
}
|
|
|
|
jsunity.run(optimizerRuleTestSuite);
|
|
|
|
return jsunity.done();
|