From 36ecc8caf18218b62379adee87755a546d50f40b Mon Sep 17 00:00:00 2001
From: Balaji Venkatesan <venkatesan.balaji@gmail.com>
Date: Tue, 18 Oct 2016 17:57:23 -0700
Subject: [PATCH] Added support for sunion and sinter

---
 .../netflix/dyno/jedis/DynoJedisClient.java   | 27 ++++++++++++++-----
 1 file changed, 21 insertions(+), 6 deletions(-)

diff --git a/dyno-jedis/src/main/java/com/netflix/dyno/jedis/DynoJedisClient.java b/dyno-jedis/src/main/java/com/netflix/dyno/jedis/DynoJedisClient.java
index c0db2513..2e7e14f7 100644
--- a/dyno-jedis/src/main/java/com/netflix/dyno/jedis/DynoJedisClient.java
+++ b/dyno-jedis/src/main/java/com/netflix/dyno/jedis/DynoJedisClient.java
@@ -15,6 +15,7 @@
  ******************************************************************************/
 package com.netflix.dyno.jedis;
 
+import com.google.common.collect.Sets;
 import com.netflix.discovery.DiscoveryClient;
 import com.netflix.dyno.connectionpool.*;
 import com.netflix.dyno.connectionpool.exception.DynoConnectException;
@@ -25,12 +26,13 @@
 import com.netflix.dyno.connectionpool.impl.lb.HttpEndpointBasedTokenMapSupplier;
 import com.netflix.dyno.connectionpool.impl.utils.CollectionUtils;
 import com.netflix.dyno.connectionpool.impl.utils.ZipUtils;
-import com.netflix.dyno.contrib.*;
-
+import com.netflix.dyno.contrib.ArchaiusConnectionPoolConfiguration;
+import com.netflix.dyno.contrib.DynoCPMonitor;
+import com.netflix.dyno.contrib.DynoOPMonitor;
+import com.netflix.dyno.contrib.EurekaHostsSupplier;
 import org.slf4j.Logger;
-
-import redis.clients.jedis.BinaryClient.LIST_POSITION;
 import redis.clients.jedis.*;
+import redis.clients.jedis.BinaryClient.LIST_POSITION;
 import redis.clients.jedis.params.geo.GeoRadiusParam;
 import redis.clients.jedis.params.sortedset.ZAddParams;
 import redis.clients.jedis.params.sortedset.ZIncrByParams;
@@ -2429,7 +2431,13 @@ public Long msetnx(String... keysvalues) {
 
     @Override
     public Set<String> sinter(String... keys) {
-        throw new UnsupportedOperationException("not yet implemented");
+        Set<String> allResults = d_smembers(keys[0]).getResult();
+        //loop through keys and intersect to a set, use d_smembers
+        for(int i = 1; i < keys.length; i++) {
+            allResults = Sets.intersection(allResults, d_smembers(keys[i]).getResult());
+        }
+
+        return allResults;
     }
 
     public Long sinterstore(final String dstkey, final String... keys) {
@@ -2448,7 +2456,14 @@ public Long sort(String key, String dstkey) {
 
     @Override
     public Set<String> sunion(String... keys) {
-        throw new UnsupportedOperationException("not yet implemented");
+        Set<String> allResults = new HashSet<String>();
+        //loop through keys and append to a set, use d_smembers
+        for(String key : keys) {
+            OperationResult<Set<String>> results = d_smembers(key);
+            allResults.addAll(results.getResult());
+        }
+
+        return allResults;
     }
 
     @Override