Merge branch 'main' into PR #3928 to update

Author: lqiu96

google-cloud-spanner/src/main/java/com/google/cloud/spanner/BuiltInMetricsProvider.java

@@ -32,6 +32,7 @@
 import com.google.cloud.opentelemetry.detection.AttributeKeys;
 import com.google.cloud.opentelemetry.detection.DetectedPlatform;
 import com.google.cloud.opentelemetry.detection.GCPPlatformDetector;
+import com.google.common.base.Strings;
 import com.google.common.hash.HashFunction;
 import com.google.common.hash.Hashing;
 import io.grpc.ManagedChannelBuilder;
@@ -43,11 +44,17 @@
 import io.opentelemetry.sdk.metrics.SdkMeterProvider;
 import io.opentelemetry.sdk.metrics.SdkMeterProviderBuilder;
 import io.opentelemetry.sdk.resources.Resource;
+import java.io.BufferedReader;
 import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
 import java.lang.management.ManagementFactory;
 import java.lang.reflect.Method;
+import java.net.HttpURLConnection;
 import java.net.InetAddress;
+import java.net.URL;
 import java.net.UnknownHostException;
+import java.nio.charset.StandardCharsets;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.UUID;
@@ -63,6 +70,10 @@ final class BuiltInMetricsProvider {
 
   private static String taskId;
 
+  private static String location;
+
+  private static final String default_location = "global";
+
   private OpenTelemetry openTelemetry;
 
   private BuiltInMetricsProvider() {}
@@ -95,6 +106,36 @@ OpenTelemetry getOrCreateOpenTelemetry(
     }
   }
 
+  // TODO: Remove when
+  // https://github.com/GoogleCloudPlatform/opentelemetry-operations-java/issues/421
+  //       has been fixed.
+  static boolean quickCheckIsRunningOnGcp() {
+    int timeout = 5000;
+    try {
+      timeout =
+          Integer.parseInt(System.getProperty("spanner.check_is_running_on_gcp_timeout", "5000"));
+    } catch (NumberFormatException ignore) {
+      // ignore
+    }
+    try {
+      URL url = new URL("http://metadata.google.internal/computeMetadata/v1/project/project-id");
+      HttpURLConnection connection = (HttpURLConnection) url.openConnection();
+      connection.setConnectTimeout(timeout);
+      connection.setRequestProperty("Metadata-Flavor", "Google");
+      if (connection.getResponseCode() == 200
+          && ("Google").equals(connection.getHeaderField("Metadata-Flavor"))) {
+        InputStream input = connection.getInputStream();
+        try (BufferedReader reader =
+            new BufferedReader(new InputStreamReader(input, StandardCharsets.UTF_8))) {
+          return !Strings.isNullOrEmpty(reader.readLine());
+        }
+      }
+    } catch (IOException ignore) {
+      // ignore
+    }
+    return false;
+  }
+
   void enableGrpcMetrics(
       InstantiatingGrpcChannelProvider.Builder channelProviderBuilder,
       String projectId,
@@ -171,14 +212,20 @@ static String generateClientHash(String clientUid) {
   }
 
   static String detectClientLocation() {
-    GCPPlatformDetector detector = GCPPlatformDetector.DEFAULT_INSTANCE;
-    DetectedPlatform detectedPlatform = detector.detectPlatform();
-    // All platform except GKE uses "cloud_region" for region attribute.
-    String region = detectedPlatform.getAttributes().get("cloud_region");
-    if (detectedPlatform.getSupportedPlatform() == GOOGLE_KUBERNETES_ENGINE) {
-      region = detectedPlatform.getAttributes().get(AttributeKeys.GKE_CLUSTER_LOCATION);
+    if (location == null) {
+      location = default_location;
+      if (quickCheckIsRunningOnGcp()) {
+        GCPPlatformDetector detector = GCPPlatformDetector.DEFAULT_INSTANCE;
+        DetectedPlatform detectedPlatform = detector.detectPlatform();
+        // All platform except GKE uses "cloud_region" for region attribute.
+        String region = detectedPlatform.getAttributes().get("cloud_region");
+        if (detectedPlatform.getSupportedPlatform() == GOOGLE_KUBERNETES_ENGINE) {
+          region = detectedPlatform.getAttributes().get(AttributeKeys.GKE_CLUSTER_LOCATION);
+        }
+        location = region == null ? location : region;
+      }
     }
-    return region == null ? "global" : region;
+    return location;
   }
 
   /**

google-cloud-spanner/src/test/java/com/google/cloud/spanner/connection/CredentialsServiceTest.java

@@ -25,8 +25,9 @@
 import com.google.auth.oauth2.ServiceAccountCredentials;
 import com.google.cloud.spanner.ErrorCode;
 import com.google.cloud.spanner.SpannerException;
-import java.io.FileInputStream;
+import java.io.File;
 import java.io.IOException;
+import java.nio.file.Files;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
@@ -35,9 +36,9 @@
 @RunWith(JUnit4.class)
 public class CredentialsServiceTest {
   private static final String FILE_TEST_PATH =
-      CredentialsServiceTest.class.getResource("test-key.json").getFile();
-  private static final String APP_DEFAULT_FILE_TEST_PATH =
-      CredentialsServiceTest.class.getResource("test-key-app-default.json").getFile();
+      CredentialsServiceTest.class.getResource("test-key.json").getPath();
+  private static final String SA_APP_DEFAULT_FILE_TEST_PATH =
+      CredentialsServiceTest.class.getResource("test-key-app-default.json").getPath();
 
   private static final String TEST_PROJECT_ID = "test-project";
   private static final String APP_DEFAULT_PROJECT_ID = "app-default-test-project";
@@ -49,7 +50,11 @@ public class CredentialsServiceTest {
         GoogleCredentials internalGetApplicationDefault() throws IOException {
           // Read application default credentials directly from a specific file instead of actually
           // fetching the default from the environment.
-          return GoogleCredentials.fromStream(new FileInputStream(APP_DEFAULT_FILE_TEST_PATH));
+          return ServiceAccountCredentials.fromStream(
+              // Calling `getResource().getPath()` on Windows returns a string that might start with
+              // something like `/C:/...`. Paths.get() interprets the leading / as part of the path
+              // and would be invalid. Use `new File().toPath()` to read from these files.
+              Files.newInputStream(new File(SA_APP_DEFAULT_FILE_TEST_PATH).toPath()));
         }
       };
 

google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITVPCNegativeTest.java

@@ -48,7 +48,6 @@
 import com.google.common.base.Strings;
 import com.google.longrunning.OperationsClient;
 import com.google.longrunning.OperationsSettings;
-import java.io.FileInputStream;
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Collections;
@@ -352,9 +351,7 @@ public void deniedListBackupOperations() throws IOException {
                 .setTransportChannelProvider(InstantiatingGrpcChannelProvider.newBuilder().build())
                 .setEndpoint("spanner.googleapis.com:443")
                 .setCredentialsProvider(
-                    FixedCredentialsProvider.create(
-                        GoogleCredentials.fromStream(
-                            new FileInputStream(System.getenv("GOOGLE_APPLICATION_CREDENTIALS")))))
+                    FixedCredentialsProvider.create(GoogleCredentials.getApplicationDefault()))
                 .build())) {
       client.listOperations(backupId.getName() + "/operations", "");
       fail("Expected PermissionDeniedException");