com.quorum.tessera.discovery.DiscoveryFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tessera-partyinfo Show documentation

Tessera is a stateless Java system that is used to enable the encryption, decryption, and distribution of private transactions for Quorum.

There is a newer version: 24.4.2

Show newest version

package com.quorum.tessera.discovery;

import com.quorum.tessera.context.RuntimeContext;
import com.quorum.tessera.partyinfo.node.NodeInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.URI;
import java.util.Set;
import java.util.concurrent.atomic.AtomicReference;
import java.util.stream.Collectors;

public class DiscoveryFactory implements Discovery {

    private static final Logger LOGGER = LoggerFactory.getLogger(DiscoveryFactory.class);

    private static final AtomicReference HOLDER = new AtomicReference<>();

    /**
     * @see java.util.ServiceLoader.Provider
     */
    public static Discovery provider() {

//        if(HOLDER.get() != null) {
//            return HOLDER.get();
//        }

        final NetworkStore networkStore = NetworkStore.getInstance();
        final RuntimeContext runtimeContext = RuntimeContext.getInstance();
        final Discovery discovery;
        if(runtimeContext.isDisablePeerDiscovery()) {
            final Set knownNodes = runtimeContext.getPeers().stream()
                                        .map(NodeUri::create)
                                        .collect(Collectors.toUnmodifiableSet());
            discovery = new DisabledAutoDiscovery(networkStore,knownNodes);
        } else {
            discovery = new AutoDiscovery(networkStore);
        }

       // HOLDER.set(discovery);

        return discovery;
    }

    private final Discovery discovery;

    public DiscoveryFactory() {
        this(provider());
    }

    protected DiscoveryFactory(Discovery discovery) {
        this.discovery = discovery;
    }

    @Override
    public void onCreate() {
        discovery.onCreate();
    }

    @Override
    public void onUpdate(NodeInfo nodeInfo) {
        discovery.onUpdate(nodeInfo);
    }

    @Override
    public void onDisconnect(URI nodeUri) {
        discovery.onDisconnect(nodeUri);
    }


}