code: /*
   * Copyright (c) 2022 Airbyte, Inc., all rights reserved.
   */
  
  package io.airbyte.container_orchestrator;
  
  import io.airbyte.commons.features.FeatureFlags;
  import io.airbyte.commons.json.Jsons;
  import io.airbyte.config.Configs;
  import io.airbyte.config.ReplicationOutput;
  import io.airbyte.config.StandardSyncInput;
  import io.airbyte.metrics.lib.MetricClient;
  import io.airbyte.metrics.lib.MetricClientFactory;
  import io.airbyte.metrics.lib.MetricEmittingApps;
  import io.airbyte.scheduler.models.IntegrationLauncherConfig;
  import io.airbyte.scheduler.models.JobRunConfig;
  import io.airbyte.workers.RecordSchemaValidator;
  import io.airbyte.workers.WorkerConfigs;
  import io.airbyte.workers.WorkerConstants;
  import io.airbyte.workers.WorkerMetricReporter;
  import io.airbyte.workers.WorkerUtils;
  import io.airbyte.workers.general.DefaultReplicationWorker;
  import io.airbyte.workers.general.ReplicationWorker;
  import io.airbyte.workers.internal.AirbyteMessageTracker;
  import io.airbyte.workers.internal.AirbyteSource;
  import io.airbyte.workers.internal.DefaultAirbyteDestination;
  import io.airbyte.workers.internal.DefaultAirbyteSource;
  import io.airbyte.workers.internal.EmptyAirbyteSource;
  import io.airbyte.workers.internal.NamespacingMapper;
  import io.airbyte.workers.process.AirbyteIntegrationLauncher;
  import io.airbyte.workers.process.IntegrationLauncher;
  import io.airbyte.workers.process.KubePodProcess;
  import io.airbyte.workers.process.ProcessFactory;
  import io.airbyte.workers.temporal.sync.ReplicationLauncherWorker;
  import java.nio.file.Path;
  import java.util.Optional;
  import lombok.extern.slf4j.Slf4j;
  
  @Slf4j
  public class ReplicationJobOrchestrator implements JobOrchestrator<StandardSyncInput> {
  
    private final ProcessFactory processFactory;
    private final WorkerConfigs workerConfigs;
    private final Configs configs;
    private final FeatureFlags featureFlags;
  
    public ReplicationJobOrchestrator(final Configs configs,
                                      final WorkerConfigs workerConfigs,
                                      final ProcessFactory processFactory,
                                      final FeatureFlags featureFlags) {
      this.configs = configs;
      this.workerConfigs = workerConfigs;
      this.processFactory = processFactory;
      this.featureFlags = featureFlags;
    }
  
    @Override
    public String getOrchestratorName() {
      return "Replication";
    }
  
    @Override
    public Class<StandardSyncInput> getInputClass() {
      return StandardSyncInput.class;
    }
  
    @Override
    public Optional<String> runJob() throws Exception {
      final JobRunConfig jobRunConfig = JobOrchestrator.readJobRunConfig();
      final StandardSyncInput syncInput = readInput();
  
      final IntegrationLauncherConfig sourceLauncherConfig = JobOrchestrator.readAndDeserializeFile(
          Path.of(KubePodProcess.CONFIG_DIR, ReplicationLauncherWorker.INIT_FILE_SOURCE_LAUNCHER_CONFIG),
          IntegrationLauncherConfig.class);
  
      final IntegrationLauncherConfig destinationLauncherConfig = JobOrchestrator.readAndDeserializeFile(
          Path.of(KubePodProcess.CONFIG_DIR, ReplicationLauncherWorker.INIT_FILE_DESTINATION_LAUNCHER_CONFIG),
          IntegrationLauncherConfig.class);
  
      log.info("Setting up source launcher...");
      final IntegrationLauncher sourceLauncher = new AirbyteIntegrationLauncher(
          sourceLauncherConfig.getJobId(),
          Math.toIntExact(sourceLauncherConfig.getAttemptId()),
          sourceLauncherConfig.getDockerImage(),
          processFactory,
          syncInput.getSourceResourceRequirements());
  
      log.info("Setting up destination launcher...");
      final IntegrationLauncher destinationLauncher = new AirbyteIntegrationLauncher(
          destinationLauncherConfig.getJobId(),
          Math.toIntExact(destinationLauncherConfig.getAttemptId()),
          destinationLauncherConfig.getDockerImage(),
          processFactory,
          syncInput.getDestinationResourceRequirements());
  
      log.info("Setting up source...");
      // reset jobs use an empty source to induce resetting all data in destination.
      final AirbyteSource airbyteSource =
          WorkerConstants.RESET_JOB_SOURCE_DOCKER_IMAGE_STUB.equals(sourceLauncherConfig.getDockerImage()) ? new EmptyAirbyteSource(
              featureFlags.useStreamCapableState())
              : new DefaultAirbyteSource(workerConfigs, sourceLauncher);
  
      MetricClientFactory.initialize(MetricEmittingApps.WORKER);
      final MetricClient metricClient = MetricClientFactory.getMetricClient();
      final WorkerMetricReporter metricReporter = new WorkerMetricReporter(metricClient, sourceLauncherConfig.getDockerImage());
  
      log.info("Setting up replication worker...");
      final ReplicationWorker replicationWorker = new DefaultReplicationWorker(
          jobRunConfig.getJobId(),
          Math.toIntExact(jobRunConfig.getAttemptId()),
          airbyteSource,
          new NamespacingMapper(syncInput.getNamespaceDefinition(), syncInput.getNamespaceFormat(), syncInput.getPrefix()),
          new DefaultAirbyteDestination(workerConfigs, destinationLauncher),
          new AirbyteMessageTracker(),
          new RecordSchemaValidator(WorkerUtils.mapStreamNamesToSchemas(syncInput)),
          metricReporter);
  
      log.info("Running replication worker...");
      final Path jobRoot = WorkerUtils.getJobRoot(configs.getWorkspaceRoot(), jobRunConfig.getJobId(), jobRunConfig.getAttemptId());
      final ReplicationOutput replicationOutput = replicationWorker.run(syncInput, jobRoot);
  
      log.info("Returning output...");
      return Optional.of(Jsons.serialize(replicationOutput));
    }
  
  }
  