package com.favorlock.atomixconnect.shared;

import com.favorlock.atomixconnect.shared.config.AtomixConfig;
import io.atomix.Atomix;
import io.atomix.AtomixClient;
import io.atomix.catalyst.serializer.Serializer;
import io.atomix.catalyst.transport.Address;
import io.atomix.catalyst.transport.netty.NettyTransport;
import io.atomix.group.DistributedGroupFactory;

import java.util.ArrayList;
import java.util.List;

public class AtomixConnector {
    private static Atomix client;

    public static Atomix getClient() {
        return client;
    }

    public static void init(AtomixConfig config, ClassLoader classLoader) {
        if (client != null) {
            return;
        }

        System.out.println("Connecting to Atomix cluster...");
        List<Address> atomixAddresses = new ArrayList<>();
        Serializer serializer = new Serializer();
        serializer.registerClassLoader(DistributedGroupFactory.class.getName(), classLoader);
        config.addresses.forEach(address -> atomixAddresses.add(new Address(address.host, address.port)));
        AtomixConnector.client = AtomixClient.builder().withSerializer(serializer).withTransport(new NettyTransport()).build()
                .connect(atomixAddresses).join();
    }
}