package com.blogspot.yawar.kitetest

import org.kitesdk.data.{
  DatasetDescriptor, Datasets, PartitionStrategy, View }

case class Person(name: String, age: Int)

object Main extends App {
  val personClass = classOf[Person]

  /*
  val partitionAge =
    new PartitionStrategy.Builder().hash("age", 10).build
  */

  val datasetDesc =
    new DatasetDescriptor.Builder()
      .schema(personClass)
      //.partitionStrategy(partitionAge)
      .build

  val people =
    Datasets
      .create(
        "dataset:file:/Users/yawar/code/kitedata",
        datasetDesc,
        personClass)
      .asInstanceOf[View[Person]]

  val peopleWriter = people.newWriter
  peopleWriter write Person("Yawar", 31)
  peopleWriter write Person("Bob", 30)
  peopleWriter.close()
}