com.lin.sql.SqlSelector Maven / Gradle / Ivy

Go to download

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

A lightweight SQL data creator.

There is a newer version: 1.0.8

package com.lin.sql;

import com.lin.asserts.Asserts;
import com.lin.helper.DatabaseHelper;
import org.jooq.Record;
import org.jooq.Result;
import org.jooq.impl.DSL;

import java.sql.Connection;
import	java.util.List;

/**
 * SQL 选择器
 * @author lkmc2
 * @since 1.0.7
 */
public final class SqlSelector {

    /**
     * 选择表中的一列数据
     * @param tableName 表名
     * @param columnName 列表
     * @return 表中的一列数据
     */
    public static List selectColumn(String tableName, String columnName) {
        String sql = String.format("select distinct %s from %s", columnName, tableName);
        return selectColumn(sql);
    }

    /**
     * 选择表中的一列数据
     * @param sql sq 语句
     * @return 表中的一列数据
     */
    public static List selectColumn(String sql) {
        Connection connection = DatabaseHelper.getConnection();
        Asserts.isTrue(connection != null, "数据库连接获取失败，请先在 DBTools 中设置连接参数");

        // 执行 sql 并获取查询结果
        Result records = DSL.using(connection).fetch(sql);

        // 获取第 1 列的值
        return records.getValues(0, String.class);
    }

}