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

package com.crobox.clickhouse.dsl.language

import com.crobox.clickhouse.dsl._

trait DictionaryFunctionTokenizer {
  self: ClickhouseTokenizerModule =>

  private def tokenizeDictionaryGet(col: DictionaryGetFuncColumn[_],
                                    typeName: String)(implicit ctx: TokenizeContext): String = {
    val default = col.default
      .map(col => ctx.delimiter + tokenizeColumn(col.column))
      .getOrElse("")

    val orDefault = col.default.map(_ => "orDefault").getOrElse("")

    s"dictGet$typeName$orDefault(${tokenizeColumn(col.dictName.column)},${tokenizeColumn(col.attrName.column)},${tokenizeColumn(col.id.column)}$default)"
  }

  def tokenizeDictionaryFunction(col: DictionaryFuncColumn[_])(implicit ctx: TokenizeContext): String = col match {
    case col: DictGetUInt8 => tokenizeDictionaryGet(col, "UInt8")
    case col: DictGetUInt16 => tokenizeDictionaryGet(col, "UInt16")
    case col: DictGetUInt32 => tokenizeDictionaryGet(col, "UInt32")
    case col: DictGetUInt64 => tokenizeDictionaryGet(col, "UInt64")
    case col: DictGetInt8 => tokenizeDictionaryGet(col, "Int8")
    case col: DictGetInt16 => tokenizeDictionaryGet(col, "Int16")
    case col: DictGetInt32 => tokenizeDictionaryGet(col, "Int32")
    case col: DictGetInt64 => tokenizeDictionaryGet(col, "Int64")
    case col: DictGetFloat32 => tokenizeDictionaryGet(col, "Float32")
    case col: DictGetFloat64 => tokenizeDictionaryGet(col, "Float64")
    case col: DictGetDate => tokenizeDictionaryGet(col, "Date")
    case col: DictGetDateTime => tokenizeDictionaryGet(col, "DateTime")
    case col: DictGetUUID => tokenizeDictionaryGet(col, "UUID")
    case col: DictGetString => tokenizeDictionaryGet(col, "String")
    case DictIsIn(dictName: StringColMagnet[_], childId: ConstOrColMagnet[_], ancestorId: ConstOrColMagnet[_]) =>
      s"dictIsIn(${tokenizeColumn(dictName.column)},${tokenizeColumn(childId.column)},${tokenizeColumn(ancestorId.column)})"
    case DictGetHierarchy(dictName: StringColMagnet[_], id: ConstOrColMagnet[_]) =>
      s"dictGetHierarchy(${tokenizeColumn(dictName.column)},${tokenizeColumn(id.column)})"
    case DictHas(dictName: StringColMagnet[_], id: ConstOrColMagnet[_]) =>
      s"dictHas(${tokenizeColumn(dictName.column)},${tokenizeColumn(id.column)})"
    case _ => throw new IllegalArgumentException(s"Unsupported dictionary")
  }
}