Set connect/read timeout when calling the metadata service

Otherwise callers of these methods can be waiting indefinitely. googleapis#4398
dmichel1 · Jan 26, 2019 · 88e049b · 88e049b
1 parent f4bac2f
commit 88e049b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/google-cloud-clients/google-cloud-core/src/main/java/com/google/cloud/MetadataConfig.java b/google-cloud-clients/google-cloud-core/src/main/java/com/google/cloud/MetadataConfig.java
@@ -34,6 +34,7 @@
 public class MetadataConfig {
 
   private static final String METADATA_URL = "http://metadata.google.internal/computeMetadata/v1/";
+  private static final int TIMEOUT_MS = 5000;
 
   private MetadataConfig() {}
 
@@ -69,6 +70,8 @@ public static String getAttribute(String attributeName) {
     try {
       URL url = new URL(METADATA_URL + attributeName);
       HttpURLConnection connection = (HttpURLConnection) url.openConnection();
+      connection.setConnectTimeout(TIMEOUT_MS);
+      connection.setReadTimeout(TIMEOUT_MS);
       connection.setRequestProperty("Metadata-Flavor", "Google");
       InputStream input = connection.getInputStream();
       if (connection.getResponseCode() == 200) {