All Downloads are FREE. Search and download functionalities are using the official Maven repository.

calasql-core_3.0.1.8.source-code.Config.scala Maven / Gradle / Ivy

package scalasql.core

/**
 * Things you to do to configure ScalaSql
 */
trait Config {

  /**
   * Render a sequence of tokens to a column label; used primarily for
   * making the generated queries more easily human readable.
   */
  def renderColumnLabel(tokens: Seq[String]): String = {
    val prefixedTokens =
      if (tokens.isEmpty || !Config.isNormalCharacter(tokens.head.head)) "res" +: tokens
      else tokens

    prefixedTokens
      .map(tableNameMapper)
      .mkString("_")
  }

  /**
   * Configures the underlying JDBC connection's `setFetchSize`. Can be overriden
   * on a per-query basis by passing `fetchSize = n` to `db.run`
   */
  def defaultFetchSize: Int = -1

  /**
   * Configures the underlying JDBC connection's `setQueryTimeout`. Can be overriden
   * on a per-query basis by passing `queryTimeoutSeconds = n` to `db.run`
   */
  def defaultQueryTimeoutSeconds: Int = -1

  /**
   * Translates table and column names from Scala `object` names to SQL names.
   *
   * Use [[tableNameMapper]] and [[columnNameMapper]] if you want different
   * translations for table and column names
   */
  def nameMapper(v: String): String = Config.camelToSnake(v)

  /**
   * Translates table names from Scala `object` names to SQL names.
   */
  def tableNameMapper(v: String): String = nameMapper(v)

  /**
   * Translates column names from Scala `case class` field names to SQL names.
   */
  def columnNameMapper(v: String): String = nameMapper(v)

  /**
   * Override this to log the executed SQL queries
   */
  def logSql(sql: String, file: String, line: Int): Unit = ()
}

object Config {
  def isNormalCharacter(c: Char) = (c >= 'a' && c <= 'z') || (c >= 'Z' && c <= 'Z') || c == '_'
  def camelToSnake(s: String) = {
    val chars = new collection.mutable.StringBuilder
    var lowercase = false
    for (c <- s) {
      if (c.isUpper) {
        if (lowercase == true) chars.append('_')
        chars.append(c.toLower)
        lowercase = false
      } else {
        chars.append(c)
        lowercase = true
      }

    }
    chars.toString()
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy