Fixed data race in GrpcProxyServer

Author: alex268

tests/common/src/main/java/tech/ydb/test/integration/docker/GrpcProxyServer.java

@@ -71,6 +71,9 @@ public void close() {
     }
 
     private static class CallProxy<ReqT, RespT> {
+        private final Lock clientCallLock = new ReentrantLock();
+        private final Lock serverCallLock = new ReentrantLock();
+
         final RequestProxy serverCallListener;
         final ResponseProxy clientCallListener;
 
@@ -80,7 +83,6 @@ private static class CallProxy<ReqT, RespT> {
         }
 
         private class RequestProxy extends ServerCall.Listener<ReqT> {
-            private final Lock clientCallLock = new ReentrantLock();
             private final ClientCall<ReqT, ?> clientCall;
             // Hold 'this' lock when accessing
             private boolean needToRequest;
@@ -101,8 +103,8 @@ public void onHalfClose() {
 
             @Override
             public void onMessage(ReqT message) {
-                clientCall.sendMessage(message);
                 clientCallLock.lock();
+                clientCall.sendMessage(message);
                 try {
                     if (clientCall.isReady()) {
                         clientCallListener.serverCall.request(1);
@@ -137,7 +139,6 @@ void onClientReady() {
         }
 
         private class ResponseProxy extends ClientCall.Listener<RespT> {
-            private final Lock serverCallLock = new ReentrantLock();
             private final ServerCall<?, RespT> serverCall;
             // Hold 'this' lock when accessing
             private boolean needToRequest;
@@ -158,8 +159,8 @@ public void onHeaders(Metadata headers) {
 
             @Override
             public void onMessage(RespT message) {
-                serverCall.sendMessage(message);
                 serverCallLock.lock();
+                serverCall.sendMessage(message);
                 try {
                     if (serverCall.isReady()) {
                         serverCallListener.clientCall.request(1);