applyFunction should have F as type param

simerplaha · Oct 7, 2019 · 5c81825 · 5c81825
1 parent 2ed4a78
commit 5c81825
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/swaydb/src/main/scala/swaydb/Map.scala b/swaydb/src/main/scala/swaydb/Map.scala
@@ -153,10 +153,10 @@ case class Map[K, V, F, T[_]](private[swaydb] val core: Core[T],
     function
   }
 
-  def applyFunction[B <: F with swaydb.Function[K, V]](key: K, function: B): T[IO.Done] =
+  def applyFunction(key: K, function: F with swaydb.Function[K, V]): T[IO.Done] =
     tag.point(core.function(key, function.id))
 
-  def applyFunction[B <: F with swaydb.Function[K, V]](from: K, to: K, function: B): T[IO.Done] =
+  def applyFunction(from: K, to: K, function: F with swaydb.Function[K, V]): T[IO.Done] =
     tag.point(core.function(from, to, function.id))
 
   def commit(prepare: Prepare[K, V]*): T[IO.Done] =

diff --git a/swaydb/src/main/scala/swaydb/Set.scala b/swaydb/src/main/scala/swaydb/Set.scala
@@ -130,11 +130,11 @@ case class Set[A, F, T[_]](private val core: Core[T],
     function
   }
 
-  def applyFunction(from: A, to: A, functionId: A): T[IO.Done] =
-    tag.point(core.function(from, to, functionId))
+  def applyFunction(from: A, to: A, function: F with swaydb.Function.GetKey[A, Nothing]): T[IO.Done] =
+    tag.point(core.function(from, to, function.id))
 
-  def applyFunction(elem: A, function: A): T[IO.Done] =
-    tag.point(core.function(elem, function))
+  def applyFunction(elem: A, function: F with swaydb.Function.GetKey[A, Nothing]): T[IO.Done] =
+    tag.point(core.function(elem, function.id))
 
   def commit(prepare: Prepare[A, Nothing]*): T[IO.Done] =
     tag.point(core.put(prepare))