injvm如何實現本地調用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"/>