org.galaxio.gatling.jdbc.JdbcDsl.scala Maven / Gradle / Ivy

Go to download
package org.galaxio.gatling.jdbc

import io.gatling.core.protocol.Protocol
import io.gatling.core.session.Expression
import org.galaxio.gatling.jdbc.actions.actions._
import org.galaxio.gatling.jdbc.check.JdbcCheckSupport
import org.galaxio.gatling.jdbc.protocol._

trait JdbcDsl extends JdbcCheckSupport {
  def DB: JdbcProtocolBuilderBase.type                                                   = JdbcProtocolBuilderBase
  def jdbc(name: Expression[String]): DBBaseAction                                       = DBBaseAction(name)
  def insertInto(tableName: Expression[String], columns: Columns): BatchInsertBaseAction =
    BatchInsertBaseAction(tableName, columns)
  def update(tableName: Expression[String]): BatchUpdateBaseAction                       = BatchUpdateBaseAction(tableName)

  implicit def configStepToProtocolBuilder(step: JdbcProtocolBuilderConnectionSettingsStep): JdbcProtocolBuilder =
    step.protocolBuilder
  implicit def jdbcProtocolBuilder2jdbcProtocol(builder: JdbcProtocolBuilder): Protocol                          = builder.build
}