org.archguard.linter.rule.sql.DatamapRuleVisitor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rule-sql Show documentation

ArchGuard is a architecture governance tool which can analysis architecture in container, component, code level, create architecture fitness functions, and anaysis system dependencies..

The newest version!

package org.archguard.linter.rule.sql

import net.sf.jsqlparser.parser.CCJSqlParserUtil
import net.sf.jsqlparser.statement.Statement
import org.archguard.rule.core.Issue
import org.archguard.rule.core.IssuePosition
import org.archguard.rule.core.Rule
import org.archguard.rule.core.RuleContext
import org.archguard.rule.core.RuleSet
import org.archguard.rule.core.RuleType
import org.archguard.rule.core.RuleVisitor
import org.archguard.context.CodeDatabaseRelation

data class CodeSqlStmt(
    val packageName: String = "",
    val className: String = "",
    val functionName: String = "",
    val statements: List = listOf()
)

class DatamapRuleVisitor(relations: List): RuleVisitor(relations) {
    private val codeSqlStmts: List = relations.map { relation ->
        val stmts = relation.sqls.mapNotNull {
            try {
                CCJSqlParserUtil.parseStatements(it)
            } catch (e: Exception) {
                null
            }
        }.flatMap {
            it.statements
        }

        CodeSqlStmt(relation.packageName, relation.className, relation.functionName, stmts)
    }

    override fun visitor(ruleSets: Iterable): List {
        val results: MutableList = mutableListOf()
        val context = RuleContext()

        ruleSets.forEach { ruleSet ->
            ruleSet.rules.forEach { rule ->
                val sqlRule = rule as SqlRule
                codeSqlStmts.map {
                    sqlRule.visit(it.statements, context, fun(rule: Rule, position: IssuePosition) {
                        results += Issue(
                            position,
                            ruleId = rule.key,
                            name = rule.name,
                            detail = rule.description,
                            ruleType = RuleType.SQL_SMELL,
                            fullName = "${it.packageName}:${it.className}:${it.functionName}",
                            source = it.toString()
                        )
                    })
                }
            }
        }

        return results
    }

}