Add option to use 50 percent memory as heap (#23)

README.md

@@ -58,7 +58,7 @@ In order to deploy both the stacks the user needs to provide a set of required a
 | account (Optional)                | string  | User provided aws account                                                                                                                                                                                                                                                                        |
 | dataNodeStorage (Optional)        | string  | User provided ebs block storage size, defaults to 100Gb                                                                                                                                                                                                                                          |
 | mlNodeStorage (Optional)          | string  | User provided ebs block storage size, defaults to 100Gb                                                                                                                                                                                                                                          |
-
+| use50PercentHeap (Optional)       | boolean | Boolean flag to use 50% of physical memory as heap. Default is 1GB.  e.g., `--context use50PercentHeap=true`                                                                                                                                                                                                     |
 
 * Before starting this step, ensure that your AWS CLI is correctly configured with access credentials.
 * Also ensure that you're running these commands in the current directory

lib/infra/infra-stack.ts

@@ -55,6 +55,7 @@
     readonly mlNodeStorage: number,
     readonly jvmSysPropsString?: string,
     readonly additionalConfig?: string,
+    readonly use50PercentHeap: boolean,
 }
 
 export class InfraStack extends Stack {
@@ -111,7 +112,7 @@
     }
 
     if (props.singleNodeCluster) {
       console.log('Single node value is true, creating single node configurations');
       singleNodeInstance = new Instance(this, 'single-node-instance', {
         vpc: props.vpc,
         instanceType: ec2InstanceType,
@@ -407,7 +408,7 @@
 
       fileContent['cluster.name'] = `${scope.stackName}-${scope.account}-${scope.region}`;
 
       console.log(dump(fileContent).toString());
       opensearchConfig = dump(fileContent).toString();
       cfnInitConfig.push(InitCommand.shellCommand(`set -ex;cd opensearch; echo "${opensearchConfig}" > config/opensearch.yml`,
         {
@@ -468,6 +469,20 @@
       }));
     }
 
+    // Check if JVM Heap Memory is set. Default is 1G in the jvm.options file
+    // @ts-ignore
+    if (props.use50PercentHeap) {
+      cfnInitConfig.push(InitCommand.shellCommand(`set -ex; cd opensearch;
+      totalMem=\`expr $(free -g | awk '/^Mem:/{print $2}') + 1\`;
+      heapSizeInGb=\`expr $totalMem / 2\`;
+      if [ $heapSizeInGb -lt 32 ];then minHeap="-Xms"$heapSizeInGb"g";maxHeap="-Xmx"$heapSizeInGb"g";else minHeap="-Xms32g";maxHeap="-Xmx32g";fi
+      sed -i -e "s/^-Xms[0-9a-z]*$/$minHeap/g" config/jvm.options;
+      sed -i -e "s/^-Xmx[0-9a-z]*$/$maxHeap/g" config/jvm.options;`, {
+        cwd: '/home/ec2-user',
+        ignoreErrors: false,
+      }));
+    }
+
     // @ts-ignore
     if (props.additionalConfig.toString() !== 'undefined') {
       // @ts-ignore

lib/os-cluster-entrypoint.ts

@@ -148,6 +148,9 @@ export class OsClusterEntrypoint {
 
       const suffix = `${scope.node.tryGetContext('suffix')}`;
 
+      const use50heap = `${scope.node.tryGetContext('use50PercentHeap')}`;
+      const use50PercentHeap = use50heap === 'true';
+
       const network = new NetworkStack(scope, 'opensearch-network-stack', {
         cidrBlock: cidrRange,
         maxAzs: 3,
@@ -192,6 +195,7 @@ export class OsClusterEntrypoint {
         mlNodeStorage,
         jvmSysPropsString: jvmSysProps,
         additionalConfig: ymlConfig,
+        use50PercentHeap,
         ...props,
       });