Question

我的postgresql数据库中的一个函数在更新表时发送通知。我正在通过scalikejdbc轮询该postgresql数据库，以获取所有通知，然后对它们进行一些处理。 here说明了该过程。典型的反应式系统会更新sql表。我从java.sql.Connection获得PGConnection。然后，我以这种方式收到通知：

val notifications = Option(pgConnection.getNotifications).getOrElse(Array[PGNotification]())

我试图通过将获取大小设置为1000并禁用自动提交功能，以1000个为大块来获取通知。但是提取大小属性会被忽略。

有什么想法可以做到吗？我不想在我的通知数据集中的一张地图中处理成千上万的通知。

pgConnection.getNotifications.size可能很大，因此，此代码无法很好地扩展。

谢谢！

Answer 1

为了更好地扩展，请考虑使用postgresql-async和Akka Streams：前者是一个可以异步获取PostgreSQL通知的库，前者是提供反压的Reactive Streams实现（无需进行分页）。例如：

import akka.actor._
import akka.stream._
import akka.stream.scaladsl._

import com.github.mauricio.async.db.postgresql.PostgreSQLConnection
import com.github.mauricio.async.db.postgresql.util.URLParser

import scala.concurrent.duration._
import scala.concurrent.Await

class DbActor(implicit materializer: ActorMaterializer) extends Actor with ActorLogging {
  private implicit val ec = context.system.dispatcher

  val queue =  
    Source.queue[String](Int.MaxValue, OverflowStrategy.backpressure)
      .to(Sink.foreach(println))
      .run()

  val configuration = URLParser.parse("jdbc:postgresql://localhost:5233/my_db?user=dbuser&password=pwd")
  val connection = new PostgreSQLConnection(configuration)
  Await.result(connection.connect, 5 seconds)

  connection.sendQuery("LISTEN my_channel")
  connection.registerNotifyListener { message =>
    val msg = message.payload
    log.debug("Sending the payload: {}", msg)
    self ! msg
  }

  def receive = {
    case payload: String =>
      queue.offer(payload).pipeTo(self)
    case QueueOfferResult.Dropped =>
      log.warning("Dropped a message.")
    case QueueOfferResult.Enqueued =>
      log.debug("Enqueued a message.")
    case QueueOfferResult.Failure(t) =>
      log.error("Stream failed: {}", t.getMessage)
    case QueueOfferResult.QueueClosed =>
      log.debug("Stream closed.")
  }
}

上面的代码仅在发生时打印来自PostgreSQL的通知；您可以将Sink.foreach(println)替换为另一个Sink。要运行它：

import akka.actor._
import akka.stream.ActorMaterializer

object Example extends App {
  implicit val system = ActorSystem()
  implicit val materializer = ActorMaterializer()
  system.actorOf(Props(classOf[DbActor], materializer))
}

在PGConnection.getNotifications中获取大小

1 个答案: