modules/data_warehouse/workflow_polling/main.tf

/**
 * Copyright 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

data "google_client_config" "current" {
}

locals {
  is_active = var.input_workflow_state == null || var.input_workflow_state == "FAILED" ? "no" : "yes"
}

## Trigger the execution of the setup workflow with an API call
data "http" "call_workflows_setup" {
  url    = "https://workflowexecutions.googleapis.com/v1/${var.workflow_id}/executions"
  method = local.is_active == "no" ? "POST" : "GET"
  # method = "POST"
  request_headers = {
    Accept = "application/json"
  Authorization = "Bearer ${data.google_client_config.current.access_token}" }
}

resource "time_sleep" "workflow_execution_wait" {
  create_duration = "120s"
  depends_on = [
    data.http.call_workflows_setup
  ]
}

# locals {
#   is_active = var.input_workflow_state == null || var.input_workflow_state == "FAILED" ? "no" : "yes"
# }

# ## Trigger the execution of the setup workflow with an API call
# data "http" "call_workflows_setup" {
#   url    = "https://workflowexecutions.googleapis.com/v1/${var.workflow_id}/executions"
#   method = local.is_active == "no" ? "POST" : "GET"
#   request_headers = {
#     Accept = "application/json"
#   Authorization = "Bearer ${data.google_client_config.current.access_token}" }
# }

# ## If the workflow last failed, sleep for 30 seconds before checking the workflow execution status.
# ## If last execution did not fail, exit as quickly as possible (1 second)
# resource "time_sleep" "workflow_execution_wait" {
#   create_duration = local.is_active == "no" ? "30s" : "1s"
#   depends_on = [
#     data.http.call_workflows_setup,
#   ]
# }

## Check the state of the setup workflow execution with an API call
data "http" "call_workflows_state" {
  url    = "https://workflowexecutions.googleapis.com/v1/${var.workflow_id}/executions"
  method = "GET"
  request_headers = {
    Accept = "application/json"
  Authorization = "Bearer ${data.google_client_config.current.access_token}" }
  depends_on = [
    data.http.call_workflows_setup,
    time_sleep.workflow_execution_wait
  ]
}

## Parse out the workflow execution state from the API call response
locals {
  response_body  = jsondecode(data.http.call_workflows_state.response_body)
  workflow_state = local.response_body.executions[0].state
}

## Output the workflow state to use as input for subsequent invocations
output "workflow_state" {
  description = "State of the most recent workflow execution. Used to determine how to proceed with next polling run."
  value       = local.workflow_state
}

# ## If workflow execution is actively running, sleep for 90 seconds to allow it to finish
# ## If not, exit as quickly as possible (1 second)
# resource "time_sleep" "complete_workflow" {
#   create_duration = local.workflow_state == "ACTIVE" ? "90s" : "1s"
#   depends_on = [
#     data.http.call_workflows_setup,
#     time_sleep.workflow_execution_wait,
#     data.http.call_workflows_state,
#     local.workflow_state
#   ]
# }