auth-server-java/common/generator-code/src/test/java/cn/bunny/JDBCTest.java

package cn.bunny;


import cn.bunny.dao.entity.ColumnMetaData;
import cn.bunny.dao.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 javax.sql.DataSource;
import java.sql.Connection;
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 DataSource dataSource;

    @BeforeEach
    public void setUp() throws Exception {
        try (Connection connection = dataSource.getConnection()) {
            metaData = connection.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)
                    .comment(remarks)
                    .tableCat(tableCat)
                    .tableSchem(tableSchem)
                    .tableType(tableType)
                    .typeCat(typeCat)
                    .typeSchem(typeSchem)
                    .typeName(typeName)
                    .selfReferencingColName(selfReferencingColName)
                    .refGeneration(refGeneration)
                    .build();

            System.out.println(tableMetaData);
        }
    }

    @Test
    void testAllTableComment() throws SQLException {
        ResultSet tables = metaData.getTables(null, null, "%" , new String[]{"TABLE"});
        List<TableMetaData> list = new ArrayList<>();

        while (tables.next()) {
            String tableName = tables.getString("TABLE_NAME" );
            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 = TableMetaData.builder()
                    .tableName(tableName).comment(remarks)
                    .tableCat(tableCat)
                    .tableSchem(tableSchem)
                    .tableType(tableType)
                    .typeCat(typeCat)
                    .typeSchem(typeSchem)
                    .typeName(typeName)
                    .selfReferencingColName(selfReferencingColName)
                    .refGeneration(refGeneration)
                    .build();
            list.add(tableMetaData);
        }

        System.out.println(list);
    }

    @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);
            }
        }

        System.out.println(columns);
    }
}