org.sisioh.aws4s.dynamodb.model.RichBatchGetItemRequest.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aws4s-dynamodb_2.10 Show documentation

aws4s-dynamodb

There is a newer version: 1.0.16

package org.sisioh.aws4s.dynamodb.model

import com.amazonaws.services.dynamodbv2.model.{ BatchGetItemRequest, KeysAndAttributes, ReturnConsumedCapacity }
import org.sisioh.aws4s.PimpedType

import scala.collection.JavaConverters._

object BatchGetItemRequestFactory {

  def create(): BatchGetItemRequest = new BatchGetItemRequest()

  def create(requestItems: Map[String, KeysAndAttributes]): BatchGetItemRequest =
    new BatchGetItemRequest(requestItems.asJava)

  def create(requestItems: Map[String, KeysAndAttributes], returnConsumedCapacity: String): BatchGetItemRequest =
    new BatchGetItemRequest(requestItems.asJava, returnConsumedCapacity)

  def create(requestItems: Map[String, KeysAndAttributes], returnConsumedCapacity: ReturnConsumedCapacity): BatchGetItemRequest =
    new BatchGetItemRequest(requestItems.asJava, returnConsumedCapacity)

}

class RichBatchGetItemRequest(val underlying: BatchGetItemRequest) extends AnyVal with PimpedType[BatchGetItemRequest] {

  def requestItemsOpt_=(values: Option[Map[String, KeysAndAttributes]]): Unit =
    underlying.setRequestItems(values.map(_.asJava).orNull)

  def requestItemsOpt: Option[Map[String, KeysAndAttributes]] =
    Option(underlying.getRequestItems).map(_.asScala.toMap)

  def withRequestItemsOpt(values: Option[Map[String, KeysAndAttributes]]): BatchGetItemRequest =
    underlying.withRequestItems(values.map(_.asJava).orNull)

  // ---

  def returnConsumedCapacityOpt: Option[String] = Option(underlying.getReturnConsumedCapacity)

  def returnConsumedCapacityOpt_=(value: Option[String]): Unit = underlying.setReturnConsumedCapacity(value.orNull)

  def setReturnConsumedCapacityOpt(value: Option[ReturnConsumedCapacity]): Unit = underlying.setReturnConsumedCapacity(value.orNull)

  def withReturnConsumedCapacityOpt(value: Option[String]): BatchGetItemRequest = underlying.withReturnConsumedCapacity(value.orNull)

  // ---

}