From 08b3ed5792ef751296c6da3ce6316c643022fd4c Mon Sep 17 00:00:00 2001
From: Min Zhang <minzhangcmu@gmail.com>
Date: Tue, 25 Sep 2018 14:16:18 -0700
Subject: [PATCH] fix: add turbo cpu/ram config

---
 README.md          | 2 ++
 index.js           | 7 +++++++
 test/index.test.js | 4 ++++
 3 files changed, 13 insertions(+)

diff --git a/README.md b/README.md
index d3f8f86..2cf0cd8 100644
--- a/README.md
+++ b/README.md
@@ -24,9 +24,11 @@ The class provides a couple options that are configurable in the instantiation o
 | config.prefix | String | '' |Prefix to container names ("") |
 | config.jobsNamespace | String | 'default' | Kubernetes namespace where builds are running on |
 | config.baseImage | String | '' | Base image used to start the VM |
+| config.kubernetes.resources.memory.turbo | Number | 16 | Value for TURBO memory (in GB) |
 | config.kubernetes.resources.memory.high | Number | 12 | Value for HIGH memory (in GB) |
 | config.kubernetes.resources.memory.low | Number | 2 | Value for LOW memory (in GB) |
 | config.kubernetes.resources.memory.micro | Number | 1 | Value for MICRO memory (in GB) |
+| config.kubernetes.resources.cpu.turbo | Number | 12 | Value for TURBO CPU (in cores) |
 | config.kubernetes.resources.cpu.high | Number | 6 | Value for HIGH CPU (in cores) |
 | config.kubernetes.resources.cpu.low | Number | 2 | Value for LOW CPU (in cores) |
 | config.kubernetes.resources.cpu.micro | Number | 1 | Value for MICRO CPU (in cores) |
diff --git a/index.js b/index.js
index e2befae..9ce283a 100644
--- a/index.js
+++ b/index.js
@@ -115,9 +115,11 @@ class K8sVMExecutor extends Executor {
      * @param  {String} [options.kubernetes.baseImage]                Base image for the pod
      * @param  {Number} [options.kubernetes.buildTimeout=90]          Number of minutes to allow a build to run before considering it is timed out
      * @param  {Number} [options.kubernetes.maxBuildTimeout=120]      Max timeout user can configure up to (in minutes)
+     * @param  {String} [options.kubernetes.resources.cpu.turbo=12]   Value for TURBO CPU (in cores)
      * @param  {String} [options.kubernetes.resources.cpu.high=6]     Value for HIGH CPU (in cores)
      * @param  {Number} [options.kubernetes.resources.cpu.low=2]      Value for LOW CPU (in cores)
      * @param  {Number} [options.kubernetes.resources.cpu.micro=1]    Value for MICRO CPU (in cores)
+     * @param  {Number} [options.kubernetes.resources.memory.turbo=16]Value for TURBO memory (in GB)
      * @param  {Number} [options.kubernetes.resources.memory.high=12] Value for HIGH memory (in GB)
      * @param  {Number} [options.kubernetes.resources.memory.low=2]   Value for LOW memory (in GB)
      * @param  {Number} [options.kubernetes.resources.memory.micro=1] Value for MICRO memory (in GB)
@@ -150,9 +152,12 @@ class K8sVMExecutor extends Executor {
         this.maxBuildTimeout = this.kubernetes.maxBuildTimeout || MAX_BUILD_TIMEOUT;
         this.podsUrl = `https://${this.host}/api/v1/namespaces/${this.jobsNamespace}/pods`;
         this.breaker = new Fusebox(requestretry, options.fusebox);
+        this.turboCpu = hoek.reach(options, 'kubernetes.resources.cpu.turbo', { default: 12 });
         this.highCpu = hoek.reach(options, 'kubernetes.resources.cpu.high', { default: 6 });
         this.lowCpu = hoek.reach(options, 'kubernetes.resources.cpu.low', { default: 2 });
         this.microCpu = hoek.reach(options, 'kubernetes.resources.cpu.micro', { default: 1 });
+        this.turboMemory = hoek.reach(options,
+            'kubernetes.resources.memory.turbo', { default: 16 });
         this.highMemory = hoek.reach(options, 'kubernetes.resources.memory.high', { default: 12 });
         this.lowMemory = hoek.reach(options, 'kubernetes.resources.memory.low', { default: 2 });
         this.microMemory = hoek.reach(options, 'kubernetes.resources.memory.micro', { default: 1 });
@@ -208,6 +213,7 @@ class K8sVMExecutor extends Executor {
             hoek.reach(config, 'annotations', { default: {} }));
         const cpuConfig = annotations[CPU_RESOURCE];
         const cpuValues = {
+            TURBO: this.turboCpu,
             HIGH: this.highCpu,
             LOW: this.lowCpu,
             MICRO: this.microCpu
@@ -215,6 +221,7 @@ class K8sVMExecutor extends Executor {
         const CPU = (cpuConfig in cpuValues) ? cpuValues[cpuConfig] : cpuValues.LOW;
 
         const memValues = {
+            TURBO: this.turboMemory,
             HIGH: this.highMemory,
             LOW: this.lowMemory,
             MICRO: this.microMemory
diff --git a/test/index.test.js b/test/index.test.js
index 3c407a8..82bc53b 100644
--- a/test/index.test.js
+++ b/test/index.test.js
@@ -218,11 +218,13 @@ describe('index', () => {
                 baseImage: 'hyperctl',
                 resources: {
                     cpu: {
+                        turbo: 10,
                         high: 8,
                         low: 2,
                         micro: 1
                     },
                     memory: {
+                        turbo: 20,
                         high: 5,
                         low: 2,
                         micro: 1
@@ -240,9 +242,11 @@ describe('index', () => {
         assert.equal(executor.host, 'kubernetes2');
         assert.equal(executor.launchVersion, 'v1.2.3');
         assert.equal(executor.jobsNamespace, 'baz');
+        assert.equal(executor.turboCpu, 10);
         assert.equal(executor.highCpu, 8);
         assert.equal(executor.lowCpu, 2);
         assert.equal(executor.microCpu, 1);
+        assert.equal(executor.turboMemory, 20);
         assert.equal(executor.highMemory, 5);
         assert.equal(executor.lowMemory, 2);
         assert.equal(executor.microMemory, 1);