Function main

batch/cpp_application/driver.cc:26–142

#include <thread>

int main(int argc, char* argv[]) try {
  if (argc != 6) {
    std::cerr << "Usage: " << argv[0]
              << " <project-id> <region-id> <job-id> <job-json-file> "
                 "<repository-name>\n";
    return 1;
  }

  namespace batch = ::google::cloud::batch_v1;

  auto const project_id = std::string(argv[1]);
  auto const location = google::cloud::Location(argv[1], argv[2]);
  auto const job_id = std::string(argv[3]);
  auto const job_file = std::string(argv[4]);
  auto const repository_name = std::string(argv[5]);

  // Parse the json and convert into protobuf format.
  std::ifstream file(job_file, std::ios::in);
  if (!file.is_open()) {
    std::cout << "Failed to open JSON file: " << job_file << '\n';
    return 0;
  }
  auto contents = std::string{std::istreambuf_iterator<char>(file), {}};
  google::cloud::batch::v1::Job job;
  google::protobuf::util::JsonParseOptions options;
  google::protobuf::util::Status status =
      google::protobuf::util::JsonStringToMessage(contents, &job, options);
  if (!status.ok()) throw status;

  // Modify the job for the containerized application
  auto container = job.mutable_task_groups()
                       ->at(0)
                       .mutable_task_spec()
                       ->mutable_runnables()
                       ->at(0)
                       .mutable_container();

  std::string image_uri =
      std::format("{}-docker.pkg.dev/{}/{}/application-image:latest",
                  location.location_id(), project_id, repository_name);
  container->set_image_uri(image_uri);

  // Create the cloud batch client.
  auto client = batch::BatchServiceClient(batch::MakeBatchServiceConnection());

  // Create a job.
  auto response = client.CreateJob(location.FullName(), job, job_id);

  if (response.status().code() != google::cloud::StatusCode::kOk) {
    if (response.status().code() ==
        google::cloud::StatusCode::kResourceExhausted) {
      std::cout << "There already exists a job for the parent `"
                << location.FullName() << "` and job_id: `" << job_id
                << "`. Please try again with a new job id.\n";
      return 0;
    }
    throw std::move(response).status();
  }