45 lines
1.1 KiB
C#
45 lines
1.1 KiB
C#
using MiaoYu.Core.CodeGenerator.Abstractions;
|
||
|
||
namespace MiaoYu.Core.CodeGenerator.Core;
|
||
|
||
/// <summary>
|
||
/// 数据源扩展方法
|
||
/// </summary>
|
||
public static class DataSourceExtensions
|
||
{
|
||
/// <summary>
|
||
/// 从 Schema 中提取数据库标识
|
||
/// </summary>
|
||
/// <param name="schema">Schema字符串</param>
|
||
/// <returns>数据库标识</returns>
|
||
public static string ExtractDatabaseKey(this string schema)
|
||
{
|
||
if (string.IsNullOrWhiteSpace(schema))
|
||
return DataSourceConstants.Admin;
|
||
|
||
if (schema.Contains("."))
|
||
{
|
||
var parts = schema.Split('.');
|
||
return parts.Length > 1 ? parts[1] : DataSourceConstants.Admin;
|
||
}
|
||
|
||
return DataSourceConstants.Admin;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 清理 Schema（移除数据库标识）
|
||
/// </summary>
|
||
/// <param name="schema">Schema字符串</param>
|
||
/// <returns>清理后的Schema</returns>
|
||
public static string CleanSchema(this string schema)
|
||
{
|
||
if (string.IsNullOrWhiteSpace(schema))
|
||
return schema;
|
||
|
||
return schema.Contains(".")
|
||
? schema.Split('.')[0]
|
||
: schema;
|
||
}
|
||
}
|
||
|