Expose instanceType and machineImage properties for EC2 deployment

restatedev · Sep 2, 2024 · 5f063ca · 5f063ca
1 parent 06b1b21
commit 5f063ca
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/lib/restate-constructs/single-node-restate-deployment.ts b/lib/restate-constructs/single-node-restate-deployment.ts
@@ -20,6 +20,12 @@ import * as cdk from "aws-cdk-lib";
 import { RemovalPolicy } from "aws-cdk-lib";
 
 export interface SingleNodeRestateProps {
+  /** EC2 instance type to use. */
+  instanceType?: ec2.InstanceType;
+
+  /** Machine image. */
+  machineImage?: ec2.IMachineImage;
+
   /** The VPC in which to launch the Restate host. */
   vpc?: ec2.IVpc;
 
@@ -49,8 +55,8 @@ export interface SingleNodeRestateProps {
     clusterName?: string;
     /** RocksDB settings. */
     rocksdb?: {
-      /** Defaults to `512.0 MB`. */
-      totalMemorySize?: string;
+      /** Defaults to 512 MB. */
+      totalMemorySize?: cdk.Size;
     };
   };
 
@@ -190,10 +196,12 @@ export class SingleNodeRestateDeployment extends Construct implements IRestateEn
     const restateInstance = new ec2.Instance(this, "Host", {
       vpc: this.vpc,
       vpcSubnets: { subnetType: ec2.SubnetType.PUBLIC },
-      instanceType: new ec2.InstanceType("t4g.micro"),
-      machineImage: ec2.MachineImage.latestAmazonLinux2023({
-        cpuType: ec2.AmazonLinuxCpuType.ARM_64,
-      }),
+      instanceType: props.instanceType ?? new ec2.InstanceType("t4g.micro"),
+      machineImage:
+        props.machineImage ??
+        ec2.MachineImage.latestAmazonLinux2023({
+          cpuType: ec2.AmazonLinuxCpuType.ARM_64,
+        }),
       role: this.instanceRole,
       blockDevices: [
         {
@@ -259,7 +267,7 @@ export class SingleNodeRestateDeployment extends Construct implements IRestateEn
         `default-thread-pool-size = 3`,
         `storage-high-priority-bg-threads = 3`,
         `storage-low-priority-bg-threads = 3`,
-        `rocksdb-total-memory-size = "${props.restateConfig?.rocksdb?.totalMemorySize ?? "512.0 MB"}"`,
+        `rocksdb-total-memory-size = "${props.restateConfig?.rocksdb?.totalMemorySize?.toMebibytes() ?? 512.0 + " MB"}"`,
         `rocksdb-total-memtables-ratio = 0.6000000238418579`,
         `rocksdb-bg-threads = 3`,
         `rocksdb-high-priority-bg-threads = 3`,

diff --git a/test/__snapshots__/restate-constructs.test.ts.snap b/test/__snapshots__/restate-constructs.test.ts.snap
@@ -702,7 +702,7 @@ exports[`Restate constructs Create a self-hosted Restate environment deployed on
 
                         storage-low-priority-bg-threads = 3
 
-                        rocksdb-total-memory-size = "512.0 MB"
+                        rocksdb-total-memory-size = "512 MB"
 
                         rocksdb-total-memtables-ratio = 0.6000000238418579