91 lines
3.1 KiB
Java
91 lines
3.1 KiB
Java
|
package cn.bunny;
|
||
|
|
|
||
|
|
|
||
|
|
import cn.bunny.config.DatabaseMetadataHolder;
|
||
|
|
import cn.bunny.entity.ColumnMetaData;
|
||
|
|
import cn.bunny.entity.TableMetaData;
|
||
|
|
import cn.bunny.utils.ConvertUtil;
|
||
|
|
import org.junit.jupiter.api.BeforeEach;
|
||
|
|
import org.junit.jupiter.api.Test;
|
||
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
||
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
||
|
|
|
||
|
|
import java.sql.DatabaseMetaData;
|
||
|
|
import java.sql.ResultSet;
|
||
|
|
import java.sql.SQLException;
|
||
|
|
import java.util.ArrayList;
|
||
|
|
import java.util.List;
|
||
|
|
|
||
|
|
@SpringBootTest
|
||
|
|
public class JDBCTest {
|
||
|
|
|
||
|
|
DatabaseMetaData metaData;
|
||
|
|
|
||
|
|
@Autowired
|
||
|
|
private DatabaseMetadataHolder metadataHolder;
|
||
|
|
|
||
|
|
@BeforeEach
|
||
|
|
public void setUp() throws Exception {
|
||
|
|
metaData = metadataHolder.getMetaData();
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* 获取表注释信息
|
||
|
|
*/
|
||
|
|
@Test
|
||
|
|
void testComment() throws SQLException {
|
||
|
|
String tableName = "sys_i18n";
|
||
|
|
|
||
|
|
TableMetaData tableMetaData;
|
||
|
|
|
||
|
|
ResultSet tables = metaData.getTables(null, null, tableName, new String[]{"TABLE"});
|
||
|
|
|
||
|
|
// 获取表的注释信息
|
||
|
|
if (tables.next()) {
|
||
|
|
String remarks = tables.getString("REMARKS" );
|
||
|
|
String tableCat = tables.getString("TABLE_CAT" );
|
||
|
|
String tableSchem = tables.getString("TABLE_SCHEM" );
|
||
|
|
String tableType = tables.getString("TABLE_TYPE" );
|
||
|
|
String typeCat = tables.getString("TYPE_CAT" );
|
||
|
|
String typeSchem = tables.getString("TYPE_SCHEM" );
|
||
|
|
String typeName = tables.getString("TYPE_NAME" );
|
||
|
|
String selfReferencingColName = tables.getString("SELF_REFERENCING_COL_NAME" );
|
||
|
|
String refGeneration = tables.getString("REF_GENERATION" );
|
||
|
|
|
||
|
|
tableMetaData = TableMetaData.builder()
|
||
|
|
.tableName(tableName)
|
||
|
|
.remarks(remarks)
|
||
|
|
.tableCat(tableCat)
|
||
|
|
.tableSchem(tableSchem)
|
||
|
|
.tableType(tableType)
|
||
|
|
.typeCat(typeCat)
|
||
|
|
.typeSchem(typeSchem)
|
||
|
|
.typeName(typeName)
|
||
|
|
.selfReferencingColName(selfReferencingColName)
|
||
|
|
.refGeneration(refGeneration)
|
||
|
|
.build();
|
||
|
|
|
||
|
|
System.out.println(tableMetaData);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
void testColumnInfo() throws SQLException {
|
||
|
|
List<ColumnMetaData> columns = new ArrayList<>();
|
||
|
|
|
||
|
|
try (ResultSet columnsRs = metaData.getColumns(null, null, "sys_i18n" , null)) {
|
||
|
|
while (columnsRs.next()) {
|
||
|
|
ColumnMetaData column = new ColumnMetaData();
|
||
|
|
column.setColumnName(columnsRs.getString("COLUMN_NAME" ));
|
||
|
|
column.setFieldName(ConvertUtil.convertToFieldName(column.getColumnName()));
|
||
|
|
column.setJdbcType(columnsRs.getString("TYPE_NAME" ));
|
||
|
|
column.setJavaType(ConvertUtil.convertToJavaType(column.getJdbcType()));
|
||
|
|
column.setComment(columnsRs.getString("REMARKS" ));
|
||
|
|
|
||
|
|
columns.add(column);
|
||
|
|
System.out.println(column);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|