com.landoop.jdbc4.rows.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lenses-jdbc Show documentation

null

The newest version!

package com.landoop.jdbc4

import com.fasterxml.jackson.databind.JsonNode
import org.apache.avro.generic.GenericData
import java.math.BigDecimal
import java.sql.SQLException

class ArrayRow(private val array: Array) : ConvertingRow() {
  override fun getObject(index: Int): Any? = array[index - 1]
}

class JsonNodeRow(private val node: JsonNode) : ConvertingRow() {
  override fun getObject(index: Int): Any? {
    // remember jdbc is 1-indexed
    val n = node.fields().asSequence().toList()[index - 1].value
    return when {
      n.isBigDecimal -> BigDecimal(n.bigIntegerValue())
      n.isBigInteger -> n.bigIntegerValue()
      n.isBinary -> n.binaryValue()
      n.isBoolean -> n.asBoolean()
      n.isDouble -> n.doubleValue()
      n.isFloat -> n.floatValue()
      n.isInt -> n.intValue()
      n.isLong -> n.longValue()
      n.isShort -> n.shortValue()
      n.isTextual -> n.asText()
      else -> throw SQLException("Unsupported node type $n")
    }
  }
}

class RecordRow(val record: GenericData.Record) : ConvertingRow() {
  override fun getObject(index: Int): Any? {
    return record.get(index - 1)
  }
}