add requestContextMap

zuojunyuan · Feb 13, 2017 · 1797112 · 1797112
1 parent 71b4235
commit 1797112
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.../com/dianping/pigeon/remoting/provider/process/threadpool/RequestThreadPoolProcessor.java b/.../com/dianping/pigeon/remoting/provider/process/threadpool/RequestThreadPoolProcessor.java
@@ -182,7 +182,7 @@ public void doStop() {
 
     public Future<InvocationResponse> doProcessRequest(final InvocationRequest request,
                                                        final ProviderContext providerContext) {
-//        requestContextMap.put(request, providerContext);
+        requestContextMap.put(request, providerContext);
 
         doMonitorData(request, providerContext);
 
@@ -201,7 +201,7 @@ public InvocationResponse call() throws Exception {
                 } catch (Throwable t) {
                     logger.error("Process request failed with invocation handler, you should never be here.", t);
                 } finally {
-//                    requestContextMap.remove(request);
+                    requestContextMap.remove(request);
                 }
                 return null;
             }
@@ -213,7 +213,7 @@ public InvocationResponse call() throws Exception {
             providerContext.getTimeline().add(new TimePoint(TimePhase.T));
             return pool.submit(requestExecutor);
         } catch (RejectedExecutionException e) {
-//            requestContextMap.remove(request);
+            requestContextMap.remove(request);
             throw new RejectedException(getProcessorStatistics(pool), e);
         }