com.alibaba.dubbo.registry.hazelcast.HazelcastRegistry Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-starter-dubbo Show documentation
Spring Boot Dubbo Rpc
There is a newer version: 1.5.33
Show newest version
package com.alibaba.dubbo.registry.hazelcast;

import com.alibaba.dubbo.common.Constants;
import com.alibaba.dubbo.common.URL;
import com.alibaba.dubbo.common.utils.UrlUtils;
import com.alibaba.dubbo.registry.NotifyListener;
import com.alibaba.dubbo.registry.support.FailbackRegistry;
import com.hazelcast.cluster.ClusterState;
import com.hazelcast.config.Config;
import com.hazelcast.config.GroupConfig;
import com.hazelcast.core.*;
import com.hazelcast.instance.HazelcastInstanceFactory;

import java.util.*;

/**
 * Created by wuyu on 2017/4/24.
 */
public class HazelcastRegistry extends FailbackRegistry {

    private HazelcastInstance hazelcastInstance;

    private final ReplicatedMap> replicatedMap;

    private final String nodeId;

    public HazelcastRegistry(URL url) {
        super(url);

        String username = url.getParameter("username","dubbo");

        Config config = new Config(username);

        if (url.getParameter("managementCenter") != null) {
            String managerUrl = url.getParameter("managementCenter");
            config.getManagementCenterConfig().setEnabled(true);
            config.getManagementCenterConfig().setUrl(managerUrl);
            config.getManagementCenterConfig().setUpdateInterval(url.getParameter("updateInterval", 5));
        }
        config.setGroupConfig(new GroupConfig(url.getUsername() == null ? username : url.getUsername(), url.getPassword() == null ? "dubbo" : url.getPassword()));
        this.hazelcastInstance = HazelcastInstanceFactory.getOrCreateHazelcastInstance(config);
        this.hazelcastInstance.getCluster().addMembershipListener(new MembershipListener() {
            @Override
            public void memberAdded(MembershipEvent membershipEvent) {

            }

            @Override
            public void memberRemoved(MembershipEvent membershipEvent) {
                replicatedMap.remove(membershipEvent.getMember().getUuid());
            }

            @Override
            public void memberAttributeChanged(MemberAttributeEvent memberAttributeEvent) {

            }
        });
        this.replicatedMap = hazelcastInstance.getReplicatedMap(username+"-dubbo-registered");
        this.nodeId = hazelcastInstance.getCluster().getLocalMember().getUuid();
        replicatedMap.put(nodeId, new LinkedHashSet());
        replicatedMap.addEntryListener(new EntryAdapter>() {
            @Override
            public void onEntryEvent(EntryEvent> event) {
                HazelcastRegistry.this.notify(toUrl(event.getValue()));
            }

            @Override
            public void onMapEvent(MapEvent event) {
                HazelcastRegistry.this.notify(new ArrayList(getRegistered()));
            }
        });
    }

    @Override
    public boolean isAvailable() {
        return hazelcastInstance.getCluster().getClusterState().equals(ClusterState.ACTIVE);
    }


    @Override
    public Set getRegistered() {
        Set registered = new LinkedHashSet<>();
        for (Set values : replicatedMap.values()) {
            registered.addAll(toUrl(values));
        }
        return registered;
    }


    @Override
    public void destroy() {
        try {
            ILock lock = hazelcastInstance.getLock(nodeId);
            lock.lock();
            try {
                Set urls = replicatedMap.remove(nodeId);
                notify(toUrl(urls));
            } finally {
                lock.unlock();
            }
            hazelcastInstance.shutdown();
        } catch (Exception e) {
            logger.warn(e.getMessage());
        }
    }


    protected void subscribed(URL url, NotifyListener listener) {
        List urls = lookup(url);
        notify(url, listener, urls);
    }

    @Override
    protected void doRegister(URL url) {
        if (logger.isInfoEnabled()) {
            logger.info("Register: " + url);
        }
        Set urls = replicatedMap.get(nodeId);
        urls.add(url.toFullString());

        ILock lock = hazelcastInstance.getLock(nodeId);
        lock.lock();
        try {
            replicatedMap.put(nodeId, urls);
        } finally {
            lock.unlock();
        }
    }

    @Override
    protected void doUnregister(URL url) {
        Set urls = replicatedMap.get(this.nodeId);
        ILock lock = hazelcastInstance.getLock(nodeId);
        lock.lock();
        try {
            urls.remove(url.toFullString());
        } finally {
            lock.unlock();
        }
    }

    @Override
    protected void doSubscribe(URL url, NotifyListener listener) {

    }

    @Override
    protected void doUnsubscribe(URL url, NotifyListener listener) {

    }

    private List toUrl(Collection urls) {
        if (urls == null) {
            return new ArrayList<>();
        }
        List converts = new ArrayList<>();
        for (String url : urls) {
            converts.add(URL.valueOf(URL.decode(url)));
        }
        return converts;
    }

    @Override
    public void subscribe(URL url, NotifyListener listener) {
        super.subscribe(url, listener);
        subscribed(url, listener);
    }

    public List lookup(URL url) {
        List urls = new ArrayList();
        Map> notifiedUrls = getNotified().get(url);
        if (notifiedUrls != null && notifiedUrls.size() > 0) {
            for (List values : notifiedUrls.values()) {
                urls.addAll(values);
            }
        }
        if (urls == null || urls.size() == 0) {
            Set cacheUrls = getRegistered();
            if (cacheUrls != null && cacheUrls.size() > 0) {
                urls.addAll(cacheUrls);
            }
        }
        if (urls == null || urls.size() == 0) {
            for (URL u : getRegistered()) {
                if (UrlUtils.isMatch(url, u)) {
                    urls.add(u);
                }
            }
        }
        if (Constants.ANY_VALUE.equals(url.getServiceInterface())) {
            for (URL u : getSubscribed().keySet()) {
                if (UrlUtils.isMatch(url, u)) {
                    urls.add(u);
                }
            }
        }
        return urls;
    }

    public String getNodeId() {
        return nodeId;
    }
}