Skip to content

Commit

Permalink
feat(helm): allow name override for Dagit deployment (#8005)
Browse files Browse the repository at this point in the history
  • Loading branch information
rexledesma committed May 23, 2022
1 parent c4d844d commit 60e8774
Show file tree
Hide file tree
Showing 5 changed files with 23 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ class Workspace(BaseModel):
class Dagit(BaseModel):
replicaCount: int
image: kubernetes.Image
nameOverride: str
service: kubernetes.Service
workspace: Workspace
env: Dict[str, str]
Expand Down
4 changes: 2 additions & 2 deletions helm/dagster/schema/schema/utils/helm_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class HelmTemplate:
subchart_paths: List[str]
output: Optional[str] = None
model: Optional[Any] = None
name: str = "release-name"
release_name: str = "release-name"
api_client: ApiClient = ApiClient()

def render(
Expand All @@ -44,7 +44,7 @@ def render(
command = [
"helm",
"template",
self.name,
self.release_name,
helm_dir_path,
"--debug",
"--values",
Expand Down
12 changes: 12 additions & 0 deletions helm/dagster/schema/schema_tests/test_dagit.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,18 @@ def test_dagit_image_tag(deployment_template: HelmTemplate):
assert image_tag == tag


@pytest.mark.parametrize("name_override", ["dagit", "new-name"])
def test_dagit_name_override(deployment_template, name_override):
helm_values = DagsterHelmValues.construct(dagit=Dagit.construct(nameOverride=name_override))
dagit_deployments = deployment_template.render(helm_values)

assert len(dagit_deployments) == 1

deployment_name = dagit_deployments[0].metadata.name

assert deployment_name == f"{deployment_template.release_name}-{name_override}"


def test_dagit_service(service_template):
helm_values = DagsterHelmValues.construct()
dagit_template = service_template.render(helm_values)
Expand Down
5 changes: 5 additions & 0 deletions helm/dagster/values.schema.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

3 changes: 3 additions & 0 deletions helm/dagster/values.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,9 @@ dagit:
tag: ~
pullPolicy: Always

# Support overriding the name prefix of the Dagit pods
nameOverride: "dagit"

service:
type: ClusterIP
# Defines the port where Dagit will serve requests; if changed, don't forget to update the
Expand Down

0 comments on commit 60e8774

Please sign in to comment.