redis.clients.util.JedisByteHashMap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jedis Show documentation
Jedis Redis java client with some modifications to support new commands in YB Redis.
There is a newer version: 2.9.0-yb-16
package redis.clients.util;

import java.io.Serializable;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

public class JedisByteHashMap implements Map, Cloneable, Serializable {
  private static final long serialVersionUID = -6971431362627219416L;
  private transient Map internalMap = new HashMap();

  public void clear() {
    internalMap.clear();
  }

  public boolean containsKey(Object key) {
    if (key instanceof byte[]) return internalMap.containsKey(new ByteArrayWrapper((byte[]) key));
    return internalMap.containsKey(key);
  }

  public boolean containsValue(Object value) {
    return internalMap.containsValue(value);
  }

  public Set> entrySet() {
    Iterator> iterator = internalMap.entrySet()
        .iterator();
    HashSet> hashSet = new HashSet>();
    while (iterator.hasNext()) {
      Entry entry = iterator.next();
      hashSet.add(new JedisByteEntry(entry.getKey().data, entry.getValue()));
    }
    return hashSet;
  }

  public byte[] get(Object key) {
    if (key instanceof byte[]) return internalMap.get(new ByteArrayWrapper((byte[]) key));
    return internalMap.get(key);
  }

  public boolean isEmpty() {
    return internalMap.isEmpty();
  }

  public Set keySet() {
    Set keySet = new HashSet();
    Iterator iterator = internalMap.keySet().iterator();
    while (iterator.hasNext()) {
      keySet.add(iterator.next().data);
    }
    return keySet;
  }

  public byte[] put(byte[] key, byte[] value) {
    return internalMap.put(new ByteArrayWrapper(key), value);
  }

  @SuppressWarnings("unchecked")
  public void putAll(Map m) {
    Iterator iterator = m.entrySet().iterator();
    while (iterator.hasNext()) {
      Entry next = (Entry) iterator
          .next();
      internalMap.put(new ByteArrayWrapper(next.getKey()), next.getValue());
    }
  }

  public byte[] remove(Object key) {
    if (key instanceof byte[]) return internalMap.remove(new ByteArrayWrapper((byte[]) key));
    return internalMap.remove(key);
  }

  public int size() {
    return internalMap.size();
  }

  public Collection values() {
    return internalMap.values();
  }

  private static final class ByteArrayWrapper {
    private final byte[] data;

    public ByteArrayWrapper(byte[] data) {
      if (data == null) {
        throw new NullPointerException();
      }
      this.data = data;
    }

    @Override
    public boolean equals(Object other) {
      if (!(other instanceof ByteArrayWrapper)) {
        return false;
      }
      return Arrays.equals(data, ((ByteArrayWrapper) other).data);
    }

    @Override
    public int hashCode() {
      return Arrays.hashCode(data);
    }
  }

  private static final class JedisByteEntry implements Entry {
    private byte[] value;
    private byte[] key;

    public JedisByteEntry(byte[] key, byte[] value) {
      this.key = key;
      this.value = value;
    }

    public byte[] getKey() {
      return this.key;
    }

    public byte[] getValue() {
      return this.value;
    }

    public byte[] setValue(byte[] value) {
      this.value = value;
      return value;
    }

  }
}