feat: add OpenTelemetry tracing (googleapis#1568)

* feat: add OpenTelemetry tracing * chore: add copyright headers * chore: add deps + reset tracing for test * fix: ClassCastException in Spring Data JDBC sample * docs: add sample for OpenTelemetry * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * docs: update sample for OpenTelemetry * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * test: check for credentials before setting up tracing * chore: cleanup * chore: add enable flag for otel to sample app * chore: add clirr diff * fix: add tracing prefix + more samples * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * chore: remove whitespace * docs: update sample to use non-snapshot versions * docs: update sample + add debugging guide * chore: cleanup * chore: actually use properties in example * docs: add link to tag sample * docs: improve documentation based on review comments * docs: clearify the names of the Spanner generated traces --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
diegomarquezp · Jun 27, 2024 · 1485a04 · 1485a04
1 parent 7191f0a
commit 1485a04
Show file tree

Hide file tree

Showing 27 changed files with 1,267 additions and 74 deletions.
diff --git a/.readme-partials.yaml b/.readme-partials.yaml
@@ -26,7 +26,7 @@ custom_content: |
    
    ### Connection URL Properties
    
-   The Cloud Spanner JDBC driver supports the following connection URL properties. Note that all of
+   The Spanner JDBC driver supports the following connection URL properties. Note that all of
    these can also be supplied in a Properties instance that is passed to the
    `DriverManager#getConnection(String url, Properties properties)` method.
 
@@ -37,6 +37,8 @@ custom_content: |
    - autoConfigEmulator (boolean): Automatically configure the connection to try to connect to the Cloud Spanner emulator. You do not need to specify any host or port in the connection string as long as the emulator is running on the default host/port (localhost:9010). The instance and database in the connection string will automatically be created if these do not yet exist on the emulator. This means that you do not need to execute any `gcloud` commands on the emulator to create the instance and database before you can connect to it. Example: `jdbc:cloudspanner:/projects/test-project/instances/test-instance/databases/test-db;autoConfigEmulator=true`
    - usePlainText (boolean): Sets whether the JDBC connection should establish an unencrypted connection to a (local) server. This option can only be used when connecting to a local emulator that does not require an encrypted connection, and that does not require authentication. Example: `jdbc:cloudspanner://localhost:9010/projects/test-project/instances/test-instance/databases/test-db;usePlainText=true`
    - optimizerVersion (String): Sets the default query optimizer version to use for this connection. See also https://cloud.google.com/spanner/docs/query-optimizer/query-optimizer-versions.
+   - enableExtendedTracing (boolean): Enables extended OpenTelemetry tracing of queries that are executed by a JDBC connection. When enabled, the SQL string of the query that is executed is added as a property to the trace.
+   - enableApiTracing (boolean): Enables API OpenTelemetry tracing of all RPCs that are executed by the JDBC driver. When enabled, a trace will be created for each RPC invocation that is executed by the JDBC driver. Enable this to investigate latency problems and/or RPCs that are being retried.
    
    #### Advanced Properties
    - minSessions (int): Sets the minimum number of sessions in the backing session pool. Defaults to 100.
@@ -46,6 +48,88 @@ custom_content: |
    - oauthToken (string): A valid pre-existing OAuth token to use for authentication for this connection. Setting this property will take precedence over any value set for a credentials file.
    - lenient (boolean): Enable this to force the JDBC driver to ignore unknown properties in the connection URL. Some applications automatically add additional properties to the URL that are not recognized by the JDBC driver. Normally, the JDBC driver will reject this, unless `lenient` mode is enabled.
    
+   ### OpenTelemetry Tracing
+   The Spanner JDBC driver supports OpenTelemetry tracing. You can configure the OpenTelemetry instance
+   that should be used in two ways:
+   1. Register a global OpenTelemetry instance. This instance will automatically be picked up by the Spanner JDBC driver.
+   2. Add an OpenTelemetry instance with the key `openTelemetry` to the `java.util.Properties` instance that is used to create the JDBC connection.
+   
+   By default, the traces that are generated by the Spanner JDBC driver do not include the SQL
+   statement. You can include the SQL statement with the traces by adding the property `enableExtendedTracing=true`
+   to the JDBC connection URL.
+   
+   #### Example Using Global OpenTelemetry
+   Create and register a global OpenTelemetry object before creating a JDBC connection.
+   See also the [Spring Data JDBC Sample](samples/spring-data-jdbc) for an example for how to
+   configure OpenTelemetry in combination with Spring Data.
+   
+   See [Latency Debugging Guide](documentation/latency-debugging-guide.md) for more information on how to use these traces.
+
+   ```java
+   TraceConfiguration traceConfiguration = TraceConfiguration.builder().setProjectId("my-project").build();
+   SpanExporter traceExporter = TraceExporter.createWithConfiguration(traceConfiguration);
+   OpenTelemetry openTelemetry =
+        OpenTelemetrySdk.builder()
+            .setTracerProvider(
+                SdkTracerProvider.builder()
+                    .setSampler(Sampler.traceIdRatioBased(0.05))
+                    .setResource(
+                        Resource.builder()
+                            .put("service.name", "my-unique-service-name")
+                            .build())
+                    .addSpanProcessor(BatchSpanProcessor.builder(traceExporter).build())
+                    .build())
+            .buildAndRegisterGlobal();
+   String projectId = "my-project";
+   String instanceId = "my-instance";
+   String databaseId = "my-database";
+   // Setting this to true instructs the JDBC driver to include the SQL statement with the traces.
+   boolean enableExtendedTracing = true;
+   // Enabling API tracing includes traces for each individual RPC invocation, including retries.
+   boolean enableApiTracing = true;
+   
+   try (Connection connection =
+       DriverManager.getConnection(
+           String.format(
+               "jdbc:cloudspanner:/projects/%s/instances/%s/databases/%s?enableExtendedTracing=%s;enableApiTracing=%s",
+               projectId, instanceId, databaseId, enableExtendedTracing, enableApiTracing))) {
+     try (Statement statement = connection.createStatement()) {
+       try (ResultSet rs = statement.executeQuery("SELECT CURRENT_TIMESTAMP()")) {
+         while (rs.next()) {
+           System.out.printf(
+               "Connected to Cloud Spanner at [%s]%n", rs.getTimestamp(1).toString());
+         }
+       }
+     }
+   }
+   ```
+   
+   #### Example Using an OpenTelemetry instance in Properties
+   Instead of registering a global `OpenTelemetry` object, you can also provide a specific instance
+   in the properties for a JDBC connection:
+   
+   ```java
+   Properties info = new Properties();
+   info.put(JdbcDriver.OPEN_TELEMETRY_PROPERTY_KEY, openTelemetry);
+   info.put("enableExtendedTracing", String.valueOf(enableExtendedTracing));
+   info.put("enableApiTracing", String.valueOf(enableApiTracing));
+   
+   try (Connection connection =
+       DriverManager.getConnection(
+           String.format(
+               "jdbc:cloudspanner:/projects/%s/instances/%s/databases/%s",
+               projectId, instanceId, databaseId))) {
+     try (Statement statement = connection.createStatement()) {
+       try (ResultSet rs = statement.executeQuery("SELECT CURRENT_TIMESTAMP()")) {
+         while (rs.next()) {
+           System.out.printf(
+               "Connected to Cloud Spanner at [%s]%n", rs.getTimestamp(1).toString());
+         }
+       }
+     }
+   }
+   ```
+   
    ### Jar with Dependencies
    A single jar with all dependencies can be downloaded from https://repo1.maven.org/maven2/com/google/cloud/google-cloud-spanner-jdbc/latest
    or be built with the command `mvn package` (select the jar that is named `google-cloud-spanner-jdbc-<version>-single-jar-with-dependencies.jar`).

diff --git a/documentation/img/example-aborted-transaction.png b/documentation/img/example-aborted-transaction.png
diff --git a/documentation/img/example-api-tracing.png b/documentation/img/example-api-tracing.png
diff --git a/documentation/img/example-gfe-latency.png b/documentation/img/example-gfe-latency.png
diff --git a/documentation/img/example-search-for-aborted-transactions.png b/documentation/img/example-search-for-aborted-transactions.png
diff --git a/documentation/img/example-search-for-tag.png b/documentation/img/example-search-for-tag.png
diff --git a/documentation/img/example-tracing.png b/documentation/img/example-tracing.png