org.sagacity.sqltoy.link.TableApi Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sagacity-sqltoy Show documentation
sqltoy core code
There is a newer version: 5.6.31.jre8
/**
 * 
 */
package org.sagacity.sqltoy.link;

import java.util.List;

import javax.sql.DataSource;

import org.sagacity.sqltoy.SqlToyContext;
import org.sagacity.sqltoy.config.model.SqlToyConfig;
import org.sagacity.sqltoy.config.model.SqlType;
import org.sagacity.sqltoy.model.ColumnMeta;
import org.sagacity.sqltoy.model.TableMeta;

/**
 * @project sagacity-sqltoy
 * @description 提供一个获取数据库表信息和操作表信息的TableApi集合
 * @author zhongxuchen
 * @version v1.0, Date:2023年5月5日
 * @modify 2023年5月5日,修改说明
 */
public class TableApi extends BaseLink {

	/**
	 * 
	 */
	private static final long serialVersionUID = -6239897514441516513L;

	/**
	 * @param sqlToyContext
	 * @param dataSource
	 */
	public TableApi(SqlToyContext sqlToyContext, DataSource dataSource) {
		super(sqlToyContext, dataSource);
	}

	public TableApi dataSource(DataSource dataSource) {
		this.dataSource = dataSource;
		this.defaultDataSource = false;
		return this;
	}

	/**
	 * @TODO 获得表的字段信息
	 * @param catalog
	 * @param schema
	 * @param tableName
	 * @return
	 */
	public List getTableColumns(final String catalog, final String schema, final String tableName) {
		return dialectFactory.getTableColumns(sqlToyContext, catalog, schema, tableName, getDataSource(null));
	}

	/**
	 * @TODO 获得数据库的表信息
	 * @param catalog
	 * @param schema
	 * @param tableName
	 * @return
	 */
	public List getTables(final String catalog, final String schema, final String tableName) {
		return dialectFactory.getTables(sqlToyContext, catalog, schema, tableName, getDataSource(null));
	}

	/**
	 * @TODO 清空表数据(根据pojo来获取实际表名称)
	 * @param entityClass
	 */
	public void truncate(Class entityClass) {
		String tableName = sqlToyContext.getEntityMeta(entityClass).getSchemaTable(null, null);
		String sql = "truncate table ".concat(tableName);
		SqlToyConfig sqlToyConfig = sqlToyContext.getSqlToyConfig(sql, SqlType.delete, super.getDialect(), null);
		dialectFactory.executeSql(sqlToyContext, sqlToyConfig, null, null, null, dataSource);
	}

	/**
	 * @TODO 清空表数据
	 * @param tableName
	 */
	public void truncate(String tableName) {
		SqlToyConfig sqlToyConfig = sqlToyContext.getSqlToyConfig("truncate table ".concat(tableName), SqlType.delete,
				super.getDialect(), null);
		dialectFactory.executeSql(sqlToyContext, sqlToyConfig, null, null, null, dataSource);
	}

	/**
	 * @TODO 删除表结构
	 * @param entityClass
	 */
	public void drop(Class entityClass) {
		String tableName = sqlToyContext.getEntityMeta(entityClass).getSchemaTable(null, null);
		String sql = "drop table ".concat(tableName);
		SqlToyConfig sqlToyConfig = sqlToyContext.getSqlToyConfig(sql, SqlType.delete, super.getDialect(), null);
		dialectFactory.executeSql(sqlToyContext, sqlToyConfig, null, null, null, dataSource);
	}

	public void drop(String tableName) {
		SqlToyConfig sqlToyConfig = sqlToyContext.getSqlToyConfig("drop table ".concat(tableName), SqlType.delete,
				super.getDialect(), null);
		dialectFactory.executeSql(sqlToyContext, sqlToyConfig, null, null, null, dataSource);
	}

//	/**
//	 * @TODO 创建表
//	 * @param tableMeta
//	 * @param createOrReplace
//	 */
//	public void create(TableMeta tableMeta, boolean createOrReplace) {
//
//	}
}