All Downloads are FREE. Search and download functionalities are using the official Maven repository.

net.dongliu.prettypb.rpc.client.RpcClientHandler Maven / Gradle / Ivy

Go to download

proto rpc libs, compatible with proto-rpc-pro

There is a newer version: 0.3.5
Show newest version
/**
 *   Copyright 2010-2014 Peter Klauser
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */
package net.dongliu.prettypb.rpc.client;

import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageDecoder;
import net.dongliu.prettypb.rpc.common.TaskCallBack;
import net.dongliu.prettypb.rpc.common.TaskSet;
import net.dongliu.prettypb.rpc.exception.ServiceException;
import net.dongliu.prettypb.rpc.listener.TcpConnectionEventListener;
import net.dongliu.prettypb.rpc.protocol.*;
import net.dongliu.prettypb.runtime.ExtensionRegistry;
import net.dongliu.prettypb.runtime.ProtoBufDecoder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;


/**
 * Handles returning RpcResponse and RpcError messages in the IO-Layer
 *
 * @author Peter Klauser
 * @author Dong Liu
 */
public class RpcClientHandler extends MessageToMessageDecoder {

    private static Logger logger = LoggerFactory.getLogger(RpcClientHandler.class);
    private final RpcClientChannel rpcClientChannel;

    private TcpConnectionEventListener eventListener;

    private final TaskSet taskSet = new TaskSet<>();

    private final ExtensionRegistry extensionRegistry;

    public RpcClientHandler(TcpConnectionEventListener eventListener,
                            RpcClientChannel rpcClientChannel,
                            ExtensionRegistry extensionRegistry) {
        this.extensionRegistry = extensionRegistry;
        if (eventListener == null) {
            throw new NullPointerException("eventListener is null");
        }
        this.eventListener = eventListener;
        this.rpcClientChannel = rpcClientChannel;
    }

    @Override
    protected void decode(ChannelHandlerContext ctx, WirePayload payload, List out)
            throws Exception {
        if (payload.hasRpcResponse()) {
            onResponse(payload.getRpcResponse());
        } else if (payload.hasRpcError()) {
            onError(payload.getRpcError());
        } else if (payload.hasOobResponse()) {
            onOobResponse(payload.getOobResponse());
        } else if (payload.hasOobMessage()) {
            onOobMessage(payload.getOobMessage());
        } else if (payload.hasTransparentMessage()) {
            // just so that it's not forgotten sometime...
            out.add(payload);
        } else {
            // rpcRequest, rpcCancel, clientMessage go further up to the RpcServerHandler
            // transparentMessage are also sent up but not handled anywhere explicitly
            out.add(payload);
        }
    }

    /**
     * receive rpc response
     *
     * @param rpcResponse
     */
    public void onResponse(final RpcResponse rpcResponse) {
        taskSet.consume(rpcResponse.getCorrelationId(), new TaskCallBack() {
            @Override
            public void onTask(ClientCallTask task) {
                try {
                    Object response = ProtoBufDecoder.fromBytes(task.getMethodInfo().getResponseType(),
                            rpcResponse.getResponseBytes(), extensionRegistry);
                    task.handleResponse(response);
                } catch (RuntimeException e) {
                    task.handleFailure(e);
                }
            }
        });
    }

    /**
     * Receipt of an RpcError reply from a remote Peer.
     *
     * @param rpcError
     */
    public void onError(final RpcError rpcError) {
        taskSet.consume(rpcError.getCorrelationId(), new TaskCallBack() {
            @Override
            public void onTask(ClientCallTask task) throws Exception {
                task.handleFailure(new ServiceException(rpcError.getErrorMessage()));
            }
        });
    }


    /**
     * For use by RpcClientHandler to dispatch an Out-of-Band server response
     * message to client code.
     *
     * @param serverMessage
     */
    public void onOobResponse(OobResponse serverMessage) {
        taskSet.consume(serverMessage.getCorrelationId(), new TaskCallBack() {
            @Override
            public void onTask(ClientCallTask task) throws Exception {
                //TODO: to be implemented
            }
        });
        logger.debug("oob response received");
    }

    public void onOobMessage(OobMessage oobMessage) {
        //TODO: to be implemented
        logger.debug("oob message received");
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        super.channelInactive(ctx);
        onClosed();
        notifyClosed();
    }

    private void onClosed() {
        taskSet.close();
    }

    public void notifyClosed() {
        eventListener.connectionClosed(rpcClientChannel);
    }

    public void notifyOpened() {
        eventListener.connectionOpened(rpcClientChannel);
    }

    public boolean registerTask(ClientCallTask task) {
        return taskSet.add(task);
    }

}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy