org.jetlinks.supports.cluster.EventBusLocalCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jetlinks-supports Show documentation
JetLinks 默认功能实现
The newest version!
package org.jetlinks.supports.cluster;

import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import org.jetlinks.core.cluster.ClusterCache;
import org.jetlinks.core.cluster.ClusterManager;
import org.jetlinks.core.codec.Codecs;
import org.jetlinks.core.codec.Encoder;
import org.jetlinks.core.event.EventBus;
import reactor.core.publisher.Mono;

import java.util.Collection;

public class EventBusLocalCache extends AbstractLocalCache {

    private final EventBus eventBus;

    private final String topicPrefix;

    private static final byte notifyData = (byte) 1;
    private static final Encoder encoder = Codecs.lookup(byte.class);


    public EventBusLocalCache(String name,
                              EventBus eventBus,
                              ClusterManager clusterManager) {
        this(name,
             eventBus,
             clusterManager,
             CacheBuilder.newBuilder().build());
    }

    public EventBusLocalCache(String name,
                              EventBus eventBus,
                              ClusterManager clusterManager,
                              Cache localCache) {
        this(name, eventBus, clusterManager.getCache(name), localCache);
    }

    public EventBusLocalCache(String name,
                              EventBus eventBus,
                              ClusterCache clusterCache,
                              Cache localCache) {
        this(name, eventBus, clusterCache, localCache, true);
    }

    public EventBusLocalCache(String name,
                              EventBus eventBus,
                              ClusterCache clusterCache,
                              Cache localCache,
                              boolean cacheEmpty) {
        super(clusterCache, localCache, cacheEmpty);
        this.eventBus = eventBus;
        this.topicPrefix = "/_sys/cluster_cache/" + name;
    }

    @Override
    protected Mono onUpdate(K key, V value) {
        return eventBus
                .publish(topicPrefix + "/update/" + key, encoder, notifyData)
                .then();
    }

    @Override
    protected Mono onRemove(K key) {
        return eventBus
                .publish(topicPrefix + "/remove/" + key, encoder, notifyData)
                .then();
    }

    @Override
    protected Mono onRemove(Collection key) {
        return eventBus
                .publish(topicPrefix + "/remove/__all", encoder, notifyData)
                .then();
    }

    @Override
    protected Mono onClear() {
        return eventBus
                .publish(topicPrefix + "/remove/__all", encoder, notifyData)
                .then();
    }
}