• Home
• com.scylladb
• spark-scylladb-connector_2.12
• 4.0.0
• source code
• CassandraLimit.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.datastax.spark.connector.rdd.CassandraLimit.scala Maven / Gradle / Ivy

package com.datastax.spark.connector.rdd

sealed trait CassandraLimit

case class CassandraPartitionLimit(rowsNumber: Long) extends CassandraLimit {
  require(rowsNumber > 0, s"$rowsNumber <= 0. Per Partition Limits must be greater than 0")
}
case class SparkPartitionLimit(rowsNumber: Long) extends CassandraLimit {
  require(rowsNumber > 0, s"$rowsNumber <= 0. Limits must be greater than 0")
}

object CassandraLimit {

  def limitToClause
  (limit: Option[CassandraLimit]): String = limit match {
    case Some(SparkPartitionLimit(rowsNumber)) => s"LIMIT $rowsNumber"
    case Some(CassandraPartitionLimit(rowsNumber)) => s"PER PARTITION LIMIT $rowsNumber"
    case None => ""
  }

  def limitForIterator(limit: Option[CassandraLimit]): Option[Long] = limit.collect {
    case SparkPartitionLimit(rowsNumber) => rowsNumber
  }
}