io.edurt.datacap.parser.mysql.MySQLHelper.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of datacap-parser-mysql Show documentation

DataCap - Parser for mysql

There is a newer version: 2024.4.0

package io.edurt.datacap.parser.mysql

import io.edurt.datacap.parser.ParserResponse
import io.edurt.datacap.parser.model.Column
import io.edurt.datacap.parser.model.Table
import io.edurt.datacap.parser.mysql.companion.MySQLEndpoint
import io.edurt.datacap.parser.mysql.companion.ParserException
import io.edurt.datacap.parser.mysql.companion.ast.ParserOptions
import io.edurt.datacap.parser.mysql.companion.tree.QuerySpecification
import io.edurt.datacap.parser.mysql.companion.tree.Statement
import io.edurt.datacap.parser.type.EngineType
import io.edurt.datacap.parser.type.StatementType
import org.slf4j.Logger
import org.slf4j.LoggerFactory

object MySQLHelper {
    private val log: Logger = LoggerFactory.getLogger(MySQLHelper::class.java)

    @JvmStatic
    fun parse(sql: String): ParserResponse {
        val response = ParserResponse()
        response.engine = EngineType.MYSQL
        try {
            response.isParser = true
            check(MySQLEndpoint().createStatement(sql, ParserOptions.DecimalLiteralTreatment.AS_DECIMAL), response)
        }
        catch (e: ParserException) {
            response.isParser = false
            response.message = e.message
            log.error("Parse error", e)
        }
        return response
    }

    @Throws(ParserException::class)
    private fun check(statement: Statement, response: ParserResponse) {
        when (statement) {
            is QuerySpecification -> {
                response.type = StatementType.SELECT

                val table = Table()
                statement.from.table.children.forEach {
                    table.name = it.name
                    table.alias = it.alias
                }

                statement.select.children.forEach {
                    val column = Column()
                    column.name = it.name
                    table.columns.add(column)
                }

                statement.limit.children.forEach {
                    table.limit = it.limit
                }
                response.table = table
            }

            else -> throw ParserException("Unsupported statement: " + statement.javaClass)
        }
    }
}