From 9d30bdec1c4d7ade03748595a6488583ce2f8948 Mon Sep 17 00:00:00 2001
From: Pahulpreet Singh <54016648+codelixir@users.noreply.github.com>
Date: Fri, 26 Dec 2025 17:24:20 +0530
Subject: [PATCH] Handle missing dataproc-role metadata in hive-lineage.sh

Replaces the `curl -f` call to the metadata server with the safer
`/usr/share/google/get_metadata_value` utility to retrieve the
`dataproc-role` attribute.

This change ensures the initialization action completes successfully even if the metadata key is not yet present, by using the standard Dataproc metadata helper and providing a fallback.
---
 hive-lineage/hive-lineage.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hive-lineage/hive-lineage.sh b/hive-lineage/hive-lineage.sh
index 84da22515..5a01d114e 100644
--- a/hive-lineage/hive-lineage.sh
+++ b/hive-lineage/hive-lineage.sh
@@ -52,7 +52,8 @@ function install_jars() {
 }
 
 function restart_hive_server2_master() {
-  ROLE=$(curl -f -s -H Metadata-Flavor:Google http://metadata/computeMetadata/v1/instance/attributes/dataproc-role)
+  # Safely get metadata without failing the script if the key is missing
+  ROLE=$(/usr/share/google/get_metadata_value attributes/dataproc-role || echo "Unknown")
   if [[ "${ROLE}" == 'Master' ]]; then
     echo "Restarting hive-server2"
     sudo systemctl restart hive-server2.service