Merge pull request #36 from robertchoi80/main

add validation for numOfAz
openinfradev · Mar 23, 2022 · e752470 · e752470
2 parents 82e7ad5 + c6e8d7f
commit e752470
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cmd/server/handlers.go b/cmd/server/handlers.go
@@ -15,7 +15,7 @@ import (
 )
 
 var (
-	filePathAzRegion  = "./az-per-region.txt"
+	filePathAzRegion = "./az-per-region.txt"
 )
 
 const MAX_SIZE_PER_AZ = 99
@@ -75,6 +75,12 @@ func constructClusterConf(rawConf *pb.ClusterRawConf) (clusterConf *pb.ClusterCo
 	numOfAz := 3
 	if rawConf != nil && rawConf.NumOfAz != 0 {
 		numOfAz = int(rawConf.NumOfAz)
+
+		if numOfAz > 3 {
+			log.Error("Error: numOfAz cannot exceed 3.")
+			temp_err := fmt.Errorf("Error: numOfAz cannot exceed 3.")
+			return nil, temp_err
+		}
 	}
 
 	sshKeyName := "tks-seoul"
@@ -471,7 +477,6 @@ func (s *server) InstallAppGroups(ctx context.Context, in *pb.InstallAppGroupsRe
 			workflowTemplate = "tks-lma-federation"
 			parameters = append(parameters, "logging_component=efk")
 
-
 		case pb.AppGroupType_SERVICE_MESH:
 			workflowTemplate = "tks-service-mesh"