(nifi) branch main updated: NIFI-14833 Add FlowFile attribute support to PutSnowflakeInternalStage (#10181)

[exceptionfactory]

This is an automated email from the ASF dual-hosted git repository.

exceptionfactory pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/nifi.git


The following commit(s) were added to refs/heads/main by this push:
     new a9bb767716 NIFI-14833 Add FlowFile attribute support to 
PutSnowflakeInternalStage (#10181)
a9bb767716 is described below

commit a9bb767716a05696308ade243522a93f040747d0
Author: Pierre Villard <pierre.villard...@gmail.com>
AuthorDate: Fri Aug 8 18:21:13 2025 +0200

    NIFI-14833 Add FlowFile attribute support to PutSnowflakeInternalStage 
(#10181)
    
    Signed-off-by: David Handermann <exceptionfact...@apache.org>
---
 .../apache/nifi/processors/snowflake/PutSnowflakeInternalStage.java | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git 
a/nifi-extension-bundles/nifi-snowflake-bundle/nifi-snowflake-processors/src/main/java/org/apache/nifi/processors/snowflake/PutSnowflakeInternalStage.java
 
b/nifi-extension-bundles/nifi-snowflake-bundle/nifi-snowflake-processors/src/main/java/org/apache/nifi/processors/snowflake/PutSnowflakeInternalStage.java
index 1e8eab1501..63b73cad17 100644
--- 
a/nifi-extension-bundles/nifi-snowflake-bundle/nifi-snowflake-processors/src/main/java/org/apache/nifi/processors/snowflake/PutSnowflakeInternalStage.java
+++ 
b/nifi-extension-bundles/nifi-snowflake-bundle/nifi-snowflake-processors/src/main/java/org/apache/nifi/processors/snowflake/PutSnowflakeInternalStage.java
@@ -75,11 +75,13 @@ public class PutSnowflakeInternalStage extends 
AbstractProcessor {
 
     public static final PropertyDescriptor DATABASE = new 
PropertyDescriptor.Builder()
             .fromPropertyDescriptor(SnowflakeProperties.DATABASE)
+            
.expressionLanguageSupported(ExpressionLanguageScope.FLOWFILE_ATTRIBUTES)
             .dependsOn(INTERNAL_STAGE_TYPE, SnowflakeInternalStageType.NAMED, 
SnowflakeInternalStageType.TABLE)
             .build();
 
     public static final PropertyDescriptor SCHEMA = new 
PropertyDescriptor.Builder()
             .fromPropertyDescriptor(SnowflakeProperties.SCHEMA)
+            
.expressionLanguageSupported(ExpressionLanguageScope.FLOWFILE_ATTRIBUTES)
             .dependsOn(INTERNAL_STAGE_TYPE, SnowflakeInternalStageType.NAMED, 
SnowflakeInternalStageType.TABLE)
             .build();
 
@@ -170,8 +172,8 @@ public class PutSnowflakeInternalStage extends 
AbstractProcessor {
 
     private SnowflakeInternalStageTypeParameters 
getSnowflakeInternalStageTypeParameters(ProcessContext context,
             FlowFile flowFile) {
-        final String database = 
context.getProperty(DATABASE).evaluateAttributeExpressions().getValue();
-        final String schema = 
context.getProperty(SCHEMA).evaluateAttributeExpressions().getValue();
+        final String database = 
context.getProperty(DATABASE).evaluateAttributeExpressions(flowFile).getValue();
+        final String schema = 
context.getProperty(SCHEMA).evaluateAttributeExpressions(flowFile).getValue();
         final String table = 
context.getProperty(TABLE).evaluateAttributeExpressions(flowFile).getValue();
         final String stageName = 
context.getProperty(INTERNAL_STAGE).evaluateAttributeExpressions(flowFile).getValue();
         return new SnowflakeInternalStageTypeParameters(database, schema, 
table, stageName);