[refactor](Nereids): compute unique and uniform property respectively (#32908)
This commit is contained in:
谢健
@ -258,12 +258,12 @@ class FunctionalDependenciesTest extends TestWithFeService {
|
||||
@Test
|
||||
void testWindow() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select row_number() over(partition by id) from agg")
|
||||
.analyze("select id, row_number() over(partition by id) from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
System.out.println(plan.getLogicalProperties().getFunctionalDependencies());
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
.getFunctionalDependencies().isUniformAndNotNull(plan.getOutput().get(1)));
|
||||
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select row_number() over(partition by name) from agg where name = '1'")
|
||||
|
||||
@ -0,0 +1,207 @@
|
||||
// Licensed to the Apache Software Foundation (ASF) under one
|
||||
// or more contributor license agreements. See the NOTICE file
|
||||
// distributed with this work for additional information
|
||||
// regarding copyright ownership. The ASF licenses this file
|
||||
// to you 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.
|
||||
|
||||
package org.apache.doris.nereids.properties;
|
||||
|
||||
import org.apache.doris.nereids.trees.expressions.Slot;
|
||||
import org.apache.doris.nereids.trees.expressions.SlotReference;
|
||||
import org.apache.doris.nereids.trees.plans.Plan;
|
||||
import org.apache.doris.nereids.types.IntegerType;
|
||||
import org.apache.doris.nereids.util.PlanChecker;
|
||||
import org.apache.doris.utframe.TestWithFeService;
|
||||
|
||||
import org.junit.jupiter.api.Assertions;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
class UniformTest extends TestWithFeService {
|
||||
Slot slot1 = new SlotReference("1", IntegerType.INSTANCE, false);
|
||||
Slot slot2 = new SlotReference("2", IntegerType.INSTANCE, false);
|
||||
Slot slot3 = new SlotReference("1", IntegerType.INSTANCE, false);
|
||||
Slot slot4 = new SlotReference("1", IntegerType.INSTANCE, false);
|
||||
|
||||
@Override
|
||||
protected void runBeforeAll() throws Exception {
|
||||
createDatabase("test");
|
||||
createTable("create table test.agg (\n"
|
||||
+ "id int not null,\n"
|
||||
+ "name varchar(128) replace not null )\n"
|
||||
+ "AGGREGATE KEY(id)\n"
|
||||
+ "distributed by hash(id) buckets 10\n"
|
||||
+ "properties('replication_num' = '1');");
|
||||
createTable("create table test.uni (\n"
|
||||
+ "id int not null,\n"
|
||||
+ "name varchar(128) not null)\n"
|
||||
+ "UNIQUE KEY(id)\n"
|
||||
+ "distributed by hash(id) buckets 10\n"
|
||||
+ "properties('replication_num' = '1');");
|
||||
connectContext.setDatabase("test");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testAgg() {
|
||||
// group by unique
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select count(id) from agg group by id")
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniform(plan.getOutput().get(0)));
|
||||
|
||||
// propagate uniform
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg where id = 1 group by id ")
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniform(plan.getOutput().get(0)));
|
||||
|
||||
// group by all/uniform
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select sum(id) from agg ")
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniform(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select sum(id) from agg where id = 1 group by id")
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniform(plan.getOutput().get(0)));
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTopNLimit() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg limit 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg order by name limit 1 ")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testSetOp() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg limit 1 except select name from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniform(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg intersect select name from agg limit 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniform(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testFilterHaving() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg where id = 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from uni group by name having name = \"\"")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testGenerate() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg lateral view explode([1,2,3]) tmp1 as e1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testJoin() {
|
||||
// foj doesn't propagate any
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id from agg join uni "
|
||||
+ "on agg.id = uni.id where uni.id = 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUniform(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testOneRowRelation() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniform(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testProject() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id as id2, 1 as id3 from uni where id = 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniformAndNotNull(plan.getOutput().get(1)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testSubQuery() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from (select id from agg where id = 1) t")
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniformAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testWindow() {
|
||||
// partition by uniform
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select row_number() over(partition by id) from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniform(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select rank() over(partition by id) from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniform(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select dense_rank() over(partition by id) from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniform(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,385 @@
|
||||
// Licensed to the Apache Software Foundation (ASF) under one
|
||||
// or more contributor license agreements. See the NOTICE file
|
||||
// distributed with this work for additional information
|
||||
// regarding copyright ownership. The ASF licenses this file
|
||||
// to you 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.
|
||||
|
||||
package org.apache.doris.nereids.properties;
|
||||
|
||||
import org.apache.doris.nereids.trees.expressions.Slot;
|
||||
import org.apache.doris.nereids.trees.expressions.SlotReference;
|
||||
import org.apache.doris.nereids.trees.plans.Plan;
|
||||
import org.apache.doris.nereids.types.IntegerType;
|
||||
import org.apache.doris.nereids.util.PlanChecker;
|
||||
import org.apache.doris.utframe.TestWithFeService;
|
||||
|
||||
import org.junit.jupiter.api.Assertions;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
class UniqueTest extends TestWithFeService {
|
||||
Slot slot1 = new SlotReference("1", IntegerType.INSTANCE, false);
|
||||
Slot slot2 = new SlotReference("2", IntegerType.INSTANCE, false);
|
||||
Slot slot3 = new SlotReference("1", IntegerType.INSTANCE, false);
|
||||
Slot slot4 = new SlotReference("1", IntegerType.INSTANCE, false);
|
||||
|
||||
@Override
|
||||
protected void runBeforeAll() throws Exception {
|
||||
createDatabase("test");
|
||||
createTable("create table test.agg (\n"
|
||||
+ "id int not null,\n"
|
||||
+ "name varchar(128) replace not null )\n"
|
||||
+ "AGGREGATE KEY(id)\n"
|
||||
+ "distributed by hash(id) buckets 10\n"
|
||||
+ "properties('replication_num' = '1');");
|
||||
createTable("create table test.uni (\n"
|
||||
+ "id int not null,\n"
|
||||
+ "name varchar(128) not null)\n"
|
||||
+ "UNIQUE KEY(id)\n"
|
||||
+ "distributed by hash(id) buckets 10\n"
|
||||
+ "properties('replication_num' = '1');");
|
||||
connectContext.setDatabase("test");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testAgg() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg group by name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select sum(id) from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUnique(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id, sum(id), avg(id), max(id), min(id) from agg group by id")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUnique(plan.getOutput().get(0)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUnique(plan.getOutput().get(1)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUnique(plan.getOutput().get(2)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUnique(plan.getOutput().get(3)));
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
void testScan() throws Exception {
|
||||
// test agg key
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
|
||||
// test unique key
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from uni")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
|
||||
// test unique constraint
|
||||
addConstraint("alter table agg add constraint uq unique(name)");
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
dropConstraint("alter table agg drop constraint uq");
|
||||
|
||||
// test primary constraint
|
||||
addConstraint("alter table agg add constraint pk primary key(name)");
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
dropConstraint("alter table agg drop constraint pk");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTopNLimit() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg limit 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg order by name limit 1 ")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testSetOp() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from agg limit 1 except select name from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg intersect select name from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg union all select name from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name, id from agg union select name, id from agg")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUnique(plan.getOutputSet()));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testFilterHaving() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg where id = 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select name from uni group by name having name = \"\"")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testGenerate() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg lateral view explode([1,2,3]) tmp1 as e1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testJoin() {
|
||||
// foj doesn't propagate any
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select * from agg full outer join uni "
|
||||
+ "on agg.id = uni.id")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isEmpty());
|
||||
|
||||
// loj propagate unique when right is unique
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select agg.id, uni.id from agg left outer join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select agg.id, uni.id from agg left outer join uni "
|
||||
+ "on agg.id = uni.id")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(0)));
|
||||
Assertions.assertFalse(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(1)));
|
||||
|
||||
// roj propagate unique when left is unique
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select agg.id, uni.id from agg right outer join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select agg.id, uni.id from agg right outer join uni "
|
||||
+ "on agg.id = uni.id")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(1)));
|
||||
Assertions.assertFalse(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(0)));
|
||||
|
||||
// semi/anti join propagate all
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select agg.id from agg left semi join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select agg.id from agg left anti join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id from agg right semi join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(0)));
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id from agg right anti join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies().isUnique(plan.getOutput().get(0)));
|
||||
|
||||
// inner join propagate unique only when join key is unique
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id, agg.id from agg inner join uni "
|
||||
+ "on agg.id = uni.id")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(1)));
|
||||
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id, agg.id from agg inner join uni "
|
||||
+ "on agg.id = uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id, agg.id from agg inner join uni "
|
||||
+ "on agg.id < uni.id")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id, agg.id from agg inner join uni "
|
||||
+ "on agg.id = uni.id or agg.name > uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select uni.id, agg.id from agg inner join uni "
|
||||
+ "on agg.id = uni.id and agg.name > uni.name")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(1)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testOneRowRelation() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select 1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testProject() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id as id2 from uni")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testRepeat() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg group by GROUPING SETS ((id, name), (id))")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isEmpty());
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg group by rollup (id, name)")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties().getFunctionalDependencies()
|
||||
.isEmpty());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testSubQuery() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from (select id from agg) t")
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testAssertRows() {
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id from agg where id = (select id from uni)")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testWindow() {
|
||||
// partition by uniform
|
||||
Plan plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id, row_number() over(partition by id) from agg where id =1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(1)));
|
||||
|
||||
// partition by None
|
||||
plan = PlanChecker.from(connectContext)
|
||||
.analyze("select id, row_number() over() from agg where id =1")
|
||||
.rewrite()
|
||||
.getPlan();
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(0)));
|
||||
Assertions.assertTrue(plan.getLogicalProperties()
|
||||
.getFunctionalDependencies().isUniqueAndNotNull(plan.getOutput().get(1)));
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user