Add TraceSpans to tensorstore

TraceSpan adds scoped traces to tensorstore, however there
is no current integration with any other telemetry libraries.

LoggingTraceSpan incorporates TraceSpan along with scoped logging.

Also inspects Executor uses and updates some TraceContext propagation.

PiperOrigin-RevId: 700720206
Change-Id: I324c544b6588516e2c0de9a701d5e49cda726065

tensorstore/driver/BUILD

@@ -151,6 +151,7 @@ tensorstore_cc_library(
         "//tensorstore/internal/json_binding",
         "//tensorstore/internal/json_binding:bindable",
         "//tensorstore/internal/json_binding:data_type",
+        "//tensorstore/internal/tracing",
         "//tensorstore/kvstore",
         "//tensorstore/serialization",
         "//tensorstore/serialization:registry",

tensorstore/driver/copy.cc

@@ -35,6 +35,7 @@
 #include "tensorstore/internal/nditerable_data_type_conversion.h"
 #include "tensorstore/internal/nditerable_util.h"
 #include "tensorstore/internal/tagged_ptr.h"
+#include "tensorstore/internal/tracing/trace_span.h"
 #include "tensorstore/open_mode.h"
 #include "tensorstore/progress.h"
 #include "tensorstore/read_write_options.h"
@@ -140,6 +141,7 @@ struct CopyState : public internal::AtomicReferenceCount<CopyState> {
   Promise<void> copy_promise;
   Promise<void> commit_promise;
   IntrusivePtr<CommitState> commit_state{new CommitState};
+  internal_tracing::TraceSpan tspan{"tensorstore.Copy"};
 
   void SetError(absl::Status error) {
     SetDeferredResult(copy_promise, std::move(error));

tensorstore/driver/read.cc

@@ -42,6 +42,7 @@
 #include "tensorstore/internal/nditerable_transformed_array.h"
 #include "tensorstore/internal/nditerable_util.h"
 #include "tensorstore/internal/tagged_ptr.h"
+#include "tensorstore/internal/tracing/trace_span.h"
 #include "tensorstore/internal/type_traits.h"
 #include "tensorstore/open_mode.h"
 #include "tensorstore/progress.h"
@@ -50,6 +51,7 @@
 #include "tensorstore/resize_options.h"
 #include "tensorstore/transaction.h"
 #include "tensorstore/util/element_pointer.h"
+#include "tensorstore/util/execution/any_receiver.h"
 #include "tensorstore/util/execution/sender.h"
 #include "tensorstore/util/executor.h"
 #include "tensorstore/util/extents.h"
@@ -108,6 +110,7 @@ struct ReadState
   Promise<PromiseValue> promise;
   std::atomic<Index> copied_elements{0};
   Index total_elements;
+  internal_tracing::TraceSpan tspan{"tensorstore.Read"};
 
   void SetError(absl::Status error) {
     SetDeferredResult(promise, std::move(error));

tensorstore/driver/write.cc

@@ -36,13 +36,15 @@
 #include "tensorstore/internal/nditerable_transformed_array.h"
 #include "tensorstore/internal/nditerable_util.h"
 #include "tensorstore/internal/tagged_ptr.h"
+#include "tensorstore/internal/tracing/trace_span.h"
 #include "tensorstore/internal/type_traits.h"
 #include "tensorstore/open_mode.h"
 #include "tensorstore/progress.h"
 #include "tensorstore/read_write_options.h"
 #include "tensorstore/resize_options.h"
 #include "tensorstore/transaction.h"
 #include "tensorstore/util/element_pointer.h"
+#include "tensorstore/util/execution/any_receiver.h"
 #include "tensorstore/util/execution/sender.h"
 #include "tensorstore/util/executor.h"
 #include "tensorstore/util/future.h"
@@ -125,6 +127,7 @@ struct WriteState : public internal::AtomicReferenceCount<WriteState> {
   Promise<void> copy_promise;
   Promise<void> commit_promise;
   IntrusivePtr<CommitState> commit_state{new CommitState};
+  internal_tracing::TraceSpan tspan{"tensorstore.Write"};
 
   void SetError(absl::Status error) {
     SetDeferredResult(copy_promise, std::move(error));

tensorstore/internal/os/BUILD

@@ -73,6 +73,7 @@ tensorstore_cc_library(
         "//tensorstore/internal/log:verbose_flag",
         "//tensorstore/internal/metrics",
         "//tensorstore/internal/metrics:metadata",
+        "//tensorstore/internal/tracing",
         "//tensorstore/util:quote_string",
         "//tensorstore/util:result",
         "//tensorstore/util:status",