orga0063
/
BootstrapAdmin
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat(#I12XME): MongoDB 自动建库 

#Issue
close https://gitee.com/LongbowEnterprise/dashboard/issues?id=I12XME
This commit is contained in:
Argo Window10 2019-11-16 14:35:27 +08:00
parent 412989012b
commit bf52b8b4b2
4 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
BootstrapAdmin.sln
View File

@ -33,7 +33,7 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "MongoDB", "MongoDB", "{A06A
db\MongoDB\Dicts.js = db\MongoDB\Dicts.js
db\MongoDB\Groups.js = db\MongoDB\Groups.js
db\MongoDB\init.js = db\MongoDB\init.js
db\MongoDB\install.cmd = db\MongoDB\install.cmd
db\MongoDB\install.ps1 = db\MongoDB\install.ps1
db\MongoDB\install.sh = db\MongoDB\install.sh
db\MongoDB\Navigations.js = db\MongoDB\Navigations.js
db\MongoDB\Roles.js = db\MongoDB\Roles.js

0
db/MongoDB/install.cmd → db/MongoDB/install.ps1
View File

8
src/admin/Bootstrap.DataAccess.MongoDB/AutoDB.cs
View File

@ -1,4 +1,7 @@
namespace Bootstrap.DataAccess.MongoDB
using Bootstrap.Security;
using MongoDB.Driver;
namespace Bootstrap.DataAccess.MongoDB
{
/// <summary>
/// 自动建库实体操作类
@ -11,7 +14,8 @@
/// </summary>
public override void EnsureCreated(string folder)
{
// UNDONE: 没有环境暂时未写代码
// 检查数据库是否存在
if (DbManager.Dicts.CountDocuments(FilterDefinition<BootstrapDict>.Empty) == 0) GenerateDB(folder);
}
}
}

22
src/admin/Bootstrap.DataAccess/AutoDB.cs
View File

@ -11,31 +11,27 @@ namespace Bootstrap.DataAccess
/// </summary>
public class AutoDB
{
private string _folder = "";
/// <summary>
/// 数据库检查方法
/// </summary>
public virtual void EnsureCreated(string folder)
{
_folder = folder;
using var db = Longbow.Data.DbManager.Create();
db.CommandTimeout = 5000;
switch (db.Provider.GetType().Name)
{
case "SQLiteDatabaseProvider":
if (db.ExecuteScalar<int>("SELECT count(*) FROM sqlite_master WHERE type='table' AND name='Users'") == 0) GenerateSQLiteDB(db);
if (db.ExecuteScalar<int>("SELECT count(*) FROM sqlite_master WHERE type='table' AND name='Users'") == 0) GenerateSQLiteDB(db, folder);
break;
case "SqlServerDatabaseProvider":
using (var newDB = ModifyConnectionString(db))
{
if (newDB.ExecuteScalar<int?>("select COUNT(*) from sys.databases where name = N'BootstrapAdmin'") == 0) GenerateDB();
if (newDB.ExecuteScalar<int?>("select COUNT(*) from sys.databases where name = N'BootstrapAdmin'") == 0) GenerateDB(folder);
}
break;
case "MySqlDatabaseProvider":
case "MariaDbDatabaseProvider":
// UNDONE: 本地没有环境此处代码未测试
if (db.ExecuteScalar<int>("select count(*) from information_schema.tables where table_name ='Users' and Table_Schema = 'BootstrapAdmin'") == 0) GenerateDB();
if (db.ExecuteScalar<int>("select count(*) from information_schema.tables where table_name ='Users' and Table_Schema = 'BootstrapAdmin'") == 0) GenerateDB(folder);
break;
}
}
@ -54,9 +50,8 @@ namespace Bootstrap.DataAccess
return new Database(string.Join(";", newsegs), provider);
}
private void GenerateSQLiteDB(IDatabase db)
private void GenerateSQLiteDB(IDatabase db, string folder)
{
var folder = _folder;
var initFile = Path.Combine(folder, "Install.sql");
if (File.Exists(initFile))
{
@ -72,13 +67,16 @@ namespace Bootstrap.DataAccess
}
}
private void GenerateDB()
/// <summary>
/// 执行建库脚本
/// </summary>
protected void GenerateDB(string folder)
{
// 检查 install.ps1 脚本
var file = Path.Combine(_folder, $"install.ps1");
var file = Path.Combine(folder, $"install.ps1");
if (File.Exists(file))
{
var psi = new ProcessStartInfo("powershell", $"{file} \"{_folder}\"");
var psi = new ProcessStartInfo("powershell", $"{file} \"{folder}\"");
var p = Process.Start(psi);
p.WaitForExit();
}