From 9b30d6572fdb988796788ad4c7b8daabdef4e961 Mon Sep 17 00:00:00 2001
From: Benjamin Lerer <b.lerer@gmail.com>
Date: Mon, 14 Dec 2015 10:17:50 +0100
Subject: [PATCH] Make Stress compiles within eclipse

patch by Benjamin Lerer; reviewed by Jake Luciani for CASSANDRA-10807
---
 CHANGES.txt                                         |  1 +
 .../cassandra/stress/generate/values/Generator.java |  4 ++--
 .../cassandra/stress/generate/values/Lists.java     | 13 +++++++------
 .../cassandra/stress/generate/values/Sets.java      | 10 +++++-----
 4 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/CHANGES.txt b/CHANGES.txt
index 96f2f4f081..091ac52738 100644
--- a/CHANGES.txt
+++ b/CHANGES.txt
@@ -1,4 +1,5 @@
 2.1.13
+ * Make Stress compiles within eclipse (CASSANDRA-10807)
  * Cassandra Daemon should print JVM arguments (CASSANDRA-10764)
  * Allow cancellation of index summary redistribution (CASSANDRA-8805)
  * Disable reloading of GossipingPropertyFileSnitch (CASSANDRA-9474)
diff --git a/tools/stress/src/org/apache/cassandra/stress/generate/values/Generator.java b/tools/stress/src/org/apache/cassandra/stress/generate/values/Generator.java
index 00f866afa8..6b39d087f2 100644
--- a/tools/stress/src/org/apache/cassandra/stress/generate/values/Generator.java
+++ b/tools/stress/src/org/apache/cassandra/stress/generate/values/Generator.java
@@ -31,13 +31,13 @@ public abstract class Generator<T>
 
     public final String name;
     public final AbstractType<T> type;
-    public final Class<T> clazz;
+    public final Class<?> clazz;
     final long salt;
     final Distribution identityDistribution;
     final Distribution sizeDistribution;
     public final Distribution clusteringDistribution;
 
-    public Generator(AbstractType<T> type, GeneratorConfig config, String name, Class<T> clazz)
+    public Generator(AbstractType<T> type, GeneratorConfig config, String name, Class<?> clazz)
     {
         this.type = type;
         this.name = name;
diff --git a/tools/stress/src/org/apache/cassandra/stress/generate/values/Lists.java b/tools/stress/src/org/apache/cassandra/stress/generate/values/Lists.java
index bfa58ea37f..d82e01f97f 100644
--- a/tools/stress/src/org/apache/cassandra/stress/generate/values/Lists.java
+++ b/tools/stress/src/org/apache/cassandra/stress/generate/values/Lists.java
@@ -26,16 +26,17 @@ import java.util.List;
 
 import org.apache.cassandra.db.marshal.ListType;
 
-public class Lists extends Generator<List>
+public class Lists<T> extends Generator<List<T>>
 {
-    final Generator valueType;
-    final Object[] buffer;
+    final Generator<T> valueType;
+    final T[] buffer;
 
-    public Lists(String name, Generator valueType, GeneratorConfig config)
+    @SuppressWarnings("unchecked")
+    public Lists(String name, Generator<T> valueType, GeneratorConfig config)
     {
         super(ListType.getInstance(valueType.type, true), config, name, List.class);
         this.valueType = valueType;
-        buffer = new Object[(int) sizeDistribution.maxValue()];
+        buffer = (T[]) new Object[(int) sizeDistribution.maxValue()];
     }
 
     public void setSeed(long seed)
@@ -45,7 +46,7 @@ public class Lists extends Generator<List>
     }
 
     @Override
-    public List generate()
+    public List<T> generate()
     {
         int size = (int) sizeDistribution.next();
         for (int i = 0 ; i < size ; i++)
diff --git a/tools/stress/src/org/apache/cassandra/stress/generate/values/Sets.java b/tools/stress/src/org/apache/cassandra/stress/generate/values/Sets.java
index 5c17b4e9a7..b74d6b22c9 100644
--- a/tools/stress/src/org/apache/cassandra/stress/generate/values/Sets.java
+++ b/tools/stress/src/org/apache/cassandra/stress/generate/values/Sets.java
@@ -26,11 +26,11 @@ import java.util.Set;
 
 import org.apache.cassandra.db.marshal.SetType;
 
-public class Sets extends Generator<Set>
+public class Sets<T> extends Generator<Set<T>>
 {
-    final Generator valueType;
+    final Generator<T> valueType;
 
-    public Sets(String name, Generator valueType, GeneratorConfig config)
+    public Sets(String name, Generator<T> valueType, GeneratorConfig config)
     {
         super(SetType.getInstance(valueType.type, true), config, name, Set.class);
         this.valueType = valueType;
@@ -43,9 +43,9 @@ public class Sets extends Generator<Set>
     }
 
     @Override
-    public Set generate()
+    public Set<T> generate()
     {
-        final Set set = new HashSet();
+        final Set<T> set = new HashSet<T>();
         int size = (int) sizeDistribution.next();
         for (int i = 0 ; i < size ; i++)
             set.add(valueType.generate());