From ec8129f06e1009beb4c261a362e805e81bbcc15c Mon Sep 17 00:00:00 2001
From: Yecheng Fu <cofyc.jackson@gmail.com>
Date: Wed, 21 Aug 2019 21:39:07 +0800
Subject: [PATCH] using real node locations

---
 deploy/modules/gcp/tidb-cluster/data.tf | 11 ++++++++++-
 deploy/modules/gcp/tidb-cluster/main.tf |  4 ++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/deploy/modules/gcp/tidb-cluster/data.tf b/deploy/modules/gcp/tidb-cluster/data.tf
index c73e032c78..0efb7629ba 100644
--- a/deploy/modules/gcp/tidb-cluster/data.tf
+++ b/deploy/modules/gcp/tidb-cluster/data.tf
@@ -18,4 +18,13 @@ data "external" "monitor_port" {
   program    = ["bash", "-c", "kubectl --kubeconfig ${var.kubeconfig_path} get svc -n ${var.cluster_name} ${var.cluster_name}-grafana -o json | jq '.spec.ports | .[] | select( .name == \"grafana\") | {port: .port|tostring}'"]
 }
 
-data "google_compute_zones" "available" {}
+locals {
+  # TODO Update related code when node locations is avaiable in attributes of cluster resource.
+  cmd_get_cluster_locations = <<EOT
+gcloud --project ${var.gcp_project} container clusters list --filter='name=${var.gke_cluster_name}' --format='json[no-heading](locations)' --region ${var.gke_cluster_location} | jq '.[0] | .locations |= join(",")'
+EOT
+}
+
+data "external" "cluster_locations" {
+  program = ["bash", "-c", local.cmd_get_cluster_locations]
+}
diff --git a/deploy/modules/gcp/tidb-cluster/main.tf b/deploy/modules/gcp/tidb-cluster/main.tf
index b6dd957a25..e7c0bfef78 100644
--- a/deploy/modules/gcp/tidb-cluster/main.tf
+++ b/deploy/modules/gcp/tidb-cluster/main.tf
@@ -125,7 +125,7 @@ resource "google_container_node_pool" "monitor_pool" {
 }
 
 locals {
-  num_availability_zones = length(data.google_compute_zones.available.names)
+  num_availability_zones = length(split(",", data.external.cluster_locations.result["locations"]))
 }
 
 module "tidb-cluster" {
@@ -143,7 +143,7 @@ module "tidb-cluster" {
 
 resource "null_resource" "wait-lb-ip" {
   depends_on = [
-    google_container_node_pool.tidb_pool,
+    module.tidb-cluster
   ]
   provisioner "local-exec" {
     interpreter = ["bash", "-c"]