com.alipay.remoting.AsyncContext Java Examples

@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, Object request) {

    BoltChannel boltChannel = new BoltChannel();
    boltChannel.setAsyncContext(asyncCtx);
    boltChannel.setConnection(bizCtx.getConnection());
    try {
        if (userProcessorHandler != null) {
            userProcessorHandler.received(boltChannel, request);
        }
    } catch (Exception e) {
        LOGGER.error("Handle request error!", e);
        throw new RuntimeException("Handle request error!", e);
    }
}
 

@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, RequestBody request) {
    logger.warn("Request received:" + request);
    invokeTimes.incrementAndGet();
    if (!delaySwitch) {
        throw new RuntimeException("Hello exception!");
    }
    try {
        Thread.sleep(delayMs);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    throw new RuntimeException("Hello exception!");
}
 

/**
 * 初始化数据
 */
protected void init() {
    // 从ThreadLocal取出当前长连接，request等信息设置进去，需要接到请求时提前设置到ThreadLocal里
    RpcInternalContext context = RpcInternalContext.getContext();
    asyncContext = (AsyncContext) context.getAttachment(RpcConstants.HIDDEN_KEY_ASYNC_CONTEXT);
    request = (SofaRequest) context.getAttachment(RpcConstants.HIDDEN_KEY_ASYNC_REQUEST);
}
 

@Override
public void registerProcessor(final RpcProcessor processor) {
    this.rpcServer.registerUserProcessor(new AsyncUserProcessor<Object>() {

        @SuppressWarnings("unchecked")
        @Override
        public void handleRequest(final BizContext bizCtx, final AsyncContext asyncCtx, final Object request) {
            final RpcContext rpcCtx = new RpcContext() {

                @Override
                public void sendResponse(final Object responseObj) {
                    asyncCtx.sendResponse(responseObj);
                }

                @Override
                public Connection getConnection() {
                    com.alipay.remoting.Connection conn = bizCtx.getConnection();
                    if (conn == null) {
                        return null;
                    }
                    return new BoltConnection(conn);
                }

                @Override
                public String getRemoteAddress() {
                    return bizCtx.getRemoteAddress();
                }
            };

            processor.handleRequest(rpcCtx, request);
        }

        @Override
        public String interest() {
            return processor.interest();
        }

        @Override
        public ExecutorSelector getExecutorSelector() {
            final RpcProcessor.ExecutorSelector realSelector = processor.executorSelector();
            if (realSelector == null) {
                return null;
            }
            return realSelector::select;
        }

        @Override
        public Executor getExecutor() {
            return processor.executor();
        }
    });
}
 

@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, Object request) {
    super.handleRequest(bizCtx, asyncCtx, request);
}
 

/**
 * unsupported here!
 *
 * @see com.alipay.remoting.rpc.protocol.UserProcessor#handleRequest(com.alipay.remoting.BizContext, com.alipay.remoting.AsyncContext, java.lang.Object)
 */
@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, T request) {
    throw new UnsupportedOperationException(
        "ASYNC handle request is unsupported in SyncUserProcessor!");
}
 

/**
 * unsupported here!
 *
 * @see com.alipay.remoting.rpc.protocol.UserProcessor#handleRequest(com.alipay.remoting.BizContext, com.alipay.remoting.AsyncContext, java.lang.Object)
 */
@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, T request) {
    throw new UnsupportedOperationException(
        "ASYNC handle request is unsupported in SyncMultiInterestUserProcessor!");
}
 

@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, RequestBody request) {
    this.asyncExecutor.execute(new InnerTask(bizCtx, asyncCtx, request));
}
 

public InnerTask(BizContext bizCtx, AsyncContext asyncCtx, RequestBody request) {
    this.bizCtx = bizCtx;
    this.asyncCtx = asyncCtx;
    this.request = request;
}
 

@Override
public void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, RequestBody request) {
    this.asyncExecutor.execute(new InnerTask(asyncCtx, request));
}
 

public InnerTask(AsyncContext asyncCtx, RequestBody request) {
    this.asyncCtx = asyncCtx;
    this.request = request;
}
 

/**
 * Getter method for property <tt>asyncContext</tt>.
 *
 * @return property value of asyncContext
 */
public AsyncContext getAsyncContext() {
    return asyncContext;
}
 

/**
 * Setter method for property <tt>asyncContext</tt>.
 *
 * @param asyncContext  value to be assigned to property asyncContext
 */
public void setAsyncContext(AsyncContext asyncContext) {
    this.asyncContext = asyncContext;
}
 

/**
 * @see com.alipay.remoting.rpc.protocol.UserProcessor#handleRequest(com.alipay.remoting.BizContext, com.alipay.remoting.AsyncContext, java.lang.Object)
 */
@Override
public abstract void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, T request);
 

/**
 * Handle request with {@link AsyncContext}.
 * @param bizCtx biz context
 * @param asyncCtx async context
 * @param request request
 */
void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, T request);
 

/**
 * @see com.alipay.remoting.rpc.protocol.UserProcessor#handleRequest(com.alipay.remoting.BizContext, com.alipay.remoting.AsyncContext, java.lang.Object)
 */
@Override
public abstract void handleRequest(BizContext bizCtx, AsyncContext asyncCtx, T request);