private List <SqlTreeDto> GetSqlTree(List <DatabaseEntity> sqlConnDtos, SqlSearchDto searchDto)
{
List <SqlTreeDto> tree = new List <SqlTreeDto>();
foreach (var node in sqlConnDtos)
{
var dto = new SqlTreeDto()
{
isParent = true,
id = "db-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = searchDto.Id,
name = node.DatabaseName,
open = false,
icon = GetIcon("database"),
SqlType = searchDto.SqlType,
SqlConnId = searchDto.SqlConnId,
Database = node.DatabaseName,
};
tree.Add(dto);
tree.Add(CreateTableTree(dto, searchDto));
tree.Add(CreateViewTree(dto, searchDto));
tree.Add(CreateProcedureTree(dto, searchDto));
tree.Add(CreateFunctionTree(dto, searchDto));
}
return(tree);
}
private List <SqlTreeDto> GetSqlTree(List <SqlConnDto> sqlConnDtos, SqlSearchDto searchDto)
{
List <SqlTreeDto> tree = new List <SqlTreeDto>();
foreach (var node in sqlConnDtos)
{
var dto = new SqlTreeDto()
{
isParent = true,
id = "c-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = "root",
name = node.Name,
open = false,
icon = GetIcon(node.SqlType.ToLower()),
SearchType = "GetDatabaseNames",
SqlConnId = node.Id,
SqlType = node.SqlType,
};
tree.Add(dto);
if (node.SqlType.Equals("sqlite", StringComparison.OrdinalIgnoreCase))
{
dto.SearchType = null;
tree.Add(CreateTableTree(dto, searchDto));
tree.Add(CreateViewTree(dto, searchDto));
}
}
return(tree);
}
private SqlTreeDto CreateFunctionTree(SqlTreeDto dto, SqlSearchDto searchDto)
{
return(new SqlTreeDto()
{
isParent = true,
id = "fs-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = dto.id,
name = "函数",
open = true,
icon = GetIcon("function"),
SqlType = searchDto.SqlType,
SqlConnId = searchDto.SqlConnId,
Database = dto.Database,
SearchType = "GetFunctionNames",
});
}
private SqlTreeDto CreateProcedureTree(SqlTreeDto dto, SqlSearchDto searchDto)
{
return(new SqlTreeDto()
{
isParent = true,
id = "ps-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = dto.id,
name = "存储过程",
open = true,
icon = GetIcon("procedure"),
SqlType = searchDto.SqlType,
SqlConnId = searchDto.SqlConnId,
Database = dto.Database,
SearchType = "GetProcedureNames",
});
}
private SqlTreeDto CreateViewTree(SqlTreeDto dto, SqlSearchDto searchDto)
{
return(new SqlTreeDto()
{
isParent = true,
id = "vs-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = dto.id,
name = "视图",
open = true,
icon = GetIcon("view"),
SqlType = dto.SqlType,
SqlConnId = dto.SqlConnId,
Database = dto.Database,
SearchType = "GetViewNames",
});
}
private SqlTreeDto CreateTableTree(SqlTreeDto dto, SqlSearchDto searchDto)
{
return(new SqlTreeDto()
{
isParent = true,
id = "ts-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = dto.id,
name = "表",
open = true,
icon = GetIcon("table"),
SqlType = dto.SqlType,
SqlConnId = dto.SqlConnId,
Database = dto.Database,
Search = dto.Search,
Schema = dto.Schema,
SearchType = "GetTableNames",
});
}
private List <SqlTreeDto> GetSqlTree(List <FunctionEntity> sqlConnDtos, SqlSearchDto searchDto)
{
List <SqlTreeDto> tree = new List <SqlTreeDto>();
foreach (var node in sqlConnDtos)
{
var dto = new SqlTreeDto()
{
id = "f-" + Guid.NewGuid().ToString().Replace("-", ""),
pId = searchDto.Id,
name = node.FunctionName,
SqlType = searchDto.SqlType,
SqlConnId = searchDto.SqlConnId,
Database = node.DatabaseName,
Schema = node.SchemaName,
Search = node.FunctionName,
icon = GetIcon("function"),
};
if (searchDto.SqlType.Equals("sqlserver", StringComparison.OrdinalIgnoreCase))
{
if (node.SchemaName != "dbo")
{
dto.name = node.SchemaName + "." + node.FunctionName;
}
}
if (searchDto.SqlType.Equals("PostgreSQL", StringComparison.OrdinalIgnoreCase))
{
if (node.SchemaName != "public")
{
dto.name = node.SchemaName + "." + node.FunctionName;
}
}
tree.Add(dto);
}
return(tree);
}
