tech.mlsql.dsl.processor.PreProcessListener.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of streamingpro-mlsql-spark_2.4 Show documentation

The newest version!

package tech.mlsql.dsl.processor

import streaming.dsl.parser.DSLSQLParser.SqlContext
import streaming.dsl.{ScriptSQLExecListener, SetAdaptor}
import streaming.parser.lisener.BaseParseListenerextends
import tech.mlsql.dsl.adaptor.{CommandAdaptor, StatementAdaptor}

import scala.collection.mutable.ArrayBuffer

/**
  * 2019-04-11 WilliamZhu([email protected])
  */
class PreProcessListener(val scriptSQLExecListener: ScriptSQLExecListener) extends BaseParseListenerextends {

  private val _statements = new ArrayBuffer[String]()

  def toScript = {
    _statements.mkString(";") + ";"
  }

  def addStatement(v: String) = {
    _statements += v
    this
  }

  override def exitSql(ctx: SqlContext): Unit = {

    ctx.getChild(0).getText.toLowerCase() match {
      case item if item.startsWith("!") =>
        new CommandAdaptor(this).parse(ctx)
      case "set" => {
        new SetAdaptor(scriptSQLExecListener).parse(ctx)
        new StatementAdaptor(this).parse(ctx)
      }
      case _ => new StatementAdaptor(this).parse(ctx)
    }

  }


}