server: fix issue with tracking custom functions that return SETOF materialized view (close #5294)

server/src-lib/Hasura/RQL/DDL/Schema/Function.hs

@@ -222,8 +222,8 @@ handleMultipleFunctions qf = \case
     throw400 NotSupported $
     "function " <> qf <<> " is overloaded. Overloaded functions are not supported"
 
-fetchRawFunctioInfo :: MonadTx m => QualifiedFunction -> m RawFunctionInfo
-fetchRawFunctioInfo qf@(QualifiedObject sn fn) =
+fetchRawFunctionInfo :: MonadTx m => QualifiedFunction -> m RawFunctionInfo
+fetchRawFunctionInfo qf@(QualifiedObject sn fn) =
   handleMultipleFunctions qf =<< map (Q.getAltJ . runIdentity) <$> fetchFromDatabase
   where
     fetchFromDatabase = liftTx $

server/src-lib/Hasura/SQL/Types.hs

@@ -195,30 +195,34 @@ instance ToTxt TableName where
 data TableType
   = TTBaseTable
   | TTView
+  | TTMaterializedView
   | TTForeignTable
   | TTLocalTemporary
   deriving (Eq)
 
 tableTyToTxt :: TableType -> T.Text
-tableTyToTxt TTBaseTable      = "BASE TABLE"
-tableTyToTxt TTView           = "VIEW"
-tableTyToTxt TTForeignTable   = "FOREIGN TABLE"
-tableTyToTxt TTLocalTemporary = "LOCAL TEMPORARY"
+tableTyToTxt TTBaseTable        = "BASE TABLE"
+tableTyToTxt TTView             = "VIEW"
+tableTyToTxt TTMaterializedView = "MATERIALIZED VIEW"
+tableTyToTxt TTForeignTable     = "FOREIGN TABLE"
+tableTyToTxt TTLocalTemporary   = "LOCAL TEMPORARY"
 
 instance Show TableType where
   show = T.unpack . tableTyToTxt
 
 instance Q.FromCol TableType where
   fromCol bs = flip Q.fromColHelper bs $ PD.enum $ \case
-    "BASE TABLE"      -> Just TTBaseTable
-    "VIEW"            -> Just TTView
-    "FOREIGN TABLE"   -> Just TTForeignTable
-    "LOCAL TEMPORARY" -> Just TTLocalTemporary
-    _                 -> Nothing
+    "BASE TABLE"        -> Just TTBaseTable
+    "VIEW"              -> Just TTView
+    "MATERIALIZED VIEW" -> Just TTMaterializedView
+    "FOREIGN TABLE"     -> Just TTForeignTable
+    "LOCAL TEMPORARY"   -> Just TTLocalTemporary
+    _                   -> Nothing
 
 isView :: TableType -> Bool
-isView TTView = True
-isView _      = False
+isView TTView             = True
+isView TTMaterializedView = True
+isView _                  = False
 
 newtype ConstraintName
   = ConstraintName { getConstraintTxt :: T.Text }

server/src-rsr/catalog_version.txt

@@ -1 +1 @@
-39
+40

server/src-rsr/initialise.sql

@@ -568,33 +568,42 @@ CREATE VIEW hdb_catalog.hdb_function_info_agg AS (
       (
         SELECT
           e
-          FROM
-              (
+        FROM
+          (
+            SELECT
+              description,
+              has_variadic,
+              function_type,
+              return_type_schema,
+              return_type_name,
+              return_type_type,
+              returns_set,
+              input_arg_types,
+              input_arg_names,
+              default_args,
+              exists(
                 SELECT
-                  description,
-                  has_variadic,
-                  function_type,
-                  return_type_schema,
-                  return_type_name,
-                  return_type_type,
-                  returns_set,
-                  input_arg_types,
-                  input_arg_names,
-                  default_args,
-                  exists(
-                    SELECT
-                      1
-                      FROM
-                          information_schema.tables
-                     WHERE
-                table_schema = return_type_schema
-            AND table_name = return_type_name
-                  ) AS returns_table
-              ) AS e
+                  1
+                FROM
+                  information_schema.tables
+                WHERE
+                  table_schema = return_type_schema
+                  AND table_name = return_type_name
+              )
+              OR exists(
+                SELECT
+                  1
+                FROM
+                  pg_matviews
+                WHERE
+                  schemaname = return_type_schema
+                  AND matviewname = return_type_name
+              ) AS returns_table
+          ) AS e
       )
     ) AS "function_info"
-    FROM
-        hdb_catalog.hdb_function_agg
+  FROM
+    hdb_catalog.hdb_function_agg
 );
 
 CREATE OR REPLACE FUNCTION

server/src-rsr/migrations/39_to_40.sql

@@ -0,0 +1,46 @@
+CREATE
+OR REPLACE VIEW hdb_catalog.hdb_function_info_agg AS (
+    SELECT
+        function_name,
+        function_schema,
+        row_to_json (
+            (
+                SELECT
+                    e
+                FROM
+                    (
+                        SELECT
+                            description,
+                            has_variadic,
+                            function_type,
+                            return_type_schema,
+                            return_type_name,
+                            return_type_type,
+                            returns_set,
+                            input_arg_types,
+                            input_arg_names,
+                            default_args,
+                            exists(
+                                SELECT
+                                    1
+                                FROM
+                                    information_schema.tables
+                                WHERE
+                                    table_schema = return_type_schema
+                                    AND table_name = return_type_name
+                            )
+                            OR exists(
+                                SELECT
+                                    1
+                                FROM
+                                    pg_matviews
+                                WHERE
+                                    schemaname = return_type_schema
+                                    AND matviewname = return_type_name
+                            ) AS returns_table
+                    ) AS e
+            )
+        ) AS "function_info"
+    FROM
+        hdb_catalog.hdb_function_agg
+);

server/src-rsr/migrations/40_to_39.sql

@@ -0,0 +1,37 @@
+CREATE
+OR REPLACE VIEW hdb_catalog.hdb_function_info_agg AS (
+    SELECT
+        function_name,
+        function_schema,
+        row_to_json (
+            (
+                SELECT
+                    e
+                FROM
+                    (
+                        SELECT
+                            description,
+                            has_variadic,
+                            function_type,
+                            return_type_schema,
+                            return_type_name,
+                            return_type_type,
+                            returns_set,
+                            input_arg_types,
+                            input_arg_names,
+                            default_args,
+                            exists(
+                                SELECT
+                                    1
+                                FROM
+                                    information_schema.tables
+                                WHERE
+                                    table_schema = return_type_schema
+                                    AND table_name = return_type_name
+                            ) AS returns_table
+                    ) AS e
+            )
+        ) AS "function_info"
+    FROM
+        hdb_catalog.hdb_function_agg
+);

server/tests-py/queries/graphql_query/functions/query_search_author_mview.yaml

@@ -0,0 +1,18 @@
+description: Custom GraphQL query using search_author_mview function which returns results from a materialized view
+url: /v1/graphql
+status: 200
+response:
+  data:
+    search_author_mview:
+    - first_name: franz
+      last_name: kafka
+query:
+  query: |
+    query {
+      search_author_mview(
+        args: {query: "kafka"}
+      ) {
+        first_name
+        last_name
+      }
+    }

server/tests-py/queries/graphql_query/functions/setup.yaml

@@ -176,3 +176,45 @@ args:
       name: get_session_var
     configuration:
       session_argument: hasura_session
+
+# track & query functions that return SETOF materialized views
+- type: run_sql
+  args:
+    sql: |
+      CREATE TABLE author(
+        id SERIAL PRIMARY KEY,
+        first_name TEXT,
+        last_name TEXT
+      );
+
+      INSERT INTO author(first_name, last_name) VALUES
+        ('enid', 'blyton'),
+        ('ruskin', 'bond'),
+        ('franz', 'kafka');
+
+      CREATE MATERIALIZED VIEW author_mat_view AS
+        SELECT * FROM author;
+
+      CREATE FUNCTION search_author_mview(query text)
+      RETURNS SETOF author_mat_view AS $FUNCTION$
+          SELECT * FROM author_mat_view WHERE
+            first_name = query OR
+            last_name = query
+      $FUNCTION$ LANGUAGE sql STABLE;
+
+- type: track_table
+  args:
+    name: author
+    schema: public
+
+- type: track_table
+  args:
+    name: author_mat_view
+    schema: public
+
+- type: track_function
+  version: 2
+  args:
+    function:
+      schema: public
+      name: search_author_mview

server/tests-py/queries/graphql_query/functions/teardown.yaml

@@ -9,4 +9,6 @@ args:
       DROP TABLE integer_column cascade;
       DROP TABLE "user" cascade;
       DROP TABLE text_result cascade;
+      DROP TABLE author cascade;
+      DROP MATERIALIZED VIEW IF EXISTS author_mat_view cascade;
     cascade: true

server/tests-py/test_graphql_queries.py

@@ -544,6 +544,10 @@ def test_track_function_v2_errors(self, hge_ctx):
     @pytest.mark.parametrize("transport", ['http', 'websocket'])
     def test_query_get_test_session_id(self, hge_ctx, transport):
         check_query_f(hge_ctx, self.dir() + '/query_get_test_session_id.yaml')
+
+    @pytest.mark.parametrize("transport", ['http', 'websocket'])
+    def test_query_search_author_mview(self, hge_ctx, transport):
+        check_query_f(hge_ctx, self.dir() + '/query_search_author_mview.yaml')
 
     @classmethod
     def dir(cls):