fix proc id parsing for legacy id (#1550)

Snowflake-Labs · Feb 21, 2023 · 266432d · 266432d
1 parent d0384b6
commit 266432d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/pkg/resources/function_grant.go b/pkg/resources/function_grant.go
@@ -344,12 +344,15 @@ func parseFunctionGrantID(s string) (*FunctionGrantID, error) {
 			objectIdentifier = objectIdentifier[0:idx]
 		}
 		objectNameParts := strings.Split(objectIdentifier, "(")
-
+		argumentDataTypes := []string{}
+		if len(objectNameParts) > 1 {
+			argumentDataTypes = helpers.SplitStringToSlice(objectNameParts[1], ",")
+		}
 		return &FunctionGrantID{
 			DatabaseName:      idParts[0],
 			SchemaName:        idParts[1],
 			ObjectName:        objectNameParts[0],
-			ArgumentDataTypes: helpers.SplitStringToSlice(objectNameParts[1], ","),
+			ArgumentDataTypes: argumentDataTypes,
 			Privilege:         idParts[3],
 			Roles:             helpers.SplitStringToSlice(idParts[4], ","),
 			Shares:            []string{},

diff --git a/pkg/resources/procedure_grant.go b/pkg/resources/procedure_grant.go
@@ -354,12 +354,15 @@ func parseProcedureGrantID(s string) (*ProcedureGrantID, error) {
 			objectIdentifier = objectIdentifier[0:idx]
 		}
 		objectNameParts := strings.Split(objectIdentifier, "(")
-
+		argumentDataTypes := []string{}
+		if len(objectNameParts) > 1 {
+			argumentDataTypes = helpers.SplitStringToSlice(objectNameParts[1], ",")
+		}
 		return &ProcedureGrantID{
 			DatabaseName:      idParts[0],
 			SchemaName:        idParts[1],
 			ObjectName:        objectNameParts[0],
-			ArgumentDataTypes: helpers.SplitStringToSlice(objectNameParts[1], ","),
+			ArgumentDataTypes: argumentDataTypes,
 			Privilege:         idParts[3],
 			Roles:             helpers.SplitStringToSlice(idParts[4], ","),
 			Shares:            []string{},