69 lines
2.4 KiB
C#
69 lines
2.4 KiB
C#
using SqlSugar;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using ARW.Model;
|
|
|
|
namespace ARW.CodeGenerator.Service
|
|
{
|
|
public class CodeGeneraterService : DbProvider
|
|
{
|
|
/// <summary>
|
|
/// 获取所有数据库名
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public List<string> GetAllDataBases()
|
|
{
|
|
var db = GetSugarDbContext();
|
|
var templist = db.DbMaintenance.GetDataBaseList(db);
|
|
|
|
return templist;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取所有表
|
|
/// </summary>
|
|
/// <param name="dbName"></param>
|
|
/// <param name="tableName"></param>
|
|
/// <param name="pager"></param>
|
|
/// <returns></returns>
|
|
public List<DbTableInfo> GetAllTables(string dbName, string tableName, PagerInfo pager)
|
|
{
|
|
var tableList = GetSugarDbContext(dbName).DbMaintenance.GetTableInfoList(true);
|
|
if (!string.IsNullOrEmpty(tableName))
|
|
{
|
|
tableList = tableList.Where(f => f.Name.ToLower().Contains(tableName.ToLower())).ToList();
|
|
}
|
|
//tableList = tableList.Where(f => !new string[] { "gen", "sys_" }.Contains(f.Name)).ToList();
|
|
pager.TotalNum = tableList.Count;
|
|
return tableList.Skip(pager.PageSize * (pager.PageNum - 1)).Take(pager.PageSize).OrderBy(f => f.Name).ToList();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取单表数据
|
|
/// </summary>
|
|
/// <param name="dbName"></param>
|
|
/// <param name="tableName"></param>
|
|
/// <returns></returns>
|
|
public DbTableInfo GetTableInfo(string dbName, string tableName)
|
|
{
|
|
var tableList = GetSugarDbContext(dbName).DbMaintenance.GetTableInfoList(true);
|
|
if (!string.IsNullOrEmpty(tableName))
|
|
{
|
|
return tableList.Where(f => f.Name.Equals(tableName, System.StringComparison.OrdinalIgnoreCase)).FirstOrDefault();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
/// <summary>
|
|
/// 获取列信息
|
|
/// </summary>
|
|
/// <param name="dbName"></param>
|
|
/// <param name="tableName"></param>
|
|
/// <returns></returns>
|
|
public List<DbColumnInfo> GetColumnInfo(string dbName, string tableName)
|
|
{
|
|
return GetSugarDbContext(dbName).DbMaintenance.GetColumnInfosByTableName(tableName, true);
|
|
}
|
|
}
|
|
}
|