mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-10-09 21:42:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add support for granting privileges to procedures and functions in Snowflake (#1930)

Browse source
This commit is contained in:
Yoav Cohen 2025-07-10 17:28:51 +02:00 • committed by GitHub
parent b1a6d11e12
commit 8f1414efff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 69 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/ast/mod.rs
View file

@ -7023,6 +7023,25 @@ pub enum GrantObjects {
ReplicationGroup(Vec<ObjectName>),
/// Grant privileges on external volumes
ExternalVolumes(Vec<ObjectName>),
/// Grant privileges on a procedure. In dialects that
/// support overloading, the argument types must be specified.
///
/// For example:
/// `GRANT USAGE ON PROCEDURE foo(varchar) TO ROLE role1`
Procedure {
name: ObjectName,
arg_types: Vec<DataType>,
},
/// Grant privileges on a function. In dialects that
/// support overloading, the argument types must be specified.
///
/// For example:
/// `GRANT USAGE ON FUNCTION foo(varchar) TO ROLE role1`
Function {
name: ObjectName,
arg_types: Vec<DataType>,
},
}
impl fmt::Display for GrantObjects {
@ -7147,6 +7166,20 @@ impl fmt::Display for GrantObjects {
GrantObjects::ExternalVolumes(objects) => {
write!(f, "EXTERNAL VOLUME {}", display_comma_separated(objects))
}
GrantObjects::Procedure { name, arg_types } => {
write!(f, "PROCEDURE {name}")?;
if !arg_types.is_empty() {
write!(f, "({})", display_comma_separated(arg_types))?;
}
Ok(())
}
GrantObjects::Function { name, arg_types } => {
write!(f, "FUNCTION {name}")?;
if !arg_types.is_empty() {
write!(f, "({})", display_comma_separated(arg_types))?;
}
Ok(())
}
}
}
}