Dubbo如何支援本地調用？injvm方式解析

Dubbo是一個遠端調用的架構，對于一個服務提供者，暴露了一個接口供外部消費者調用，那麼對于提供者自己是否可以調用這個接口，需要什麼特殊處理嗎？

這篇文章就分享下Dubbo關于本地調用的實作機制，以及如何開啟和關閉本地調用。

injvm支援本地調用

使用 Dubbo 本地調用不需做特殊配置，按正常 Dubbo 服務暴露服務即可。

任一服務在暴露遠端服務的同時，也會同時以 injvm 的協定暴露本地服務。injvm 是一個僞協定，不會像其他協定那樣對外開啟端口，隻用于本地調用的目的。

InjvmProtocol源碼學習

injvm相關的代碼實作在Dubbo-rpc-injvm子產品中，主要是InjvmExporter、InjvmInvoker和InjvmProtocol。

InjvmProtocol 繼承 AbstractProtocol，

public class InjvmProtocol extends AbstractProtocol implements Protocol {

    public static final String NAME = Constants.LOCAL_PROTOCOL;

    public static final int DEFAULT_PORT = 0;
    private static InjvmProtocol INSTANCE;

    public InjvmProtocol() {
        INSTANCE = this;
    }

    public static InjvmProtocol getInjvmProtocol() {
        if (INSTANCE == null) {
            ExtensionLoader.getExtensionLoader(Protocol.class).getExtension(InjvmProtocol.NAME); // load
        }
        return INSTANCE;
    }

    @Override
    public <T> Exporter<T> export(Invoker<T> invoker) throws RpcException {
        return new InjvmExporter<T>(invoker, invoker.getUrl().getServiceKey(), exporterMap);
    }

    @Override
    public <T> Invoker<T> refer(Class<T> serviceType, URL url) throws RpcException {
        return new InjvmInvoker<T>(serviceType, url, url.getServiceKey(), exporterMap);
    }
}           

除了export和refer方法，InjvmProtocol提供了isInjvmRefer()方法，

isInjvmRefer會讀取配置檔案，判斷是否開啟本地調用。

public boolean isInjvmRefer(URL url) {
        String scope = url.getParameter(Constants.SCOPE_KEY);
        // Since injvm protocol is configured explicitly, we don't need to set any extra flag, use normal refer process.
        if (Constants.SCOPE_LOCAL.equals(scope) || (url.getParameter(Constants.LOCAL_PROTOCOL, false))) {
            // if it's declared as local reference
            // 'scope=local' is equivalent to 'injvm=true', injvm will be deprecated in the future release
            return true;
        } else if (Constants.SCOPE_REMOTE.equals(scope)) {
            // it's declared as remote reference
            return false;
        } else if (url.getParameter(Constants.GENERIC_KEY, false)) {
            // generic invocation is not local reference
            return false;
        } else if (getExporter(exporterMap, url) != null) {
            // by default, go through local reference if there's the service exposed locally
            return true;
        } else {
            return false;
        }
    }
           

本地調用同樣經過Filter鍊

與真正的本地方法調用不同的是，Dubbo 本地調用會經過 Filter 鍊，其中包括了 Consumer 端的 Filter 鍊以及 Provider 端的 Filter 鍊。通過這樣的機制，本地消費者和其他消費者都是統一對待，統一監控，服務統一進行治理。

如何開啟本地調用

預設情況下，本地調用是自動開啟的，不需要做額外的配置。隻有隻有當需要關閉的時候，才需要通過 scope 的配置來顯式的關閉。

但是，特别需要指出的是，在下面的幾種情況下，本地調用是無法使用的：

第一，泛化調用的時候無法使用本地調用。

第二，消費者明确指定 URL 發起直連調用。當然，如果消費者指定的是 injvm 的 URL，最終的調用也是走本地調用的，比如：

<Dubbo:reference id="demoService" interface="org.apache.Dubbo.samples.local.api.DemoService" url="injvm://127.0.0.1/org.apache.Dubbo.samples.local.api.DemoService"/>
           

如何關閉本地調用

本地調用是可以顯示關閉的，通過這種方式，服務提供者可以做到對遠端服務消費者和本地消費者一視同仁。具體做法是通過 scope="remote" 來關閉 injvm 協定的暴露，這樣，即使是本地調用者，也需要從注冊中心上擷取服務位址清單，然後才能發起調用，而這個時候的調用過程，與遠端的服務消費者的過程是一緻的。

<bean id="target" class="org.apache.Dubbo.samples.local.impl.DemoServiceImpl"/>
<!-- 服務提供者指定 scope="remote" -->
<Dubbo:service interface="org.apache.Dubbo.samples.local.api.DemoService" ref="target" scope="remote"/>
<Dubbo:reference id="demoService" interface="org.apache.Dubbo.samples.local.api.DemoService"/>