2015-09-16 15 views
6

Nella versione precedente, avevamo un metodo 'saveAsOrcFile()' su RDD. Questo è ora andato! Come posso salvare i dati in DataFrame nel formato di file ORC?Spark: Salva Dataframe in formato ORC

def main(args: Array[String]) { 
println("Creating Orc File!") 
val sparkConf = new SparkConf().setAppName("orcfile") 
val sc = new SparkContext(sparkConf) 
val hiveContext = new org.apache.spark.sql.hive.HiveContext(sc) 

val people = sc.textFile("/apps/testdata/people.txt") 
val schemaString = "name age" 
val schema = StructType(schemaString.split(" ").map(fieldName => {if(fieldName == "name") StructField(fieldName, StringType, true) else StructField(fieldName, IntegerType, true)})) 
val rowRDD = people.map(_.split(",")).map(p => Row(p(0), new Integer(p(1).trim))) 

//# Infer table schema from RDD** 
val peopleSchemaRDD = hiveContext.createDataFrame(rowRDD, schema) 

//# Create a table from schema** 
peopleSchemaRDD.registerTempTable("people") 
val results = hiveContext.sql("SELECT * FROM people") 
results.map(t => "Name: " + t.toString).collect().foreach(println) 

// Now I want to save this Dataframe(peopleSchemaRDD) in ORC Format. How do I do that? 

}

risposta

8

Dal Spark 1.4 si può semplicemente utilizzare DataFrameWriter e impostare format-orc:

peopleSchemaRDD.write.format("orc").save("people") 

o

peopleSchemaRDD.write.orc("people") 
Problemi correlati