Merge pull request #97 from ltybc-coder/rootio-configure-port

tinkerbell · Mar 13, 2024 · 3536dca · 3536dca
2 parents eb6fa85 + 29c0423
commit 3536dca
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rootio/storage/metadata.go b/rootio/storage/metadata.go
@@ -86,12 +86,16 @@ func RetrieveData() (*Metadata, error) {
 	if metadataURL == "" {
 		return nil, errors.New("unable to discover the metadata server from environment variable [MIRROR_HOST]")
 	}
+	port, exist := os.LookupEnv("METADATA_SERVICE_PORT")
+	if !exist {
+		port = "50061"
+	}
 
 	metadataClient := http.Client{
 		Timeout: time.Second * 60, // Timeout after 60 seconds (seems massively long is this dial-up?)
 	}
 
-	req, err := http.NewRequestWithContext(context.TODO(), http.MethodGet, fmt.Sprintf("http://%s:50061/metadata", metadataURL), nil)
+	req, err := http.NewRequestWithContext(context.TODO(), http.MethodGet, fmt.Sprintf("http://%s:%s/metadata", metadataURL, port), nil)
 	if err != nil {
 		return nil, err
 	}