googleapis / java-spanner

Apache License 2.0
64 stars 119 forks source link

Possibility to define grpc-timeout dynamically without interacting directly with GrpcCallContext #2321

Open dominity opened 1 year ago

dominity commented 1 year ago

Thanks for stopping by to let us know something could be better!

Is your feature request related to a problem? Please describe. My current system receives request from other system which defines deadline/finishby timeout which varies from request type to request type. I'd like to instruct spanner client to propagate this value to spanner using 'grpc-timeout' header. Currently it's possible to do this interacting directly with GrpcCallContext via SpannerOptions.CallContextConfigurator. It would be great to have it as part of SpannerStubSettings, so I don't need to provide my own implementation of SpannerOptions.CallContextConfigurator and attach it to GrpcContext each time it's needed.

Describe the solution you'd like Not sure of what would be the best approach here. I have one idea in mind: SpannerStubSettings.Builder already contains method applyToAllUnaryMethods(...). But it's invoked once when settings are build. applyDynamicallyToAllUnaryMethods(ApiFunction f) method could be introduced which causes invocation of provided ApiFunction on each method invocation (or limit it to specific methods only to move it to UnaryCallSettings level).

Describe alternatives you've considered Currently my system interact with GrpcCallContext directly via SpannerOptions.CallContextConfigurator:

    public static class SpannerGrpcCallContextConfigurator implements SpannerOptions.CallContextConfigurator {
        public <ReqT, RespT> ApiCallContext configure(ApiCallContext context, ReqT request,
                MethodDescriptor<ReqT, RespT> method) {
            if (method == SpannerGrpc.getStreamingReadMethod()) {
                 return GrpcCallContext.createDefault().withCallOptions(CallOptions.DEFAULT.withDeadlineAfter(/*some deadline*/));            
            }
            return null;
        }
    }

then spanner invocation is wrapped with custom grpc context:

        Context.current()
                .withValue(SpannerOptions.CALL_CONTEXT_CONFIGURATOR_KEY, new SpannerGrpcCallContextConfigurator())
                .run(() -> { /* spanner invocation here */ });
rahul2393 commented 1 year ago

@arpan14 can you please help triage?

psinghbay1 commented 2 months ago

Is Context.current() per server's grpc request or per thread basis? Say an app receives grpc call which may perform Spanner operations from thread executor. How does Context.current() ensures that the context is not shared across the threads?

rahul2393 commented 2 months ago

@dominity Does this answer your question here https://cloud.google.com/spanner/docs/statement-timeout#java?