org.sisioh.aws4s.dynamodb.document.RichTableKeysAndAttributes.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.document

import com.amazonaws.services.dynamodbv2.document.{ PrimaryKey, TableKeysAndAttributes }
import org.sisioh.aws4s.PimpedType

import scala.collection.JavaConverters._

object TableKeysAndAttributeFactory {

  def apply(tableName: String): TableKeysAndAttributes = new TableKeysAndAttributes(tableName)

}

class RichTableKeysAndAttributes(val underlying: TableKeysAndAttributes) extends AnyVal with PimpedType[TableKeysAndAttributes] {

  def primaryKeys: Seq[PrimaryKey] = underlying.getPrimaryKeys.asScala.toVector

  def withAttributeNames(attributeNames: Seq[String]): TableKeysAndAttributes = underlying.withAttributeNames(attributeNames.asJava)

  def attributeNames: Set[String] = underlying.getAttributeNames.asScala.toSet

  def tableName: String = underlying.getTableName

  def projectionExpression: String = underlying.getProjectionExpression

  def nameMap: Map[String, String] = underlying.getNameMap.asScala.toMap

  def withNameMap(nameMap: Map[String, String]): TableKeysAndAttributes = underlying.withNameMap(nameMap.asJava)

}