@@ -61,6 +61,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
6161 .setFullMethodName ("google.datastore.v1.Datastore/Lookup" )
6262 .setRequestMarshaller (ProtoUtils .marshaller (LookupRequest .getDefaultInstance ()))
6363 .setResponseMarshaller (ProtoUtils .marshaller (LookupResponse .getDefaultInstance ()))
64+ .setSampledToLocalTracing (true )
6465 .build ();
6566
6667 private static final MethodDescriptor <RunQueryRequest , RunQueryResponse >
@@ -70,6 +71,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
7071 .setFullMethodName ("google.datastore.v1.Datastore/RunQuery" )
7172 .setRequestMarshaller (ProtoUtils .marshaller (RunQueryRequest .getDefaultInstance ()))
7273 .setResponseMarshaller (ProtoUtils .marshaller (RunQueryResponse .getDefaultInstance ()))
74+ .setSampledToLocalTracing (true )
7375 .build ();
7476
7577 private static final MethodDescriptor <RunAggregationQueryRequest , RunAggregationQueryResponse >
@@ -81,6 +83,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
8183 ProtoUtils .marshaller (RunAggregationQueryRequest .getDefaultInstance ()))
8284 .setResponseMarshaller (
8385 ProtoUtils .marshaller (RunAggregationQueryResponse .getDefaultInstance ()))
86+ .setSampledToLocalTracing (true )
8487 .build ();
8588
8689 private static final MethodDescriptor <BeginTransactionRequest , BeginTransactionResponse >
@@ -92,6 +95,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
9295 ProtoUtils .marshaller (BeginTransactionRequest .getDefaultInstance ()))
9396 .setResponseMarshaller (
9497 ProtoUtils .marshaller (BeginTransactionResponse .getDefaultInstance ()))
98+ .setSampledToLocalTracing (true )
9599 .build ();
96100
97101 private static final MethodDescriptor <CommitRequest , CommitResponse > commitMethodDescriptor =
@@ -100,6 +104,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
100104 .setFullMethodName ("google.datastore.v1.Datastore/Commit" )
101105 .setRequestMarshaller (ProtoUtils .marshaller (CommitRequest .getDefaultInstance ()))
102106 .setResponseMarshaller (ProtoUtils .marshaller (CommitResponse .getDefaultInstance ()))
107+ .setSampledToLocalTracing (true )
103108 .build ();
104109
105110 private static final MethodDescriptor <RollbackRequest , RollbackResponse >
@@ -109,6 +114,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
109114 .setFullMethodName ("google.datastore.v1.Datastore/Rollback" )
110115 .setRequestMarshaller (ProtoUtils .marshaller (RollbackRequest .getDefaultInstance ()))
111116 .setResponseMarshaller (ProtoUtils .marshaller (RollbackResponse .getDefaultInstance ()))
117+ .setSampledToLocalTracing (true )
112118 .build ();
113119
114120 private static final MethodDescriptor <AllocateIdsRequest , AllocateIdsResponse >
@@ -119,6 +125,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
119125 .setRequestMarshaller (ProtoUtils .marshaller (AllocateIdsRequest .getDefaultInstance ()))
120126 .setResponseMarshaller (
121127 ProtoUtils .marshaller (AllocateIdsResponse .getDefaultInstance ()))
128+ .setSampledToLocalTracing (true )
122129 .build ();
123130
124131 private static final MethodDescriptor <ReserveIdsRequest , ReserveIdsResponse >
@@ -128,6 +135,7 @@ public class GrpcDatastoreStub extends DatastoreStub {
128135 .setFullMethodName ("google.datastore.v1.Datastore/ReserveIds" )
129136 .setRequestMarshaller (ProtoUtils .marshaller (ReserveIdsRequest .getDefaultInstance ()))
130137 .setResponseMarshaller (ProtoUtils .marshaller (ReserveIdsResponse .getDefaultInstance ()))
138+ .setSampledToLocalTracing (true )
131139 .build ();
132140
133141 private final UnaryCallable <LookupRequest , LookupResponse > lookupCallable ;
0 commit comments