Github user srowen commented on a diff in the pull request: https://github.com/apache/spark/pull/19553#discussion_r147593655 --- Diff: core/src/main/scala/org/apache/spark/api/java/JavaUtils.scala --- @@ -43,10 +43,17 @@ private[spark] object JavaUtils { override def size: Int = underlying.size - override def get(key: AnyRef): B = try { - underlying.getOrElse(key.asInstanceOf[A], null.asInstanceOf[B]) - } catch { - case ex: ClassCastException => null.asInstanceOf[B] + // Delegate to implementation because AbstractMap implementation iterates over whole key set + override def containsKey(key: AnyRef): Boolean = { + underlying.contains(key.asInstanceOf[A]) --- End diff -- Really, this should return `false` if the key isn't an `A`. This will throw an exception now. It should be prefixed with `key.isInstanceOf[A] && ...`
--- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org