add index matching test

Author: pengpeng-lu

fdb-relational-core/src/test/java/com/apple/foundationdb/relational/recordlayer/query/StandardQueryTests.java

@@ -964,39 +964,39 @@ void testNotDistinctFrom() throws Exception {
                 Assertions.assertEquals(1, statement.executeInsert("STUDENT", row1), "Incorrect insertion count");
                 Assertions.assertEquals(1, statement.executeInsert("STUDENT", row2), "Incorrect insertion count");
 
-                Assertions.assertTrue(statement.execute("SELECT * from STUDENT f WHERE score is not distinct from null"), "Did not return a result set from a select statement!");
+                Assertions.assertTrue(statement.execute("SELECT * from STUDENT WHERE score is not distinct from null"), "Did not return a result set from a select statement!");
                 try (final RelationalResultSet resultSet = statement.getResultSet()) {
                     ResultSetAssert.assertThat(resultSet)
                             .hasNextRow()
                             .hasColumn("NAME", "Bob")
                             .hasNoNextRow();
                 }
 
-                Assertions.assertTrue(statement.execute("SELECT * from STUDENT f WHERE score is not distinct from 100"), "Did not return a result set from a select statement!");
+                Assertions.assertTrue(statement.execute("SELECT * from STUDENT WHERE score is not distinct from 100"), "Did not return a result set from a select statement!");
                 try (final RelationalResultSet resultSet = statement.getResultSet()) {
                     ResultSetAssert.assertThat(resultSet)
                             .hasNextRow()
                             .hasColumn("NAME", "Alice")
                             .hasNoNextRow();
                 }
 
-                Assertions.assertTrue(statement.execute("SELECT * from STUDENT f WHERE null is not distinct from score"), "Did not return a result set from a select statement!");
+                Assertions.assertTrue(statement.execute("SELECT * from STUDENT WHERE null is not distinct from score"), "Did not return a result set from a select statement!");
                 try (final RelationalResultSet resultSet = statement.getResultSet()) {
                     ResultSetAssert.assertThat(resultSet)
                             .hasNextRow()
                             .hasColumn("NAME", "Bob")
                             .hasNoNextRow();
                 }
 
-                Assertions.assertTrue(statement.execute("SELECT * from STUDENT f WHERE 100 is not distinct from score"), "Did not return a result set from a select statement!");
+                Assertions.assertTrue(statement.execute("SELECT * from STUDENT WHERE 100 is not distinct from score"), "Did not return a result set from a select statement!");
                 try (final RelationalResultSet resultSet = statement.getResultSet()) {
                     ResultSetAssert.assertThat(resultSet)
                             .hasNextRow()
                             .hasColumn("NAME", "Alice")
                             .hasNoNextRow();
                 }
 
-                Assertions.assertTrue(statement.execute("SELECT * from STUDENT f WHERE 100 is not distinct from 100"), "Did not return a result set from a select statement!");
+                Assertions.assertTrue(statement.execute("SELECT * from STUDENT WHERE 100 is not distinct from 100"), "Did not return a result set from a select statement!");
                 try (final RelationalResultSet resultSet = statement.getResultSet()) {
                     ResultSetAssert.assertThat(resultSet)
                             .hasNextRow()
@@ -1006,7 +1006,7 @@ void testNotDistinctFrom() throws Exception {
                             .hasNoNextRow();
                 }
 
-                Assertions.assertTrue(statement.execute("SELECT * from STUDENT f WHERE null is not distinct from null"), "Did not return a result set from a select statement!");
+                Assertions.assertTrue(statement.execute("SELECT * from STUDENT WHERE null is not distinct from null"), "Did not return a result set from a select statement!");
                 try (final RelationalResultSet resultSet = statement.getResultSet()) {
                     ResultSetAssert.assertThat(resultSet)
                             .hasNextRow()
@@ -1019,6 +1019,71 @@ void testNotDistinctFrom() throws Exception {
         }
     }
 
+    @Test
+    void testExplainDistinctFrom() throws Exception {
+        final String schemaTemplate = "CREATE TYPE AS STRUCT contact_detail(phone_number string, address string) " +
+                "CREATE TABLE STUDENT(id bigint, name string, contact contact_detail, score bigint, primary key(id)) " +
+                "CREATE index score_idx as select score from STUDENT";
+        try (var ddl = Ddl.builder().database(URI.create("/TEST/QT")).relationalExtension(relationalExtension).schemaTemplate(schemaTemplate).build()) {
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where score is distinct from null")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("ISCAN\\(SCORE_IDX \\(\\[null\\],>\\) \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where score is distinct from 100")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("COVERING\\(SCORE_IDX <,> -> \\[ID: KEY\\[.*], SCORE: KEY\\[.*]\\]\\) \\| FILTER _\\.SCORE IS_DISTINCT_FROM promote\\(@.* AS LONG\\) \\| FETCH \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where null is distinct from score")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("ISCAN\\(SCORE_IDX \\(\\[null\\],>\\) \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where 100 is distinct from score")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("COVERING\\(SCORE_IDX <,> -> \\[ID: KEY\\[.*], SCORE: KEY\\[.*]\\]\\) \\| FILTER _\\.SCORE IS_DISTINCT_FROM promote\\(@.* AS LONG\\) \\| FETCH \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where score is not distinct from null")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("ISCAN\\(SCORE_IDX \\[\\[null\\],\\[null\\]\\]\\) \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where score is not distinct from 100")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("COVERING\\(SCORE_IDX <,> -> \\[ID: KEY\\[.*], SCORE: KEY\\[.*]\\]\\) \\| FILTER _\\.SCORE NOT_DISTINCT_FROM promote\\(@.* AS LONG\\) \\| FETCH \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where null is not distinct from score")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("ISCAN\\(SCORE_IDX \\[\\[null\\],\\[null\\]\\]\\) \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+            try (var statement = ddl.setSchemaAndGetConnection().createStatement()) {
+                try (final RelationalResultSet resultSet = statement.executeQuery("EXPLAIN SELECT name FROM STUDENT where 100 is not distinct from score")) {
+                    resultSet.next();
+                    String plan = resultSet.getString(1);
+                    assertThat(plan).matches("COVERING\\(SCORE_IDX <,> -> \\[ID: KEY\\[.*], SCORE: KEY\\[.*]\\]\\) \\| FILTER _\\.SCORE NOT_DISTINCT_FROM promote\\(@.* AS LONG\\) \\| FETCH \\| MAP \\(_\\.NAME AS NAME\\)");
+                }
+            }
+        }
+    }
+
     @Test
     void testBitmap() throws Exception {
         final String query = "SELECT BITMAP_CONSTRUCT_AGG(BITMAP_BIT_POSITION(uid)) as bitmap, category, BITMAP_BUCKET_OFFSET(uid) as offset FROM T1\n" +