feat: allow custom args for Mongo

techfromsage · Sep 9, 2022 · a2fa174 · a2fa174
1 parent 07c7ee1
commit a2fa174
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 7 deletions.
diff --git a/lib/mongo/mongo-props.ts b/lib/mongo/mongo-props.ts
@@ -17,6 +17,11 @@ export interface MongoProps {
    */
   readonly storageEngine?: "wiredTiger" | "inMemory" | "mmapv1";
 
+  /**
+   * Override the default arguments (storageEngine) to the container.
+   */
+  readonly customArgs?: string[];
+
   /**
    * Storage for the mongo pod
    * @default Quantity.fromString("20Gi")

diff --git a/lib/mongo/mongo.ts b/lib/mongo/mongo.ts
@@ -19,7 +19,6 @@ export class Mongo extends Construct {
     const app = props.selectorLabels?.app ?? chart.labels.app;
     const release = props.release ?? "3.2.8";
     const port = 27017;
-    const storageEngine = props.storageEngine ?? "mmapv1";
     const labels = {
       ...chart.labels,
       release: release,
@@ -44,11 +43,7 @@ export class Mongo extends Construct {
       ...selectorLabels,
     };
 
-    const args = ["--storageEngine", storageEngine];
-
-    if (storageEngine === "mmapv1") {
-      args.push("--smallfiles");
-    }
+    const args = this.getCommandArgs(props);
 
     this.service = new KubeService(this, id, {
       metadata: {
@@ -150,4 +145,19 @@ export class Mongo extends Construct {
   public getDnsName(replica = 0): string {
     return `${this.statefulSet.name}-${replica}.${this.service.name}`;
   }
+
+  private getCommandArgs(props: MongoProps): string[] {
+    if (props.customArgs) {
+      return props.customArgs;
+    }
+
+    const storageEngine = props.storageEngine ?? "mmapv1";
+    const args = ["--storageEngine", storageEngine];
+
+    if (storageEngine === "mmapv1") {
+      args.push("--smallfiles");
+    }
+
+    return args;
+  }
 }
diff --git a/test/mongo/mongo.test.ts b/test/mongo/mongo.test.ts
@@ -39,7 +39,7 @@ describe("Mongo", () => {
         instance: "test",
       };
 
-      const allProps: Required<MongoProps> = {
+      const allProps: Required<Omit<MongoProps, "customArgs">> = {
         ...requiredProps,
         selectorLabels,
         storageEngine: "wiredTiger",
@@ -158,6 +158,21 @@ describe("Mongo", () => {
     });
   });
 
+  describe("customArgs", () => {
+    test("It accepts custom arg that override the default", () => {
+      const results = synthMongo({
+        ...requiredProps,
+        customArgs: ["--smallfiles", "--quotaFiles", "8"],
+      });
+      const mongo = results.find((obj) => obj.kind === "StatefulSet");
+      expect(mongo).toHaveProperty("spec.template.spec.containers[0].args", [
+        "--smallfiles",
+        "--quotaFiles",
+        "8",
+      ]);
+    });
+  });
+
   describe("getDnsName", () => {
     test("Builds a DNS name for the first Pod", () => {
       const chart = makeChart();