diff --git a/src/main/java/io/kestra/plugin/dbt/cloud/CheckStatus.java b/src/main/java/io/kestra/plugin/dbt/cloud/CheckStatus.java index 50e4dbe..61f49ba 100644 --- a/src/main/java/io/kestra/plugin/dbt/cloud/CheckStatus.java +++ b/src/main/java/io/kestra/plugin/dbt/cloud/CheckStatus.java @@ -3,6 +3,7 @@ import io.kestra.core.exceptions.IllegalVariableEvaluationException; import io.kestra.core.models.annotations.Example; import io.kestra.core.models.annotations.Plugin; +import io.kestra.core.models.annotations.PluginProperty; import io.kestra.core.models.property.Property; import io.kestra.core.models.tasks.RunnableTask; import io.kestra.core.runners.RunContext; @@ -77,14 +78,16 @@ public class CheckStatus extends AbstractDbtCloud implements RunnableTask pollFrequency = Property.of(Duration.ofSeconds(5)); + Duration pollFrequency = Duration.ofSeconds(5); @Schema( title = "The maximum duration the task should poll for the job completion." ) + @PluginProperty(dynamic = true) @Builder.Default - Property maxDuration = Property.of(Duration.ofMinutes(60)); + Duration maxDuration = Duration.ofMinutes(60); @Builder.Default @Schema( @@ -139,8 +142,8 @@ public CheckStatus.Output run(RunContext runContext) throws Exception { return null; }), - this.pollFrequency.as(runContext, Duration.class), - this.maxDuration.as(runContext, Duration.class) + this.pollFrequency, + this.maxDuration ); // final response @@ -212,7 +215,7 @@ private Optional fetchRunResponse(RunContext runContext, Long id, B )) ), Argument.of(RunResponse.class), - maxDuration.as(runContext, Duration.class) + maxDuration ) .getBody(); } diff --git a/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java b/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java index 727d949..8365529 100644 --- a/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java +++ b/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java @@ -2,6 +2,7 @@ import io.kestra.core.models.annotations.Example; import io.kestra.core.models.annotations.Plugin; +import io.kestra.core.models.annotations.PluginProperty; import io.kestra.core.models.property.Property; import io.kestra.core.models.tasks.RunnableTask; import io.kestra.core.runners.RunContext; @@ -126,14 +127,16 @@ public class TriggerRun extends AbstractDbtCloud implements RunnableTask pollFrequency = Property.of(Duration.ofSeconds(5)); + Duration pollFrequency = Duration.ofSeconds(5); @Schema( title = "The maximum total wait duration." ) + @PluginProperty(dynamic = true) @Builder.Default - Property maxDuration = Property.of(Duration.ofMinutes(60)); + Duration maxDuration = Duration.ofMinutes(60); @Builder.Default @Schema( diff --git a/src/test/java/io/kestra/plugin/dbt/cloud/CheckStatusTest.java b/src/test/java/io/kestra/plugin/dbt/cloud/CheckStatusTest.java index 7a06771..3fb1602 100644 --- a/src/test/java/io/kestra/plugin/dbt/cloud/CheckStatusTest.java +++ b/src/test/java/io/kestra/plugin/dbt/cloud/CheckStatusTest.java @@ -52,7 +52,7 @@ void run() throws Exception { .runId(Property.of(runOutput.getRunId().toString())) .token(Property.of(this.token)) .accountId(Property.of(this.accountId)) - .maxDuration(Property.of(Duration.ofMinutes(60))) + .maxDuration(Duration.ofMinutes(60)) .parseRunResults(Property.of(false)) .build();