feat(cubesql): Improve DataGrip compatibility

Signed-off-by: Alex Qyoun-ae <[email protected]>

Author: MazterQyou

packages/cubejs-backend-native/Cargo.lock

@@ -10,14 +10,14 @@ homepage = "https://cube.dev"
 
 [dependencies]
 arc-swap = "1"
-datafusion = { git = 'https://github.com/cube-js/arrow-datafusion.git', rev = "7803f7e1cb8fa3d7b25451de35b8765e1ce4a3f6", default-features = false, features = [
+datafusion = { git = 'https://github.com/cube-js/arrow-datafusion.git', rev = "ea7364eee57e573fb0abea7d3a89af9a11adaccb", default-features = false, features = [
     "regex_expressions",
     "unicode_expressions",
 ] }
 thiserror = "2"
 cubeclient = { path = "../cubeclient" }
 pg-srv = { path = "../pg-srv" }
-sqlparser = { git = 'https://github.com/cube-js/sqlparser-rs.git', rev = "34f22de680caa5fe586def5b336d56efe43c8cc4" }
+sqlparser = { git = 'https://github.com/cube-js/sqlparser-rs.git', rev = "b4e4c844a0e9b3e58c61ec2d1520b01a6e1437b7" }
 base64 = "0.13.0"
 tokio = { version = "^1.35", features = ["full", "rt", "tracing"] }
 serde = { version = "^1.0", features = ["derive"] }

rust/cubenativeutils/Cargo.lock

@@ -17,16 +17,21 @@ use super::information_schema::postgres::{
     InfoSchemaSqlImplementationInfoProvider as PostgresInfoSchemaSqlImplementationInfoProvider,
     InfoSchemaSqlSizingProvider as PostgresInfoSchemaSqlSizingProvider,
     InfoSchemaTestingBlockingProvider, InfoSchemaTestingDatasetProvider, PgCatalogAmProvider,
-    PgCatalogAttrdefProvider, PgCatalogAttributeProvider, PgCatalogClassProvider,
+    PgCatalogAttrdefProvider, PgCatalogAttributeProvider, PgCatalogAuthMembersProvider,
+    PgCatalogAvailableExtensionVersionsProvider, PgCatalogCastProvider, PgCatalogClassProvider,
     PgCatalogConstraintProvider, PgCatalogDatabaseProvider, PgCatalogDependProvider,
-    PgCatalogDescriptionProvider, PgCatalogEnumProvider, PgCatalogExtensionProvider,
-    PgCatalogIndexProvider, PgCatalogInheritsProvider, PgCatalogMatviewsProvider,
-    PgCatalogNamespaceProvider, PgCatalogPartitionedTableProvider, PgCatalogProcProvider,
-    PgCatalogRangeProvider, PgCatalogRolesProvider, PgCatalogSequenceProvider,
+    PgCatalogDescriptionProvider, PgCatalogEnumProvider, PgCatalogEventTriggerProvider,
+    PgCatalogExtensionProvider, PgCatalogForeignDataWrapperProvider,
+    PgCatalogForeignServerProvider, PgCatalogForeignTableProvider, PgCatalogIndexProvider,
+    PgCatalogInheritsProvider, PgCatalogLanguageProvider, PgCatalogLocksProvider,
+    PgCatalogMatviewsProvider, PgCatalogNamespaceProvider, PgCatalogOperatorProvider,
+    PgCatalogPartitionedTableProvider, PgCatalogProcProvider, PgCatalogRangeProvider,
+    PgCatalogRewriteProvider, PgCatalogRolesProvider, PgCatalogSequenceProvider,
     PgCatalogSettingsProvider, PgCatalogShdescriptionProvider, PgCatalogStatActivityProvider,
     PgCatalogStatUserTablesProvider, PgCatalogStatioUserTablesProvider, PgCatalogStatsProvider,
-    PgCatalogTableProvider, PgCatalogTypeProvider, PgCatalogUserProvider, PgCatalogViewsProvider,
-    PgPreparedStatementsProvider,
+    PgCatalogTableProvider, PgCatalogTablespaceProvider, PgCatalogTimezoneAbbrevsProvider,
+    PgCatalogTimezoneNamesProvider, PgCatalogTypeProvider, PgCatalogUserMappingProvider,
+    PgCatalogUserProvider, PgCatalogViewsProvider, PgPreparedStatementsProvider,
 };
 use crate::{
     compile::{
@@ -138,6 +143,36 @@ impl DatabaseProtocol {
             "pg_catalog.pg_stat_user_tables".to_string()
         } else if let Some(_) = any.downcast_ref::<PgCatalogShdescriptionProvider>() {
             "pg_catalog.pg_shdescription".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogLocksProvider>() {
+            "pg_catalog.pg_locks".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogTimezoneNamesProvider>() {
+            "pg_catalog.pg_timezone_names".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogTimezoneAbbrevsProvider>() {
+            "pg_catalog.pg_timezone_abbrevs".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogAuthMembersProvider>() {
+            "pg_catalog.pg_auth_members".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogTablespaceProvider>() {
+            "pg_catalog.pg_tablespace".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogEventTriggerProvider>() {
+            "pg_catalog.pg_event_trigger".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogForeignDataWrapperProvider>() {
+            "pg_catalog.pg_foreign_data_wrapper".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogForeignServerProvider>() {
+            "pg_catalog.pg_foreign_server".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogUserMappingProvider>() {
+            "pg_catalog.pg_user_mapping".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogAvailableExtensionVersionsProvider>() {
+            "pg_catalog.pg_available_extension_versions".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogLanguageProvider>() {
+            "pg_catalog.pg_language".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogCastProvider>() {
+            "pg_catalog.pg_cast".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogForeignTableProvider>() {
+            "pg_catalog.pg_foreign_table".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogOperatorProvider>() {
+            "pg_catalog.pg_operator".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogRewriteProvider>() {
+            "pg_catalog.pg_rewrite".to_string()
         } else if let Some(_) = any.downcast_ref::<RedshiftPgExternalSchemaProvider>() {
             "pg_catalog.pg_external_schema".to_string()
         } else if let Some(_) = any.downcast_ref::<RedshiftSvvTablesTableProvider>() {
@@ -404,6 +439,31 @@ impl DatabaseProtocol {
                     )))
                 }
                 "pg_shdescription" => return Some(Arc::new(PgCatalogShdescriptionProvider::new())),
+                "pg_locks" => return Some(Arc::new(PgCatalogLocksProvider::new())),
+                "pg_timezone_names" => {
+                    return Some(Arc::new(PgCatalogTimezoneNamesProvider::new()))
+                }
+                "pg_timezone_abbrevs" => {
+                    return Some(Arc::new(PgCatalogTimezoneAbbrevsProvider::new()))
+                }
+                "pg_auth_members" => return Some(Arc::new(PgCatalogAuthMembersProvider::new())),
+                "pg_tablespace" => return Some(Arc::new(PgCatalogTablespaceProvider::new())),
+                "pg_event_trigger" => return Some(Arc::new(PgCatalogEventTriggerProvider::new())),
+                "pg_foreign_data_wrapper" => {
+                    return Some(Arc::new(PgCatalogForeignDataWrapperProvider::new()))
+                }
+                "pg_foreign_server" => {
+                    return Some(Arc::new(PgCatalogForeignServerProvider::new()))
+                }
+                "pg_user_mapping" => return Some(Arc::new(PgCatalogUserMappingProvider::new())),
+                "pg_available_extension_versions" => {
+                    return Some(Arc::new(PgCatalogAvailableExtensionVersionsProvider::new()))
+                }
+                "pg_language" => return Some(Arc::new(PgCatalogLanguageProvider::new())),
+                "pg_cast" => return Some(Arc::new(PgCatalogCastProvider::new())),
+                "pg_foreign_table" => return Some(Arc::new(PgCatalogForeignTableProvider::new())),
+                "pg_operator" => return Some(Arc::new(PgCatalogOperatorProvider::new())),
+                "pg_rewrite" => return Some(Arc::new(PgCatalogRewriteProvider::new())),
                 "pg_external_schema" => {
                     return Some(Arc::new(RedshiftPgExternalSchemaProvider::new()))
                 }

rust/cubesql/Cargo.lock

@@ -16,21 +16,32 @@ pub mod views;
 mod pg_am;
 mod pg_attrdef;
 mod pg_attribute;
+mod pg_auth_members;
+mod pg_available_extension_versions;
+mod pg_cast;
 mod pg_class;
 mod pg_constraint;
 mod pg_database;
 mod pg_depend;
 mod pg_description;
 mod pg_enum;
+mod pg_event_trigger;
 mod pg_extension;
+mod pg_foreign_data_wrapper;
+mod pg_foreign_server;
+mod pg_foreign_table;
 mod pg_index;
 mod pg_inherits;
+mod pg_language;
+mod pg_locks;
 mod pg_matviews;
 mod pg_namespace;
+mod pg_operator;
 mod pg_partitioned_table;
 mod pg_prepared_statements;
 mod pg_proc;
 mod pg_range;
+mod pg_rewrite;
 mod pg_roles;
 mod pg_sequence;
 mod pg_settings;
@@ -40,8 +51,12 @@ mod pg_stat_user_tables;
 mod pg_statio_user_tables;
 mod pg_stats;
 mod pg_tables;
+mod pg_tablespace;
+mod pg_timezone_abbrevs;
+mod pg_timezone_names;
 mod pg_type;
 mod pg_user;
+mod pg_user_mapping;
 mod pg_views;
 mod role_column_grants;
 mod role_table_grants;
@@ -52,21 +67,32 @@ use super::utils;
 pub use pg_am::*;
 pub use pg_attrdef::*;
 pub use pg_attribute::*;
+pub use pg_auth_members::*;
+pub use pg_available_extension_versions::*;
+pub use pg_cast::*;
 pub use pg_class::*;
 pub use pg_constraint::*;
 pub use pg_database::*;
 pub use pg_depend::*;
 pub use pg_description::*;
 pub use pg_enum::*;
+pub use pg_event_trigger::*;
 pub use pg_extension::*;
+pub use pg_foreign_data_wrapper::*;
+pub use pg_foreign_server::*;
+pub use pg_foreign_table::*;
 pub use pg_index::*;
 pub use pg_inherits::*;
+pub use pg_language::*;
+pub use pg_locks::*;
 pub use pg_matviews::*;
 pub use pg_namespace::*;
+pub use pg_operator::*;
 pub use pg_partitioned_table::*;
 pub use pg_prepared_statements::*;
 pub use pg_proc::*;
 pub use pg_range::*;
+pub use pg_rewrite::*;
 pub use pg_roles::*;
 pub use pg_sequence::*;
 pub use pg_settings::*;
@@ -76,8 +102,12 @@ pub use pg_stat_user_tables::*;
 pub use pg_statio_user_tables::*;
 pub use pg_stats::*;
 pub use pg_tables::*;
+pub use pg_tablespace::*;
+pub use pg_timezone_abbrevs::*;
+pub use pg_timezone_names::*;
 pub use pg_type::*;
 pub use pg_user::*;
+pub use pg_user_mapping::*;
 pub use pg_views::*;
 pub use role_column_grants::*;
 pub use role_table_grants::*;