• Home
• com.emarsys
• rdb-connector-redshift_2.12
• 4e6e2e40
• source code
• RedshiftSimpleSelect.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.emarsys.rdb.connector.redshift.RedshiftSimpleSelect.scala Maven / Gradle / Ivy

package com.emarsys.rdb.connector.redshift

import akka.NotUsed
import akka.stream.scaladsl.Source
import com.emarsys.rdb.connector.common.ConnectorResponse
import com.emarsys.rdb.connector.common.models.SimpleSelect
import com.emarsys.rdb.connector.common.defaults.SqlWriter._
import RedshiftSqlWriters._

import scala.concurrent.duration.FiniteDuration

trait RedshiftSimpleSelect extends RedshiftStreamingQuery {
  self: RedshiftConnector =>

  override def simpleSelect(
      select: SimpleSelect,
      timeout: FiniteDuration
  ): ConnectorResponse[Source[Seq[String], NotUsed]] = {
    streamingQuery(timeout)(select.toSql)
  }

  override def runSelectWithGroupLimit(
      select: SimpleSelect,
      groupLimit: Int,
      references: Seq[String],
      timeout: FiniteDuration
  ): ConnectorResponse[Source[Seq[String], NotUsed]] = {
    val sql = select.toSql(selectWithGroupLimitWriter(groupLimit, references))
    streamingQuery(timeout)(sql).map(_.map(_.map(_.dropRight(1))))
  }
}