Fixes #15

Author: asm0dey

core/pom.xml

@@ -4,7 +4,7 @@
   <parent>
     <artifactId>kotlin-spark-api-parent</artifactId>
     <groupId>org.jetbrains.kotlin.spark</groupId>
-    <version>0.2.0</version>
+    <version>0.2.1</version>
   </parent>
   <artifactId>core</artifactId>
   <name>Kotlin Spark API: Scala core</name>

core/src/main/scala/org/jetbrains/spark/extensions/KSparkExtensions.scala

@@ -19,6 +19,7 @@
  */
 package org.jetbrains.spark.extensions
 
+import org.apache.spark.SparkContext
 import org.apache.spark.sql._
 
 object KSparkExtensions {
@@ -37,4 +38,6 @@ object KSparkExtensions {
     import org.apache.spark.sql.execution.debug._
     df.debug()
   }
+
+  def sparkContext(s:SparkSession): SparkContext = s.sparkContext
 }

examples/pom.xml

@@ -7,7 +7,7 @@
     <parent>
         <artifactId>kotlin-spark-api-parent</artifactId>
         <groupId>org.jetbrains.kotlin.spark</groupId>
-        <version>0.2.0</version>
+        <version>0.2.1</version>
     </parent>
 
     <groupId>org.jetbrains.kotlin.spark.examples</groupId>

kotlin-spark-api/pom.xml

@@ -7,7 +7,7 @@
     <parent>
         <artifactId>kotlin-spark-api-parent</artifactId>
         <groupId>org.jetbrains.kotlin.spark</groupId>
-        <version>0.2.0</version>
+        <version>0.2.1</version>
     </parent>
 
     <artifactId>kotlin-spark-api</artifactId>

kotlin-spark-api/src/main/kotlin/org/jetbrains/spark/api/ApiV1.kt

@@ -21,6 +21,7 @@
 
 package org.jetbrains.spark.api
 
+import org.apache.spark.SparkContext
 import org.apache.spark.api.java.function.*
 import org.apache.spark.sql.*
 import org.apache.spark.sql.Encoders.*
@@ -141,6 +142,9 @@ inline fun <reified T> Dataset<T>.forEach(noinline func: (T) -> Unit) = foreach(
  */
 fun <T> Dataset<T>.debugCodegen() = also { KSparkExtensions.debugCodegen(it) }
 
+val SparkSession.sparkContext
+    get() = KSparkExtensions.sparkContext(this)
+
 /**
  * It's hard to call `Dataset.debug` from kotlin, so here is utility for that
  */

pom.xml

@@ -4,7 +4,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>org.jetbrains.kotlin.spark</groupId>
     <artifactId>kotlin-spark-api-parent</artifactId>
-    <version>0.2.0</version>
+    <version>0.2.1</version>
     <name>kotlin-spark-api-parent</name>
     <url>http://maven.apache.org</url>
     <packaging>pom</packaging>