Merge pull request #19 from kurtosis-tech/barnabasbusa-patch-1

feat: Add node selectors to postgres

Author: barnabasbusa

main.star

@@ -32,6 +32,7 @@ def run(
     max_cpu=POSTGRES_MAX_CPU,
     min_memory=POSTGRES_MIN_MEMORY,
     max_memory=POSTGRES_MAX_MEMORY,
+    node_selectors=None,
 ):
     """Launches a Postgresql database instance, optionally seeding it with a SQL file script
 
@@ -52,6 +53,7 @@ def run(
         max_cpu (int): Define how much CPU millicores the service should be assign max.
         min_memory (int): Define how much MB of memory the service should be assigned at least.
         max_memory (int): Define how much MB of memory the service should be assigned max.
+        node_selectors (dict): Define a dict of node selectors - only works in kubernetes example: {"kubernetes.io/hostname": node-name-01}
     Returns:
         An object containing useful information about the Postgres database running inside the enclave:
         ```
@@ -95,7 +97,8 @@ def run(
             persistent_key= "data-{0}".format(service_name),
         )
         env_vars["PGDATA"] = DATA_DIRECTORY_PATH + "/pgdata"
-
+    if node_selectors == None:
+        node_selectors = {}
     if config_file_artifact_name != "":
         config_filepath = CONFIG_FILE_MOUNT_DIRPATH + "/" + CONFIG_FILENAME
         cmd += ["-c", "config_file=" + config_filepath]
@@ -126,6 +129,7 @@ def run(
             max_cpu=max_cpu,
             min_memory=min_memory,
             max_memory=max_memory,
+            node_selectors=node_selectors,
         ),
     )
 
@@ -158,6 +162,7 @@ def run(
         max_cpu=max_cpu,
         min_memory=min_memory,
         max_memory=max_memory,
+        node_selectors=node_selectors,
     )