com.netflix.astyanax.connectionpool.impl.OldHostSupplierAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of astyanax-core Show documentation

astyanax-core

There is a newer version: 3.10.2

package com.netflix.astyanax.connectionpool.impl;

import java.math.BigInteger;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import com.google.common.base.Supplier;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import com.netflix.astyanax.connectionpool.Host;

public class OldHostSupplierAdapter implements Supplier> {
    private final Supplier>> source;
    
    public OldHostSupplierAdapter(Supplier>> source) {
        this.source = source;
    }
    
    @Override
    public List get() {
        Map> tokenHostMap = source.get();
        Set hosts = Sets.newHashSet();
        for (Entry> entry : tokenHostMap.entrySet()) { 
            for (Host host : entry.getValue()) {
                if (!hosts.contains(host)) {
                    hosts.add(host);
                }
                
                String token = entry.getKey().toString();
                host.getTokenRanges().add(new TokenRangeImpl(token, token, null));
            }
        }
        return Lists.newArrayList(hosts);
    }

}