From eacb4eac990ebf18b51f9269e4115b1b83a7be62 Mon Sep 17 00:00:00 2001 From: hyukjinkwon Date: Mon, 14 Dec 2015 11:53:49 +0900 Subject: [PATCH] Add a case for isnotnull 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 1c348ed62fc7..1c5ece563ba9 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 IsNotNull(attr) => s"$attr IS NOT 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 8c24aa3151bc..1211483c67cd 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 NOT NULL").collect().size === 0) } test("SELECT * WHERE (quoted strings)") {