def add_app_specific_vars()

in tools/deploy/app.py [0:0]

• 88 lines of code
• 23 McCabe index (conditional complexity)

    def add_app_specific_vars(self, env_vars):
        """
        Adds env variables specific to the application
        """
        if self.name == "placement_stateful":
            env_vars["TASK_TYPE"] = "STATEFUL"
            env_vars["APP"] = "placement"
            env_vars["APP_TYPE"] = "placement_stateful"
            if self.cluster.use_host_pool:
                env_vars["USE_HOST_POOL"] = self.cluster.use_host_pool

        if self.name == "placement_stateless":
            env_vars["TASK_TYPE"] = "STATELESS"
            env_vars["APP"] = "placement"
            env_vars["APP_TYPE"] = "placement_stateless"
            if self.cluster.use_host_pool:
                env_vars["USE_HOST_POOL"] = self.cluster.use_host_pool

        if self.name == "placement":
            if getattr(self, "task_dequeue_limit", ""):
                env_vars["TASK_DEQUEUE_LIMIT"] = self.task_dequeue_limit
            if getattr(self, "task_dequeue_period", ""):
                env_vars["TASK_DEQUEUE_PERIOD"] = self.task_dequeue_period
            if self.cluster.use_host_pool:
                env_vars["USE_HOST_POOL"] = self.cluster.use_host_pool

        if self.name == "jobmgr":
            env_vars["JOB_TYPE"] = getattr(self, "job_type", "BATCH")
            env_vars["JOB_RUNTIME_CALCULATION_VIA_CACHE"] = getattr(
                self, "job_runtime_calculation_via_cache", False
            )
            env_vars["TASK_KILL_RATE_LIMIT"] = getattr(
                self, "task_kill_rate_limit", 0.0
            )
            env_vars["TASK_KILL_BURST_LIMIT"] = getattr(
                self, "task_kill_burst_limit", 0
            )
            env_vars["TASK_LAUNCH_TIMEOUT"] = getattr(
                self, "task_launch_timeout", "0"
            )
            env_vars["TASK_START_TIMEOUT"] = getattr(
                self, "task_start_timeout", "0"
            )
            env_vars["EXECUTOR_SHUTDOWN_RATE_LIMIT"] = getattr(
                self, "executor_shutdown_rate_limit", 0.0
            )
            env_vars["EXECUTOR_SHUTDOWN_BURST_LIMIT"] = getattr(
                self, "executor_shutdown_burst_limit", 0
            )

        if self.name == "archiver":
            env_vars["ENABLE_ARCHIVER"] = self.enable_archiver
            env_vars["STREAM_ONLY_MODE"] = self.stream_only_mode
            env_vars["POD_EVENTS_CLEANUP"] = self.pod_events_cleanup
            env_vars["ARCHIVE_AGE"] = self.archive_age
            env_vars["ARCHIVE_INTERVAL"] = self.archive_interval
            env_vars["ARCHIVE_STEP_SIZE"] = self.archive_step_size
            env_vars["KAFKA_TOPIC"] = self.kafka_topic

        if self.name == "resmgr":
            env_vars["TASK_PREEMPTION_PERIOD"] = getattr(
                self, "task_preemption_period", "60s"
            )
            env_vars["ENABLE_SLA_TRACKING"] = getattr(
                self, "enable_sla_tracking", False
            )
            env_vars["ENABLE_PREEMPTION"] = getattr(
                self, "enable_preemption", False
            )

        if self.name == "hostmgr":
            if self.scarce_resource_types:
                env_vars["SCARCE_RESOURCE_TYPES"] = ",".join(
                    self.scarce_resource_types
                )
            if self.slack_resource_types:
                env_vars["SLACK_RESOURCE_TYPES"] = ",".join(
                    self.slack_resource_types
                )
            if self.enable_revocable_resources:
                env_vars[
                    "ENABLE_REVOCABLE_RESOURCES"
                ] = self.enable_revocable_resources
            if self.bin_packing:
                env_vars["BIN_PACKING"] = self.bin_packing
            if self.qos_advisor_service_address:
                env_vars["QOS_ADVISOR_SERVICE_ADDRESS"] \
                    = self.qos_advisor_service_address
            if self.cluster.use_host_pool:
                env_vars["ENABLE_HOST_POOL"] = self.cluster.use_host_pool

        if self.name == "aurorabridge":
            if self.respool_path:
                env_vars["RESPOOL_PATH"] = self.respool_path
            if self.gpu_respool_path:
                env_vars["GPU_RESPOOL_PATH"] = self.gpu_respool_path
            if self.enable_inplace_update:
                env_vars["ENABLE_INPLACE_UPDATE"] = "true"