medium_ref_sstable_cql_scripts.scala

private def getCreateTableStatement(keyspace: String, tableName: String, df: DataFrame, keys: Array[String]) = {
  val primaryKey = keys.mkString(",")
  val fields = df.schema.map(field => field.name + " " + field.dataType.simpleString.trim())
  val fieldsDesc = fields.mkString(", ")
  "CREATE TABLE if not exists " + keyspace + "." + tableName + " (" + fieldsDesc + ", PRIMARY KEY (" + primaryKey + "))"
}

private def getInsertStatement(keyspace: String, tableName: String, df: DataFrame) = {
  val fieldsDesc = df.schema.fields.map(f => f.name).mkString(",")
  val valuesHolder = df.schema.fields.map(_ => "?").mkString(",")
  "insert into " + keyspace + "." + tableName + " (" + fieldsDesc + ") values (" + valuesHolder + ")"
}