scala.collection.JavaConverters.asScalaBufferConverter Scala Examples

The following examples show how to use scala.collection.JavaConverters.asScalaBufferConverter. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example.
Example 1
Source File: ItemConverter.scala    From spark-dynamodb   with Apache License 2.0 5 votes vote down vote up 
package com.github.traviscrawford.spark.dynamodb

import com.amazonaws.services.dynamodbv2.document.Item
import org.apache.spark.sql.Row
import org.apache.spark.sql.types._

import scala.collection.JavaConverters.asScalaBufferConverter


private[dynamodb] object ItemConverter {

  def toRow(item: Item, schema: StructType): Row = {
    val values: Seq[Any] = schema.map(field => {
      if (item.hasAttribute(field.name)) {
        field.dataType.typeName match {
          case "string" => item.getString(field.name)
          case "integer" => item.getInt(field.name)
          case "long" => item.getLong(field.name)
          case "float" => item.getFloat(field.name)
          case "double" => item.getDouble(field.name)
          case "array" => getArrayValue(field, item)
          case _ => throw new IllegalArgumentException(
            s"Unexpected data type ${field.dataType.typeName} field: $field item: $item")
        }
      } else {
        // scalastyle:off null
        null
        // scalastyle:on null
      }
    })

    Row.fromSeq(values)
  }

  private def getArrayValue(field: StructField, item: Item): Any = {
    def getList(fieldName: String): List[java.math.BigDecimal] = {
      item.getList[java.math.BigDecimal](field.name).asScala.toList
    }

    field.dataType.asInstanceOf[ArrayType].elementType match {
      case StringType => item.getList[String](field.name).asScala.toList
      case IntegerType => getList(field.name).map(_.intValue())
      case LongType => getList(field.name).map(_.longValue())
      case FloatType => getList(field.name).map(_.floatValue())
      case DoubleType => getList(field.name).map(_.doubleValue())
      case _ => throw new IllegalArgumentException(
        s"Unexpected array element type ${field.dataType.typeName} field: $field item: $item")
    }
  }
}