initial commit

Author: ianbibby

infrared/manifest.star

@@ -66,6 +66,9 @@ clusters_to_monitor = [
         "name"     : "pga-r2-subreddit-15",
         "endpoint" : "pga-r2-subreddit-15.cluster-cgzk1xobcfng.us-east-1.rds.amazonaws.com:5432/newreddit",
     },
+]
+
+clusters_to_monitor_k8s_vpc = [
     {
         "name"     : "pga-reddit-service-preference",
         "endpoint" : "pga-reddit-service-preference.cluster-cgzk1xobcfng.us-east-1.rds.amazonaws.com",
@@ -76,98 +79,114 @@ clusters_to_monitor = [
     },
 ]
 
-if env() == Production:
-    repository = "artifactory.build.ue1.snooguts.net/reddit-docker-prod/" + git_repo_name
-
-    iam_role = "arn:aws:iam::210246326331:role/prod-3.kubernetes.ue1.k8s-postgres-exporter-pod"
-    service_account = "postgres-exporter"
+def create_deployment():
+    if env() == Production:
+        repository = "artifactory.build.ue1.snooguts.net/reddit-docker-prod/" + git_repo_name
+
+        iam_role = "arn:aws:iam::210246326331:role/prod-3.kubernetes.ue1.k8s-postgres-exporter-pod"
+        service_account = "postgres-exporter"
+
+        requests = bp.resources(cpu = "200m", mem = "128Mi")
+        autoscaler = bp.autoscale(
+            cpu_utilization = 70,
+            min_replicas = 1,
+            max_replicas = 1,
+        )
+    else:
+        config_file = "development.yaml"
+        requests = bp.resources(cpu = "200m", mem = "128Mi")
+        service_account = "snoodev"
+
+    release = bp.release(name = fullname)
+
+    conf = release.config(
+        name = fullname,
+        filename = "production.yaml",
+        contents = file("%s.yaml" % env()),
+        tags = [],
+    )
 
-    requests = bp.resources(cpu = "200m", mem = "128Mi")
-    autoscaler = bp.autoscale(
-        cpu_utilization = 70,
-        min_replicas = 1,
-        max_replicas = 1,
+    metrics_port = release.port(
+        name = "metrics",
+        port = 9187,
+        container_port = 9187,
     )
-else:
-    config_file = "development.yaml"
-    requests = bp.resources(cpu = "200m", mem = "128Mi")
-    service_account = "snoodev"
-
-release = bp.release(name = fullname)
-
-conf = release.config(
-    name = fullname,
-    filename = "production.yaml",
-    contents = file("%s.yaml" % env()),
-    tags = [],
-)
-
-metrics_port = release.port(
-    name = "metrics",
-    port = 9187,
-    container_port = 9187,
-)
-
-container = release.container(
-    name = "exporter",
-    repository = repository,
-    ports = [metrics_port],
-    requests = requests,
-    args = [
-        "--config.file", "/mnt/secrets/postgres-exporter.yaml",
-    ],
-    secret_files = [
-        bp.secret_file(
-            output_file = "postgres-exporter.yaml",
-            secret_path = "secret/postgres-exporter/config",
-            content_key = "current",
-        ),
-    ],
-)
-
-
-if env() == Production:
-    node_specialization = "prod-vpc-apps"
-
-svc = release.service(
-    name = fullname,
-    type = service_type,
-    ports = [metrics_port],
-)
-
-for cluster in clusters_to_monitor:
-    release.podmonitor(
-        name   = cluster["name"],
-        port   = "metrics",
-        path   = "/probe",
-        params = {
-            "target": [cluster["endpoint"]],
-            "auth_module": ["repl_lag_monitor"],
-        },
-        relabelings = [
-            bp.relabel(
-                sourcelabels = ["__param_target"],
-                targetlabel  = "cluster_endpoint",
-            ),
-            bp.relabel(
-                replacement = cluster["name"],
-                targetlabel = "cluster_name",
+
+    container = release.container(
+        name = "exporter",
+        repository = repository,
+        ports = [metrics_port],
+        requests = requests,
+        args = [
+            "--config.file", "/mnt/secrets/postgres-exporter.yaml",
+        ],
+        secret_files = [
+            bp.secret_file(
+                output_file = "postgres-exporter.yaml",
+                secret_path = "secret/postgres-exporter/config",
+                content_key = "current",
             ),
         ],
     )
 
-release.deploy(
-    name = fullname,
-    container = container,
-    config = conf,
-    services = [svc],
-    replicas = 1,
-    service_account = service_account,
-    iam_role = iam_role,
-    node_specialization = node_specialization,
-    autoscale = autoscaler,
-    sidecars = sidecars,
-)
+
+    if env() == Production:
+        node_specialization = "prod-vpc-apps"
+
+    svc = release.service(
+        name = fullname,
+        type = service_type,
+        ports = [metrics_port],
+    )
+
+    for cluster in clusters_to_monitor:
+        release.podmonitor(
+            name   = cluster["name"],
+            port   = "metrics",
+            path   = "/probe",
+            params = {
+                "target": [cluster["endpoint"]],
+                "auth_module": ["repl_lag_monitor"],
+            },
+            relabelings = [
+                bp.relabel(
+                    sourcelabels = ["__param_target"],
+                    targetlabel  = "cluster_endpoint",
+                ),
+                bp.relabel(
+                    replacement = cluster["name"],
+                    targetlabel = "cluster_name",
+                ),
+            ],
+        )
+
+    release.deploy(
+        name = fullname,
+        container = container,
+        config = conf,
+        services = [svc],
+        replicas = 1,
+        service_account = service_account,
+        iam_role = iam_role,
+        node_specialization = node_specialization,
+        autoscale = autoscaler,
+        sidecars = sidecars,
+    )
+
+    release.deploy(
+        name = fullname,
+        container = container,
+        config = conf,
+        services = [svc],
+        replicas = 1,
+        service_account = service_account,
+        iam_role = iam_role,
+        node_specialization = "",
+        autoscale = autoscaler,
+        sidecars = sidecars,
+    )
+
+create_deployment()
 
 # This fixes an inconsistency in our naming scheme for node taints.
 # Remove when the starlark resource generator handles this issue gracefully.