From ff9d8b21e5464169fc4cbda87896c5e7e66fa303 Mon Sep 17 00:00:00 2001 From: hyukjinkwon Date: Mon, 14 Dec 2015 11:50:04 +0900 Subject: [PATCH] Add a case for isnull operator --- .../apache/spark/sql/execution/datasources/jdbc/JDBCRDD.scala | 1 + .../src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala | 1 + 2 files changed, 2 insertions(+) diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/jdbc/JDBCRDD.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/jdbc/JDBCRDD.scala index 1c348ed62fc78..62c4d4f0b1731 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/jdbc/JDBCRDD.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/jdbc/JDBCRDD.scala @@ -285,6 +285,7 @@ private[sql] class JDBCRDD( case GreaterThan(attr, value) => s"$attr > ${compileValue(value)}" case LessThanOrEqual(attr, value) => s"$attr <= ${compileValue(value)}" case GreaterThanOrEqual(attr, value) => s"$attr >= ${compileValue(value)}" + case IsNull(attr) => s"$attr IS NULL" case _ => null } diff --git a/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala index 8c24aa3151bc1..9c1cc49b9befe 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala @@ -182,6 +182,7 @@ class JDBCSuite extends SparkFunSuite with BeforeAndAfter with SharedSQLContext assert(sql("SELECT * FROM foobar WHERE NAME = 'fred'").collect().size === 1) assert(sql("SELECT * FROM foobar WHERE NAME > 'fred'").collect().size === 2) assert(sql("SELECT * FROM foobar WHERE NAME != 'fred'").collect().size === 2) + assert(sql("SELECT * FROM nulltypes WHERE A IS NULL").collect().size === 1) } test("SELECT * WHERE (quoted strings)") {