5 - 教你如何实现 JDBC 列权限控制

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门，即可获取！

val stmt = SQLUtils.parseSingleStatement(sql, dbType)

val visitor = SQLUtils.createSchemaStatVisitor(dbType)

stmt.accept(visitor)

visitor.getTables().asScala.map { f =>

val dbAndTable = f._1.getName

if (dbAndTable.contains(“.”)) {

val Array(db, table) = dbAndTable.split(“\.”, 2)

(db ,table)

} else (dbAndTable, None)

}.toList

}

2. 查询表的元数据信息

//streaming.core.datasource.JDBCUtils

def queryTableWithColumnsInDriver(options: Map[String, String] ,tableList: List[String]) = {

val tableAndCols = mutable.HashMap.empty[String, mutable.HashMap[String ,String]]

val driver = options(“driver”)

val url = options(“url”)

Class.forName(driver)

val connection = java.sql.DriverManager.getConnection(url, options(“user”), options(“password”))

try {

val dbMetaData = connection.getMetaData()

tableList.foreach(table => {

val rs = dbMetaData.getColumns(null, null, table, “%”)

val value = tableAndCols.getOrElse(table, mutable.HashMap.empty[String ,String])

while(rs.next()){

value += (rs.getString(“COLUMN_NAME”) -> rs.getString(“TYPE_NAME”))

}

tableAndCols.update(table, value)

rs.close()

})

} finally {

if (connection != null)

connection.close()

}

tableAndCols

}

3. 根据表的元数据信息，构造表的创建语句，用于构造SchemaRepository

//streaming.core.datasource.JDBCUtils

def tableColumnsToCreateSql(tableClos: mutable.HashMap[String, mutable.HashMap[String, String]]) = {

val createSqlList = mutable.ArrayBuffer.empty[String]

tableClos.foreach(table => {

var createSql = “create table " + table._1 + " (” +

table._2.map(m => m._1 + " " + m._2)

.mkString(“,”) +

" )"

createSqlList += createSql

})

createSqlList.toList

}

4. 解析SQL的表和列

//tech.mlsql.sql.MLSQLSQLParser

def extractTableWithColumns(dbType :String ,sql :String ,createSchemaList :List[String]) = {

val tableAndCols = mutable.HashMap.empty[String, mutable.HashSet[String]]

val repository = new SchemaRepository(dbType)

createSchemaList.foreach(repository.console(_))

val stmtList = SQLUtils.parseStatements(sql, dbType)

val stmt = stmtList.get(0).asInstanceOf[SQLSelectStatement]

repository.resolve(stmt)

val statVisitor = SQLUtils.createSchemaStatVisitor(dbType)

stmt.accept(statVisitor)

val iter = statVisitor.getColumns().iterator()

while(iter.hasNext()){

val c = iter.next()

if(c.isSelect()){

val value = tableAndCols.getOrElse(c.getTable, mutable.HashSet.empty[String])

value.add(c.getName)

tableAndCols.update(c.getTable, value)

}

}

tableAndCols

}

下面来看一下整个流程：

val sql =

“”"

|select name ,country

| from c5_user a

| left join c5_country b

| on a.id = b.id

“”".stripMargin

val params = Map(“driver”->“com.mysql.jdbc.Driver” ,

“url”->“jdbc:mysql://127.0.0.1:3306/test” ,

“user”->“root” ,

“password”->“mlsql”)

val rest = extractTablesFromSQL(sql)

val tableList = extractTablesFromSQL(sql).map(_._1)

val tableColsMap = queryTableWithColumnsInDriver(params, tableList)

val createSqlList = tableColumnsToCreateSql(tableColsMap)

println(“创建SQL：”)

createSqlList.foreach(println(_))

val tableAndCols = extractTableWithColumns(JdbcConstants.MYSQL, sql, createSqlList)

println(“表列信息：”)

println(tableAndCols)

--------------------------------------------------------------------------------------

输出结果：

创建SQL：

create table c5_country (country VARCHAR,id VARCHAR )

create table c5_user (name VARCHAR,id VARCHAR )

表列信息：

Map(c5_country -> Set(country), c5_user -> Set(name))

写在最后

很多人感叹“学习无用”，实际上之所以产生无用论，是因为自己想要的与自己所学的匹配不上，这也就意味着自己学得远远不够。无论是学习还是工作，都应该有主动性，所以如果拥有大厂梦，那么就要自己努力去实现它。

最后祝愿各位身体健康，顺利拿到心仪的offer！

由于文章的篇幅有限，所以这次的蚂蚁金服和京东面试题答案整理在了PDF文档里

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门，即可获取！
大厂梦，那么就要自己努力去实现它。

最后祝愿各位身体健康，顺利拿到心仪的offer！

由于文章的篇幅有限，所以这次的蚂蚁金服和京东面试题答案整理在了PDF文档里

[外链图片转存中…(img-pcOHOoQG-1714406470523)]

[外链图片转存中…(img-ghZt1XPK-1714406470524)]

[外链图片转存中…(img-yGVospSr-1714406470524)]
《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门，即可获取！