update procedures

Snowflake-Labs · Dec 4, 2023 · a0f9ccd · a0f9ccd
1 parent d63b18c
commit a0f9ccd
Show file tree

Hide file tree

Showing 10 changed files with 1,059 additions and 1,336 deletions.
diff --git a/pkg/sdk/common_types.go b/pkg/sdk/common_types.go
@@ -207,3 +207,31 @@ func (row *propertyRow) toBoolProperty() *BoolProperty {
 		Description:  row.Description,
 	}
 }
+
+type ExecuteAs string
+
+func ExecuteAsPointer(v ExecuteAs) *ExecuteAs {
+	return &v
+}
+
+const (
+	ExecuteAsCaller ExecuteAs = "EXECUTE AS CALLER"
+	ExecuteAsOwner  ExecuteAs = "EXECUTE AS OWNER"
+)
+
+type NullInputBehavior string
+
+func NullInputBehaviorPointer(v NullInputBehavior) *NullInputBehavior {
+	return &v
+}
+
+const (
+	NullInputBehaviorCalledOnNullInput NullInputBehavior = "CALLED ON NULL INPUT"
+	NullInputBehaviorReturnNullInput   NullInputBehavior = "RETURN NULL ON NULL INPUT"
+	NullInputBehaviorStrict            NullInputBehavior = "STRICT"
+)
+
+type Secret struct {
+	VariableName string `ddl:"keyword,single_quotes"`
+	Name         string `ddl:"parameter,no_quotes"`
+}
diff --git a/pkg/sdk/poc/generator/keyword_builders.go b/pkg/sdk/poc/generator/keyword_builders.go
@@ -79,7 +79,7 @@ func (v *QueryStruct) SetTags() *QueryStruct {
 }
 
 func (v *QueryStruct) OptionalSetTags() *QueryStruct {
-	return v.setTags(nil)
+	return v.setTags(KeywordOptions().SQL("SET TAG"))
 }
 
 func (v *QueryStruct) setTags(transformer *KeywordTransformer) *QueryStruct {
@@ -91,7 +91,7 @@ func (v *QueryStruct) UnsetTags() *QueryStruct {
 }
 
 func (v *QueryStruct) OptionalUnsetTags() *QueryStruct {
-	return v.unsetTags(nil)
+	return v.unsetTags(KeywordOptions().SQL("UNSET TAG"))
 }
 
 func (v *QueryStruct) unsetTags(transformer *KeywordTransformer) *QueryStruct {