Scala 2中如何向PostgreSQL发送Option None类型数据

我需要从Scala向PostgreSQL数据库的表中写入数据，当前遇到空值处理问题：数据库各列有严格的数据类型，而从Kafka读取的数据可能存在空值，最终要将这些数据写入数据库表。

我的尝试代码如下：

import java.sql.{Connection, DriverManager, PreparedStatement}

case class Person(name: Option[String], value: Option[Double])

val value1 = Person(Some("brak"), Some(26.5))
println(value1)

val value2 = Person(Some("mrk"),None)
println(value2)

对应的PostgreSQL表定义：

create table person(
name varchar(256) nullable,
value numeric(3,1) nullable)

使用JDBC写入的代码如下：

// Define the connection parameters for the Postgresql database
val url = "jdbc:postgresql://url/db"
val username = "username"
val password = "pass"

val insertStatement = "INSERT INTO person(name, value) VALUES (?,?)"

Class.forName("org.postgresql.Driver")
val connection: Connection = DriverManager.getConnection(url, username, password)
val preparedStatement: PreparedStatement = connection.prepareStatement(insertStatement)
value2.name.foreach(preparedStatement.setString(1, _))
value2.value.foreach {
  case Some(value) => preparedStatement.setDouble(2, value)
  case None => preparedStatement.setNull(2, java.sql.Types.DOUBLE)
}
preparedStatement.executeUpdate()

// Close the resources
preparedStatement.close()
connection.close()

处理value2时出现如下错误：

<console>:49: error: constructor cannot be instantiated to expected type;
 found   : Some[A]
 required: Double
         case Some(value) => preparedStatement.setDouble(2, value)
              ^
<console>:50: error: pattern type is incompatible with expected type;
 found   : None.type
 required: Double
         case None => preparedStatement.setNull(2, java.sql.Types.DOUBLE)
              ^

错误原因 #

value2.value是Option[Double]类型，调用foreach时，传入的函数参数是被取出的Double值（而非Option本身），但你却用Some/None去匹配这个Double值，自然会触发类型不匹配错误。

修正方案 #

直接对Option对象做模式匹配，而非在foreach里处理：

import java.sql.{Connection, DriverManager, PreparedStatement}

case class Person(name: Option[String], value: Option[Double])

val value2 = Person(Some("mrk"), None)

// 数据库连接参数
val url = "jdbc:postgresql://url/db"
val username = "username"
val password = "pass"

val insertStatement = "INSERT INTO person(name, value) VALUES (?,?)"

Class.forName("org.postgresql.Driver")
val connection: Connection = DriverManager.getConnection(url, username, password)
val preparedStatement: PreparedStatement = connection.prepareStatement(insertStatement)

// 处理name字段
value2.name match {
  case Some(nameStr) => preparedStatement.setString(1, nameStr)
  case None => preparedStatement.setNull(1, java.sql.Types.VARCHAR)
}

// 处理value字段：注意PostgreSQL的numeric类型对应java.sql.Types.NUMERIC更准确
value2.value match {
  case Some(num) => preparedStatement.setDouble(2, num)
  case None => preparedStatement.setNull(2, java.sql.Types.NUMERIC)
}

preparedStatement.executeUpdate()

// 关闭资源
preparedStatement.close()
connection.close()

更简洁的优化方案 #

1. 封装通用工具函数 #

把重复的Option处理逻辑抽成工具函数，减少冗余：

def setNullableString(ps: PreparedStatement, index: Int, value: Option[String]): Unit = {
  value match {
    case Some(v) => ps.setString(index, v)
    case None => ps.setNull(index, java.sql.Types.VARCHAR)
  }
}

def setNullableDouble(ps: PreparedStatement, index: Int, value: Option[Double]): Unit = {
  value match {
    case Some(v) => ps.setDouble(index, v)
    case None => ps.setNull(index, java.sql.Types.NUMERIC)
  }
}

// 使用示例
setNullableString(preparedStatement, 1, value2.name)
setNullableDouble(preparedStatement, 2, value2.value)

2. 利用Scala-Java Option转换 #

Scala的Option可以转换成Java的Optional，结合JDBC的setObject方法自动处理null：

import scala.jdk.OptionConverters._

// 直接用setObject，自动处理null值
preparedStatement.setObject(1, value2.name.toJava)
preparedStatement.setObject(2, value2.value.toJava)

这种方式代码最简洁，PostgreSQL JDBC驱动完全支持该用法。

内容的提问来源于stack exchange，提问作者bbgghh