• Home
• com.crobox.clickhouse
• dsl_3
• 1.2.6
• source code
• StringFunctions.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.crobox.clickhouse.dsl.column.StringFunctions.scala Maven / Gradle / Ivy

package com.crobox.clickhouse.dsl.column

import com.crobox.clickhouse.dsl.{Column, ExpressionColumn}

trait StringFunctions { self: Magnets =>

  abstract class StringFunctionCol[+V](val innerCol: Column) extends ExpressionColumn[V](innerCol)

  case class Length(col: EmptyNonEmptyCol[_])     extends StringFunctionCol[Int](col.column)
  case class LengthUTF8(col: EmptyNonEmptyCol[_]) extends StringFunctionCol[String](col.column)
  case class Lower(col: StringColMagnet[_])       extends StringFunctionCol[String](col.column)
  case class Upper(col: StringColMagnet[_])       extends StringFunctionCol[String](col.column)
  case class LowerUTF8(col: StringColMagnet[_])   extends StringFunctionCol[String](col.column)
  case class UpperUTF8(col: StringColMagnet[_])   extends StringFunctionCol[String](col.column)
  case class Reverse(col: StringColMagnet[_])     extends StringFunctionCol[String](col.column)
  case class ReverseUTF8(col: StringColMagnet[_]) extends StringFunctionCol[String](col.column)
  case class Concat(col1: StringColMagnet[_], col2: StringColMagnet[_], columns: StringColMagnet[_]*)
      extends StringFunctionCol[String](col1.column)
  case class Substring(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_])
      extends StringFunctionCol[String](col.column)
  case class SubstringUTF8(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_])
      extends StringFunctionCol[String](col.column)
  case class AppendTrailingCharIfAbsent(col: StringColMagnet[_], c: StringColMagnet[_])
      extends StringFunctionCol[String](col.column)
  case class ConvertCharset(col: StringColMagnet[_], from: StringColMagnet[_], to: StringColMagnet[_])
      extends StringFunctionCol[String](col.column)

  // TODO: Enum the charsets?

  trait StringOps { self: StringColMagnet[_] with EmptyNonEmptyCol[_] =>

    def length()      = Length(self)
    def lengthUTF8()  = LengthUTF8(self)
    def lower()       = Lower(self)
    def upper()       = Upper(self)
    def lowerUTF8()   = LowerUTF8(self)
    def upperUTF8()   = UpperUTF8(self)
    def reverse()     = Reverse(self)
    def reverseUTF8() = ReverseUTF8(self)

    def ||(col2: StringColMagnet[_]) = Concat(self, col2)

    def concat(col2: StringColMagnet[_], coln: StringColMagnet[_]*) =
      Concat(self, col2: StringColMagnet[_], coln: _*)

    def substring(offset: NumericCol[_], length: NumericCol[_]) =
      Substring(self, offset: NumericCol[_], length: NumericCol[_])

    def substringUTF8(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_]) =
      SubstringUTF8(self, offset: NumericCol[_], length: NumericCol[_])

    def appendTrailingCharIfAbsent(col: StringColMagnet[_], c: StringColMagnet[_]) =
      AppendTrailingCharIfAbsent(self, c: StringColMagnet[_])

    def convertCharset(from: StringColMagnet[_], to: StringColMagnet[_]) =
      ConvertCharset(self, from: StringColMagnet[_], to: StringColMagnet[_])
  }

  def length(col: EmptyNonEmptyCol[_])     = Length(col: EmptyNonEmptyCol[_])
  def lengthUTF8(col: EmptyNonEmptyCol[_]) = LengthUTF8(col: EmptyNonEmptyCol[_])
  def lower(col: StringColMagnet[_])       = Lower(col: StringColMagnet[_])
  def upper(col: StringColMagnet[_])       = Upper(col: StringColMagnet[_])
  def lowerUTF8(col: StringColMagnet[_])   = LowerUTF8(col: StringColMagnet[_])
  def upperUTF8(col: StringColMagnet[_])   = UpperUTF8(col: StringColMagnet[_])
  def reverse(col: StringColMagnet[_])     = Reverse(col: StringColMagnet[_])
  def reverseUTF8(col: StringColMagnet[_]) = ReverseUTF8(col: StringColMagnet[_])

  def concat(col: StringColMagnet[_], col2: StringColMagnet[_], coln: StringColMagnet[_]*) =
    Concat(col: StringColMagnet[_], col2: StringColMagnet[_], coln: _*)

  def substring(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_]) =
    Substring(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_])

  def substringUTF8(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_]) =
    SubstringUTF8(col: StringColMagnet[_], offset: NumericCol[_], length: NumericCol[_])

  def appendTrailingCharIfAbsent(col: StringColMagnet[_], c: StringColMagnet[_]) =
    AppendTrailingCharIfAbsent(col: StringColMagnet[_], c: StringColMagnet[_])

  def convertCharset(col: StringColMagnet[_], from: StringColMagnet[_], to: StringColMagnet[_]) =
    ConvertCharset(col: StringColMagnet[_], from: StringColMagnet[_], to: StringColMagnet[_])
}