com.taobao.remoting.impl.DefaultClient.java

//同步調(diào)用遠(yuǎn)程接口

public Object invokeWithSync(Object appRequest, RequestControl control) throws RemotingException, InterruptedException {

        byte protocol = getProtocol(control);

        if (!TRConstants.isValidProtocol(protocol)) {

            throw new RemotingException("Invalid serialization protocol [" + protocol + "] on invokeWithSync.");

        }

        ResponseFuture future = invokeWithFuture(appRequest, control);

        return future.get();  //獲取結(jié)果時(shí)讓當(dāng)前線(xiàn)程等待，ResponseFuture其實(shí)就是前面說(shuō)的callback

}

public ResponseFuture invokeWithFuture(Object appRequest, RequestControl control) {

         byte protocol = getProtocol(control);

         long timeout = getTimeout(control);

         ConnectionRequest request = new ConnectionRequest(appRequest);

         request.setSerializeProtocol(protocol);

         Callback2FutureAdapter adapter = new Callback2FutureAdapter(request);

         connection.sendRequestWithCallback(request, adapter, timeout);

         return adapter;

}

Callback2FutureAdapter implements ResponseFuture

public Object get() throws RemotingException, InterruptedException {

synchronized (this) {  // 旋鎖

   while (!isDone) {  // 是否有結(jié)果了

wait(); //沒(méi)結(jié)果是釋放鎖，讓當(dāng)前線(xiàn)程處于等待狀態(tài)

   }

}

if (errorCode == TRConstants.RESULT_TIMEOUT) {

   throw new TimeoutException("Wait response timeout, request["

   + connectionRequest.getAppRequest() + "].");

}

else if (errorCode > 0) {

   throw new RemotingException(errorMsg);

}

else {

   return appResp;

}

}

客戶(hù)端收到服務(wù)端結(jié)果后，回調(diào)時(shí)相關(guān)方法，即設(shè)置isDone = true并notifyAll()

public void handleResponse(Object _appResponse) {

         appResp = _appResponse; //將遠(yuǎn)程調(diào)用結(jié)果設(shè)置到callback中來(lái)

         setDone();

}

public void onRemotingException(int _errorType, String _errorMsg) {

         errorCode = _errorType;

         errorMsg = _errorMsg;

         setDone();

}

private void setDone() {

         isDone = true;

         synchronized (this) { //獲取鎖，因?yàn)榍懊鎤ait()已經(jīng)釋放了callback的鎖了

             notifyAll(); // 喚醒處于等待的線(xiàn)程

         }

}

com.taobao.remoting.impl.DefaultConnection.java

// 用來(lái)存放請(qǐng)求和回調(diào)的MAP

private final ConcurrentHashMap<Long, Object[]> requestResidents;

//發(fā)送消息出去

void sendRequestWithCallback(ConnectionRequest connRequest, ResponseCallback callback, long timeoutMs) {

         long requestId = connRequest.getId();

         long waitBegin = System.currentTimeMillis();

         long waitEnd = waitBegin + timeoutMs;

         Object[] queue = new Object[4];

         int idx = 0;

         queue[idx++] = waitEnd;

         queue[idx++] = waitBegin;   //用于記錄日志

         queue[idx++] = connRequest; //用于記錄日志

         queue[idx++] = callback;

         requestResidents.put(requestId, queue); // 記錄響應(yīng)隊(duì)列

         write(connRequest);

         // 埋點(diǎn)記錄等待響應(yīng)的Map的大小

         StatLog.addStat("TBRemoting-ResponseQueues", "size", requestResidents.size(),

                   1L);

}

public void write(final Object connectionMsg) {

//mina里的IoSession.write()發(fā)送消息

         WriteFuture writeFuture = ioSession.write(connectionMsg);

         // 注冊(cè)FutureListener，當(dāng)請(qǐng)求發(fā)送失敗后，能夠立即做出響應(yīng)

         writeFuture.addListener(new MsgWrittenListener(this, connectionMsg));

}

/**

* 在得到響應(yīng)后，刪除對(duì)應(yīng)的請(qǐng)求隊(duì)列，并執(zhí)行回調(diào)

* 調(diào)用者：MINA線(xiàn)程

*/

public void putResponse(final ConnectionResponse connResp) {

         final long requestId = connResp.getRequestId();

         Object[] queue = requestResidents.remove(requestId);

         if (null == queue) {

             Object appResp = connResp.getAppResponse();

             String appRespClazz = (null == appResp) ? "null" : appResp.getClass().getName();

             StringBuilder sb = new StringBuilder();

             sb.append("Not found response receiver for requestId=[").append(requestId).append("],");

             sb.append("from [").append(connResp.getHost()).append("],");

             sb.append("response type [").append(appRespClazz).append("].");

             LOGGER.warn(sb.toString());

             return;

         }

         int idx = 0;

         idx++;

         long waitBegin = (Long) queue[idx++];

         ConnectionRequest connRequest = (ConnectionRequest) queue[idx++];

         ResponseCallback callback = (ResponseCallback) queue[idx++];

         // ** 把回調(diào)任務(wù)交給業(yè)務(wù)提供的線(xiàn)程池執(zhí)行 **

         Executor callbackExecutor = callback.getExecutor();

         callbackExecutor.execute(new CallbackExecutorTask(connResp, callback));

         long duration = System.currentTimeMillis() - waitBegin; // 實(shí)際讀響應(yīng)時(shí)間

         logIfResponseError(connResp, duration, connRequest.getAppRequest());

}

CallbackExecutorTask

static private class CallbackExecutorTask implements Runnable {

         final ConnectionResponse resp;

         final ResponseCallback callback;

         final Thread createThread;

         CallbackExecutorTask(ConnectionResponse _resp, ResponseCallback _cb) {

             resp = _resp;

             callback = _cb;

             createThread = Thread.currentThread();

         }

         public void run() {

             // 預(yù)防這種情況：業(yè)務(wù)提供的Executor，讓調(diào)用者線(xiàn)程來(lái)執(zhí)行任務(wù)

             if (createThread == Thread.currentThread()

                       && callback.getExecutor() != DIYExecutor.getInstance()) {

                   StringBuilder sb = new StringBuilder();

                   sb.append("The network callback task [" + resp.getRequestId() + "] cancelled, cause:");

                   sb.append("Can not callback task on the network io thhread.");

                   LOGGER.warn(sb.toString());

                   return;

             }

             if (TRConstants.RESULT_SUCCESS == resp.getResult()) {

                   callback.handleResponse(resp.getAppResponse()); //設(shè)置調(diào)用結(jié)果

             }

             else {

                   callback.onRemotingException(resp.getResult(), resp

                            .getErrorMsg());  //處理調(diào)用異常

             }

         }

}