From 46994a689028ef87515175a29257ee3e7ff2bd40 Mon Sep 17 00:00:00 2001
From: Wojciech Tyczynski <wojtekt@google.com>
Date: Fri, 11 Mar 2016 09:45:22 +0100
Subject: [PATCH] Fix AWS default master sizes

---
 cluster/aws/config-default.sh | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/cluster/aws/config-default.sh b/cluster/aws/config-default.sh
index 69c0856d860..d3a240e67a1 100644
--- a/cluster/aws/config-default.sh
+++ b/cluster/aws/config-default.sh
@@ -31,12 +31,22 @@ if [[ -z ${NODE_SIZE} ]]; then
 fi
 
 # Dynamically set the master size by the number of nodes, these are guesses
-# TODO: gather some data
 if [[ -z ${MASTER_SIZE} ]]; then
-  if (( ${NUM_NODES} < 150 )); then
-    MASTER_SIZE="m3.medium"
-  else
-    MASTER_SIZE="m3.large"
+  MASTER_SIZE="m3.medium"
+  if [[ "${NUM_NODES}" -gt "5" ]]; then
+    suggested_master_size="m3.large"
+  fi
+  if [[ "${NUM_NODES}" -gt "10" ]]; then
+    suggested_master_size="m3.xlarge"
+  fi
+  if [[ "${NUM_NODES}" -gt "100" ]]; then
+    suggested_master_size="m3.2xlarge"
+  fi
+  if [[ "${NUM_NODES}" -gt "250" ]]; then
+    suggested_master_size="c4.4xlarge"
+  fi
+  if [[ "${NUM_NODES}" -gt "500" ]]; then
+    suggested_master_size="c4.8xlarge"
   fi
 fi