stamp infra vnet.

Author: sivakami

.pipelines/swiftv2-long-running/scripts/create_aks.sh

@@ -48,6 +48,26 @@ for i in $(seq 1 "$CLUSTER_COUNT"); do
   echo " - waiting for AKS provisioning state..."
   wait_for_provisioning "$RG" "$CLUSTER_NAME"
 
+  echo "Retrieving VNET ID from AKS cluster..."
+
+  node_subnet=$(az aks show \
+    --resource-group "$RG" \
+    --name "$CLUSTER_NAME" \
+    --query "agentPoolProfiles[0].vnetSubnetID" \
+    -o tsv)
+
+  if [[ -z "$node_subnet" ]]; then
+    echo "ERROR: Could not retrieve AKS node subnet ID"
+    exit 1
+  fi
+
+  vnet_id="${node_subnet%/subnets/*}"
+
+  echo "Found VNET: $vnet_id"
+
+  # Call your stamp function
+  stamp_vnet "$vnet_id"
+
   echo "Adding multi-tenant nodepool ' to '$CLUSTER_NAME'"
   make -C ./hack/aks linux-swiftv2-nodepool-up \
   AZCLI=az REGION=$LOCATION \
@@ -67,5 +87,37 @@ for i in $(seq 1 "$CLUSTER_COUNT"); do
     --file "$KUBECONFIG_PATH"
   echo "Kubeconfig saved: ${KUBECONFIG_PATH}"
 
+#  Stamp infra vnet. 
+function stamp_vnet() {
+    local vnet_id
+    vnet_id="${1}"
+
+    responseFile="response.txt"
+    modified_vnet="${vnet_id//\//%2F}"
+    cmd_stamp_curl="'curl -v -X PUT http://localhost:8080/VirtualNetwork/$modified_vnet/stampcreatorservicename'"
+    cmd_containerapp_exec="az containerapp exec -n subnetdelegator-westus-u3h4j -g subnetdelegator-westus --subscription 9b8218f9-902a-4d20-a65c-e98acec5362f --command $cmd_stamp_curl"
+    
+    max_retries=6
+    sleep_seconds=15
+    retry_count=0
+    while [[ $retry_count -lt $max_retries ]]; do
+        script --quiet -c "$cmd_containerapp_exec" "$responseFile"
+        if grep -qF "200 OK" "$responseFile"; then
+            echo "Subnet Delegator successfully stamped the vnet"
+            return 0
+        else
+            echo "Subnet Delegator failed to stamp the vnet, attempt $((retry_count+1))"
+            cat "$responseFile"
+            retry_count=$((retry_count+1))
+            sleep "$sleep_seconds"
+        fi
+    done
+    if [[ $retry_count -eq $max_retries ]]; then
+        echo "Failed to stamp the vnet even after $max_retries attempts"
+        exit 1
+    fi
+}
+
+
 done
 echo "All done. Created $CLUSTER_COUNT cluster set(s)."