import org.apache.zookeeper.*;
import org.apache.zookeeper.ZooDefs.Ids;
import org.apache.zookeeper.data.Stat;

public class ZooKeeperIntegration {
    private static final String ZOOKEEPER_HOST = "localhost:2181";
    private static final String SERVICE_PATH = "/services";
    private static final int SESSION_TIMEOUT = 3000;

    private ZooKeeper zooKeeper;

    public ZooKeeperIntegration() throws Exception {
        this.zooKeeper = new ZooKeeper(ZOOKEEPER_HOST, SESSION_TIMEOUT, null);
    }

    public void registerService(String serviceName, String serviceAddress) throws Exception {
        String serviceNode = SERVICE_PATH + "/" + serviceName;
        String serviceData = serviceAddress.getBytes();

        // Create service node if it doesn't exist
        if (zooKeeper.exists(serviceNode, false) == null) {
            zooKeeper.create(serviceNode, null, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
        }

        // Create service instance node
        String instancePath = zooKeeper.create(serviceNode + "/instance", serviceData, Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
    }

    public void discoverServices(String serviceName) throws Exception {
        String serviceNode = SERVICE_PATH + "/" + serviceName;

        // Get all service instances
        List<String> instances = zooKeeper.getChildren(serviceNode, false);

        for (String instance : instances) {
            byte[] data = zooKeeper.getData(serviceNode + "/" + instance, false, null);
            String instanceAddress = new String(data);

            System.out.println("Instance Address: " + instanceAddress);
        }
    }
}