fix: use RunContext storage object

kestra-io · Jun 12, 2024 · 7952274 · 7952274
1 parent 20affe4
commit 7952274
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/plugin-jdbc-duckdb/src/main/java/io/kestra/plugin/jdbc/duckdb/Query.java b/plugin-jdbc-duckdb/src/main/java/io/kestra/plugin/jdbc/duckdb/Query.java
@@ -235,7 +235,7 @@ public Query.Output run(RunContext runContext) throws Exception {
 
         if (outputFiles != null) {
             outputFiles
-                .forEach(throwBiConsumer((k, v) -> uploaded.put(k, runContext.putTempFile(new File(runContext.render(v, additionalVars))))));
+                .forEach(throwBiConsumer((k, v) -> uploaded.put(k, runContext.storage().putFile(new File(runContext.render(v, additionalVars))))));
         }
 
         return Output.builder()

diff --git a/plugin-jdbc-pinot/src/test/java/io/kestra/plugin/jdbc/pinot/PinotTest.java b/plugin-jdbc-pinot/src/test/java/io/kestra/plugin/jdbc/pinot/PinotTest.java
@@ -3,6 +3,7 @@
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.runners.RunContext;
 import io.kestra.core.runners.RunContextFactory;
+import io.kestra.core.utils.VersionProvider;
 import io.kestra.plugin.jdbc.AbstractJdbcQuery;
 import io.kestra.core.junit.annotations.KestraTest;
 import jakarta.inject.Inject;

diff --git a/plugin-jdbc-postgres/src/main/java/io/kestra/plugin/jdbc/postgresql/CopyOut.java b/plugin-jdbc-postgres/src/main/java/io/kestra/plugin/jdbc/postgresql/CopyOut.java
@@ -74,7 +74,7 @@ public Output run(RunContext runContext) throws Exception {
 
             return Output
                 .builder()
-                .uri(runContext.putTempFile(path.toFile()))
+                .uri(runContext.storage().putFile(path.toFile()))
                 .rowCount(rowsAffected)
                 .build();
         }

diff --git a/plugin-jdbc-snowflake/src/main/java/io/kestra/plugin/jdbc/snowflake/Download.java b/plugin-jdbc-snowflake/src/main/java/io/kestra/plugin/jdbc/snowflake/Download.java
@@ -98,7 +98,7 @@ public Download.Output run(RunContext runContext) throws Exception {
 
             return Output
                 .builder()
-                .uri(runContext.putTempFile(tempFile))
+                .uri(runContext.storage().putFile(tempFile))
                 .build();
         }
     }