AOI_V1/MaiMuAOI/Service/InitDB.cs

123 lines
5.9 KiB
C#
Raw Normal View History

2023-11-20 13:26:38 +08:00
using Models;
using MySql.Data.MySqlClient;
using SqlSugar;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
namespace Service
{
public class InitDB
{
public static string ConnectionString;
public static void initDB(bool dropTable=false)
{
ConnectionConfig connectionConfig = new ConnectionConfig() {
ConnectionString = ConnectionString,
DbType = DbType.MySql,
IsAutoCloseConnection = true
};
//创建数据库对象
using (SqlSugarClient db = new SqlSugarClient(connectionConfig))
{
db.Aop.OnLogExecuting = (sql, pars) =>
{
Console.WriteLine(sql);//输出sql,查看执行sql 性能无影响
};
//create db
db.DbMaintenance.CreateDatabase();
//db.DbMaintenance.TruncateTable<Order>();//删除记录
//===建表
if (dropTable && db.DbMaintenance.IsAnyTable("Classes", false)) db.DbMaintenance.DropTable("Classes");
if (dropTable && db.DbMaintenance.IsAnyTable("Step", false)) db.DbMaintenance.DropTable("Steps");
if (dropTable && db.DbMaintenance.IsAnyTable("StepProcess", false)) db.DbMaintenance.DropTable("StepProcess");
if (dropTable && db.DbMaintenance.IsAnyTable("Product", false)) db.DbMaintenance.DropTable("Product");
if (dropTable && db.DbMaintenance.IsAnyTable("Attachment", false)) db.DbMaintenance.DropTable("Attachment");
if (dropTable && db.DbMaintenance.IsAnyTable("BatchHistory", false)) db.DbMaintenance.DropTable("BatchHistory");
if (dropTable && db.DbMaintenance.IsAnyTable("QualifiedCriterion", false)) db.DbMaintenance.DropTable("QualifiedCriterion");
if (dropTable && db.DbMaintenance.IsAnyTable("ProductReviseProcess", false)) db.DbMaintenance.DropTable("ProductReviseProcess");//????ProductReviseProcessList
if (dropTable && db.DbMaintenance.IsAnyTable("ProductAssistProcess", false)) db.DbMaintenance.DropTable("ProductAssistProcess");
if (dropTable && db.DbMaintenance.IsAnyTable("Order", false)) db.DbMaintenance.DropTable("Order");
if (dropTable && db.DbMaintenance.IsAnyTable("OrderHistory", false)) db.DbMaintenance.DropTable("OrderHistory");
if (dropTable && db.DbMaintenance.IsAnyTable("SizeTagData", false)) db.DbMaintenance.DropTable("SizeTagData");
if (dropTable && db.DbMaintenance.IsAnyTable("DefectInfo", false)) db.DbMaintenance.DropTable("DefectInfo");
//===添加与更新表
db.CodeFirst.InitTables<Models.Classes>();
db.CodeFirst.InitTables<Models.Step>();
db.CodeFirst.InitTables<Models.StepProcess>();
db.CodeFirst.InitTables<Models.Product>();
db.CodeFirst.InitTables<Models.Attachment>();
db.CodeFirst.InitTables<Models.BatchHistory>();
db.CodeFirst.InitTables<Models.QualifiedCriterion>();
db.CodeFirst.InitTables<Models.ProductProcess>();
db.CodeFirst.InitTables<Models.ProductReviseProcess>();
db.CodeFirst.InitTables<Models.ProductAssistProcess>();
db.CodeFirst.InitTables<Models.Order>();
db.CodeFirst.InitTables<Models.OrderHistory>();
db.CodeFirst.InitTables<Models.SizeTagData>();
db.CodeFirst.InitTables<Models.DefectInfo>();
//更新新列数据
//==触发器
string rootPath = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
string sqlPath = rootPath + "\\SqlPath\\";
if (Directory.Exists(sqlPath))
{
int count;
string triggerName = "trigger_order_update";
if (File.Exists(sqlPath + triggerName+".sql"))
{
count = db.Ado.GetInt($"SELECT count(*) FROM information_schema.TRIGGERS where TRIGGER_NAME = '{triggerName}'");
if (count > 0)
db.Ado.ExecuteCommand($"drop trigger {triggerName}");
db.Ado.ExecuteCommand(File.ReadAllText(sqlPath + triggerName + ".sql"));
}
}
//==清理垃圾数据
db.Ado.ExecuteCommand($"delete from `Order` where ProductId not in (select id from product)");
}
}
/// <summary>
/// 备份DB (还原mysql -uroot -p < d:\dbName.sql)
/// </summary>
/// <param name="result"></param>
public static void BackupDataBase(string outFilePath)
{
using (var conn = new MySqlConnection(ConnectionString))
{
using (var cmd = new MySqlCommand())
{
using (MySqlBackup mb = new MySqlBackup(cmd))
{
// 设置数据库连接
cmd.Connection = conn;
cmd.Connection.Open();
// 导出数据库到文件
mb.ExportToFile(outFilePath);
conn.Close();
}
}
}
}
private static string GetMD5(string str)
{
byte[] data = Encoding.UTF8.GetBytes(str);
data = new System.Security.Cryptography.MD5CryptoServiceProvider().ComputeHash(data);
string ret = "";
for (int i = 0; i < data.Length; i++)
{
ret += data[i].ToString("x1").PadLeft(2, '0');//ToString("x1"):转换为16进制
}
return ret.ToUpper();
}
}
}