net.dubboclub.protocol.akka.AkkaInvoker Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of akka-protocol Show documentation

The newest version!

package net.dubboclub.protocol.akka;

import com.alibaba.dubbo.common.URL;
import com.alibaba.dubbo.remoting.RemotingException;
import com.alibaba.dubbo.remoting.exchange.ResponseFuture;
import com.alibaba.dubbo.rpc.Invocation;
import com.alibaba.dubbo.rpc.Invoker;
import com.alibaba.dubbo.rpc.Result;
import com.alibaba.dubbo.rpc.RpcException;
import com.alibaba.dubbo.rpc.protocol.AbstractInvoker;
import net.dubboclub.akka.remoting.actor.AkkaFuture;
import net.dubboclub.akka.remoting.actor.BasicActor;

/**
 * Created by bieber on 2015/7/8.
 */
public class AkkaInvoker implements Invoker {
    
    private BasicActor actor;

    private URL url;

    private Class type;

    private volatile boolean isAvailable =false;

    public AkkaInvoker(BasicActor actor, URL url,Class type) {
        this.actor = actor;
        this.url = url;
        this.type = type;
        this.isAvailable = true;
    }

    @Override
    public Class getInterface() {
        return type;
    }

    @Override
    public Result invoke(Invocation invocation) throws RpcException {
        if(isAvailable){
            ResponseFuture future = actor.tell(invocation);
            try {
                return (Result) future.get();
            } catch (RemotingException e) {
                throw new RpcException("Akka invoker for type "+type+" is not available",e);
            }
        }
        throw new RpcException("Akka invoker for type "+type+" is not available");
    }

    @Override
    public URL getUrl() {
        return url;
    }

    @Override
    public boolean isAvailable() {
        return isAvailable;
    }

    @Override
    public void destroy() {
        isAvailable=false;
        actor.destroy();
    }
}