Optimize CDB_UserDataSize (on behalf of @javisantana) #65

scripts-available/CDB_Quota.sql

@@ -3,50 +3,37 @@ CREATE OR REPLACE FUNCTION CDB_UserDataSize(schema_name TEXT)
 RETURNS bigint AS
 $$
 DECLARE
-  quota_vector INT8;
+  total_size INT8;
-  quota_raster INT8;
 BEGIN
-  -- TODO: double check queries. Maybe use CDB_TableMetadata for lookup?
+  WITH raster_tables AS (
-  --  Also, "table_name" sounds sensible to search_path
+    SELECT o_table_name, r_table_name FROM raster_overviews
-
-  -- Division by 2 is for not counting the_geom_webmercator
-  SELECT COALESCE(INT8(SUM(pg_total_relation_size('"' || schema_name || '"."' || table_name || '"')) / 2), 0) INTO quota_vector
-  FROM information_schema.tables
-  WHERE table_catalog = current_database() AND table_schema = schema_name
-    AND table_name != 'spatial_ref_sys'
-    AND table_name != 'cdb_tablemetadata'
-    AND table_type = 'BASE TABLE'
-    -- exclude raster overview tables
-    AND table_name NOT IN (
-      SELECT o_table_name FROM raster_overviews
       WHERE o_table_schema = schema_name AND o_table_catalog = current_database()
-    )
+  ),
-    -- exclude raster "main" tables
+  user_tables AS (
-    AND table_name NOT IN (
+    SELECT table_name FROM information_schema.tables
-      SELECT r_table_name FROM raster_overviews
+      WHERE table_catalog = current_database() AND table_schema = schema_name
-      WHERE r_table_name = table_name
+        AND table_name != 'spatial_ref_sys'
-        AND o_table_schema = schema_name AND o_table_catalog = current_database()
+        AND table_name != 'cdb_tablemetadata'
-    );
+        AND table_type = 'BASE TABLE'
-
+  ),
-  SELECT COALESCE(INT8(SUM(pg_total_relation_size('"' || schema_name || '"."' || table_name || '"'))), 0) INTO quota_raster
+  table_cat AS (
-  FROM information_schema.tables
+    SELECT
-  WHERE table_catalog = current_database() AND table_schema = schema_name
+      table_name,
-    AND table_name != 'spatial_ref_sys'
+      EXISTS(select * from raster_tables where o_table_name = table_name) AS is_overview,
-    AND table_name != 'cdb_tablemetadata'
+      EXISTS(SELECT * FROM raster_tables WHERE r_table_name = table_name) AS is_raster
-    AND table_type = 'BASE TABLE'
+    FROM user_tables
-    -- exclude raster overview tables
+  ),
-    AND table_name NOT IN (
+  sizes AS (
-      SELECT o_table_name FROM raster_overviews
+    SELECT COALESCE(INT8(SUM(pg_total_relation_size('"' || schema_name || '"."' || table_name || '"')))) table_size,
-      WHERE o_table_schema = schema_name AND o_table_catalog = current_database()
+      CASE
-    )
+        WHEN is_overview THEN 0
-    -- filter to raster "main" tables
+	WHEN is_raster THEN 1
-    AND table_name IN (
+	ELSE 0.5
-      SELECT r_table_name FROM raster_overviews
+      END AS multiplier FROM table_cat GROUP BY is_overview, is_raster
-      WHERE r_table_name = table_name
+  )
-        AND o_table_schema = schema_name AND o_table_catalog = current_database()
+  SELECT sum(table_size*multiplier)::int8 INTO total_size FROM sizes;
-    );
+
-
+  RETURN total_size;
-  RETURN quota_vector + quota_raster;
 END;
 $$
 LANGUAGE 'plpgsql' VOLATILE;