shutdown + set fixes

wavesplatform · Nov 7, 2017 · d495b95 · d495b95
1 parent f04c986
commit d495b95
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/discovery/src/main/scala/com.wavesplatform.discovery/collections/ExpirationSet.scala b/discovery/src/main/scala/com.wavesplatform.discovery/collections/ExpirationSet.scala
@@ -5,13 +5,15 @@ import scala.collection.JavaConverters._
 
 import com.google.common.cache.CacheBuilder
 
-class ExpirationSet[T](expirationTimeMilis: Long) extends scala.collection.mutable.Set[T]{
+class ExpirationSet[T <: Object](expirationTimeMilis: Long) extends scala.collection.mutable.Set[T]{
+  private val emptyItem = new Object()
+
   private var inner =  CacheBuilder.newBuilder()
     .expireAfterWrite(expirationTimeMilis, TimeUnit.MILLISECONDS)
-    .build[T, Boolean]()
+    .build[T, Object]()
 
   override def +=(elem: T): ExpirationSet.this.type = {
-    inner.put(elem, true)
+    inner.put(elem, emptyItem)
     this
   }