subprojects/s4-core/src/main/java/org/apache/s4/core/ZkClientProvider.java

package org.apache.s4.core;

import org.I0Itec.zkclient.serialize.ZkSerializer;
import org.apache.s4.comm.topology.ZNRecordSerializer;
import org.apache.s4.comm.topology.ZkClient;

import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.name.Named;

/**
 * 
 * Provides a connection to ZooKeeper through the {@link ZkClient} class.
 * <p>
 * This connection can easily be shared by specifying singleton scope at binding time (i.e. when binding the ZkClient
 * class, see {@link BaseModule}).
 * 
 * 
 */
public class ZkClientProvider implements Provider<ZkClient> {

    private final ZkClient zkClient;

    @Inject
    public ZkClientProvider(@Named("s4.cluster.zk_address") String zookeeperAddress,
            @Named("s4.cluster.zk_session_timeout") int sessionTimeout,
            @Named("s4.cluster.zk_connection_timeout") int connectionTimeout) {
        zkClient = new ZkClient(zookeeperAddress, sessionTimeout, connectionTimeout);
        ZkSerializer serializer = new ZNRecordSerializer();
        zkClient.setZkSerializer(serializer);
    }

    @Override
    public ZkClient get() {
        return zkClient;
    }
}