com.netflix.dyno.demo.redis.CustomTokenSupplierExample Maven / Gradle / Ivy

Go to download
package com.netflix.dyno.demo.redis;

import com.netflix.dyno.connectionpool.Host;
import com.netflix.dyno.connectionpool.HostSupplier;
import com.netflix.dyno.connectionpool.OperationResult;
import com.netflix.dyno.connectionpool.TokenMapSupplier;
import com.netflix.dyno.connectionpool.impl.ConnectionPoolConfigurationImpl;
import com.netflix.dyno.connectionpool.impl.lb.HostToken;
import com.netflix.dyno.jedis.DynoJedisClient;
import org.apache.log4j.BasicConfigurator;

import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Set;

/**
 * Simple example that illustrates using a custom token map supplier. This example works with a local redis
 * installation on the default port. To setup and run:
 * 
 *     brew install redis
 *     redis-server /usr/local/etc/redis.conf
 *     java -classpath <...classpath...> com.netflix.dyno.demo.redis.CustomTokenSupplierExample
 * 
 */
public class CustomTokenSupplierExample {

    private DynoJedisClient client;

    public CustomTokenSupplierExample() {
    }

    public void init() throws Exception {

        final int port = 6379;

        final Host localHost = new Host("localhost", port, Host.Status.Up);

        final HostSupplier localHostSupplier = new HostSupplier() {

            @Override
            public Collection getHosts() {
                return Collections.singletonList(localHost);
            }
        };

        final TokenMapSupplier supplier = new TokenMapSupplier() {

            @Override
            public List getTokens(Set activeHosts) {
                return Collections.singletonList(localHostToken);
            }

            @Override
            public HostToken getTokenForHost(Host host, Set activeHosts) {
                return localHostToken;
            }

            final HostToken localHostToken = new HostToken(100000L, localHost);

        };

        client = new DynoJedisClient.Builder()
                .withApplicationName("tokenSupplierExample")
                .withDynomiteClusterName("tokenSupplierExample")
                .withHostSupplier(localHostSupplier)
                .withCPConfig(new ConnectionPoolConfigurationImpl("tokenSupplierExample")
                        .withTokenSupplier(supplier))
                .withPort(port)
                .build();

    }

    public void runSimpleTest() throws Exception {

        // write
        for (int i=0; i<10; i++) {
            client.set("" + i, "" + i);
        }
        // read
        for (int i=0; i<10; i++) {
            OperationResult result = client.d_get(""+i);
            System.out.println("Key: " + i + ", Value: " + result.getResult() + " " + result.getNode());
        }
    }


    public void stop() {
        if (client != null) {
            client.stopClient();
        }
    }

    public static void main(String args[]) {

        BasicConfigurator.configure();
        CustomTokenSupplierExample example = new CustomTokenSupplierExample();

        try {
            example.init();
            example.runSimpleTest();
            Thread.sleep(1000);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            example.stop();
        }
    }
}