/// <summary>
/// 通过指定的文件名，为它创建目录
///
/// 错误处理：IEFS_Error.LastErrorMessage
///
/// </summary>
/// <param name="aFileName"></param>
/// <returns></returns>
public static I3MsgInfo CreateDirectoryByFileName(string aFileName)
{
if (string.IsNullOrEmpty(aFileName))
{
return(new I3MsgInfo(false, "未指定文件名"));
}
string dir = Path.GetDirectoryName(aFileName);
return(I3DirectoryUtil.CreateDirctory(dir));
}
/// <summary>
/// 检查并清空目录，最后有空的目录存在则返回true
/// </summary>
/// <param name="aDir"></param>
/// <returns></returns>
public static I3MsgInfo CheckAndClearDirctory(string aDir)
{
I3MsgInfo msg = I3DirectoryUtil.DeleteDirctory(aDir);
if (!msg.State)
{
return(msg);
}
return(I3DirectoryUtil.CreateDirctory(aDir));
}
/// <summary>
/// 从数据库字段中读取文件并保存到目的文件，
/// 注意，此字段必须是由WriteFileToDataRow写入！
/// </summary>
/// <param name="row"></param>
/// <param name="fieldName"></param>
/// <param name="destFileName"></param>
/// <returns></returns>
public static I3MsgInfo ReadFileFromDataRowBeforeRar(DataRow row, string fieldName, string destFileName)
{
#region 初始化临时文件
string tmpDir = I3DirectoryUtil.GetAppTmpDir();
string tmpRarFileName = Path.Combine(tmpDir, I3DateTimeUtil.ConvertDateTimeToLongString(DateTime.Now) + ".rar");
I3MsgInfo msg = I3DirectoryUtil.CreateDirctory(tmpDir);
if (!msg.State)
{
return(msg);
}
if (!I3FileUtil.CheckFileNotExists(tmpRarFileName))
{
return(new I3MsgInfo(false, ""));
}
#endregion
try
{
#region 从数据库读取文件到临时文件
try
{
byte[] buffer = (byte[])row[fieldName];
if (buffer.Length == 0)
{
return(new I3MsgInfo(false, "数据字段长度为0!"));
}
using (FileStream stream = new FileStream(tmpRarFileName, FileMode.Create, FileAccess.ReadWrite, FileShare.None))
{
stream.Write(buffer, 0, buffer.Length);
stream.Flush();
stream.Close();
}
}
catch (Exception ex)
{
return(new I3MsgInfo(false, ex.Message, ex));
}
#endregion
#region 解压
using (I3Rar rar = new I3Rar())
{
msg = rar.UnCompressASingleFile(tmpRarFileName, "IEFS_Rar.tmp", destFileName);
if (!msg.State)
{
return(msg);
}
}
#endregion
return(I3MsgInfo.Default);
}
finally
{
#region 除临时Rar文件
I3FileUtil.CheckFileNotExists(tmpRarFileName);
#endregion
}
}
/// <summary>
/// 将文件经过rar压缩后，写入数据库字段中
/// 写入数据库中的实际是一个rar压缩文件，里面只有一个文件，文件名为 IEFS_Rar.tmp
/// </summary>
/// <param name="row"></param>
/// <param name="fieldName"></param>
/// <param name="fileName"></param>
/// <returns></returns>
public static I3MsgInfo WriteFileToDataRowAfterRar(DataRow row, string fieldName, string sourceFileName, bool deleteSource)
{
if (!File.Exists(sourceFileName))
{
return(new I3MsgInfo(false, "文件不存在！文件名：" + sourceFileName));
}
string tmpDir = I3DirectoryUtil.GetAppTmpDir();
string tmpRarFileName = Path.Combine(tmpDir, I3DateTimeUtil.ConvertDateTimeToLongString(DateTime.Now) + ".rar");
string tmpSourceFileName = Path.Combine(tmpDir, I3DateTimeUtil.ConvertDateTimeToLongString(DateTime.Now));
tmpSourceFileName = Path.Combine(tmpSourceFileName, "IEFS_Rar.tmp");
#region 生成临时目录并删除临时文件
if (!I3FileUtil.CheckFileNotExists(tmpRarFileName))
{
return(new I3MsgInfo(false, "临时文件删除失败"));
}
I3MsgInfo msg = I3DirectoryUtil.CreateDirctory(tmpDir);
if (!msg.State)
{
return(msg);
}
if (!I3FileUtil.CheckFileNotExists(tmpSourceFileName))
{
return(new I3MsgInfo(false, "临时源文件删除失败"));
}
msg = I3DirectoryUtil.CreateDirectoryByFileName(tmpSourceFileName);
if (!msg.State)
{
return(msg);
}
msg = I3FileUtil.MoveFile(sourceFileName, tmpSourceFileName, deleteSource);
if (!msg.State)
{
return(msg);
}
#endregion
try
{
#region 压缩
using (I3Rar rar = new I3Rar())
{
msg = rar.CompressASingleFile(tmpSourceFileName, tmpRarFileName);
if (!msg.State)
{
return(msg);
}
}
if (!File.Exists(tmpRarFileName))
{
return(new I3MsgInfo(false, "未知原因，压缩失败！文件名：" + tmpRarFileName));
}
#endregion
#region 写入
using (FileStream stream = File.OpenRead(tmpRarFileName))
{
byte[] buffer = new byte[stream.Length];
stream.Position = 0;
int readSize = stream.Read(buffer, 0, buffer.Length);
if (readSize != buffer.Length)
{
return(new I3MsgInfo(false, "数据缓冲读取失败！目标长度：" + buffer.Length.ToString() + "，读取长度：" + readSize.ToString()));
}
try
{
row.BeginEdit();
row[fieldName] = buffer;
row.EndEdit();
stream.Close();
return(I3MsgInfo.Default);
}
catch (Exception ex)
{
return(new I3MsgInfo(false, ex.Message, ex));
}
}
#endregion
}
finally
{
#region 除临时文件
I3FileUtil.CheckFileNotExists(tmpRarFileName);
I3DirectoryUtil.DeleteDirctory(Path.GetDirectoryName(tmpSourceFileName));
#endregion
}
}
/// <summary>
/// 根据UDL文件中的连接信息和备份文件名，创建或者恢复数据库
/// runMode:创建，恢复
/// </summary>
/// <param name="udlFileName"></param>
/// <returns></returns>
public static bool RestoreSQLServerDataBase(string connectionString, string bakFileName, I3SQLServerRestoreMode restoreMode, I3SQLServerRestoreInfo restoreInfo, string dataFilePath)
{
#region 操作关键字
string runMode = "";
if (restoreMode == I3SQLServerRestoreMode.rmCreate)
{
runMode = "创建";
}
else
{
runMode = "恢复";
}
#endregion
#region 从udl文件获取连接字符串
//restoreInfo("一、连接字符串：" + connectionString, false, MessageBoxIcon.Warning);
restoreInfo("一、获取连接字符串", false, MessageBoxIcon.Warning);
//MessageBox.Show(connectionString);
#endregion
#region 获取服务器名，登录方式，数据库名等信息
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("二、解析服务器名，数据库名", false, MessageBoxIcon.Warning);
//OleDbConnectionStringBuilder ob;
SqlConnectionStringBuilder sb;
try
{
sb = new SqlConnectionStringBuilder(connectionString);
}
catch (Exception ex)
{
restoreInfo("数据库连接字符串解析失败，错误消息：\r\n" + ex.Message, true, MessageBoxIcon.Error);
return false;
}
string serverName = sb.DataSource.ToUpper();
string dataBasename = sb["Initial Catalog"].ToString();
restoreInfo("即将在服务器" + serverName + "上" + runMode + "数据库" + dataBasename, false, MessageBoxIcon.Warning);
#endregion
I3Data masterCon = null;
try
{
#region 登录到master数据库，并检查数据库是否已经存在
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("三、连接数据库引擎", false, MessageBoxIcon.Warning);
sb["Initial Catalog"] = "master";
masterCon = I3Data.CreateDataSql(sb.ConnectionString);
if (!masterCon.Active)
{
restoreInfo("无法启动数据库引擎，错误消息：\r\n" + masterCon.LastErrorInfo, true, MessageBoxIcon.Error);
return false;
#region 检查是否本机，如果不是本机，则直接返回false
//if (!I3PCUtil.CheckServerNameIsLocal(serverName))
//{
// restoreInfo("连接数据库引擎失败，但连接字符串中指定的服务器" + serverName + "不是本机，因此无法启动数据库引擎！", true, MessageBoxIcon.Error);
// return false;
//}
#endregion
#region 启动数据库引擎 并重新连接
//if (MessageBox.Show("数据库引擎连接失败，需要启动数据库引擎吗？", "连接数据库引擎", MessageBoxButtons.YesNo, MessageBoxIcon.Warning) == DialogResult.Yes)
//{
// //IEFS_PC.RunService("MSSQLServer");
// //为了有提示界面，通过调用SCM.exe来启动服务
// I3MsgInfo msg = I3PCUtil.CreateAndWaitProcess("SCM.exe", " -action 1 -service MSSQLServer -SvcStartType 2 ", false);
// if (msg.State)
// {
// Thread.Sleep(5000);
// masterCon = I3Data.CreateDataOle(sb.ConnectionString);
// if (!masterCon.Active)
// {
// restoreInfo("启动了数据库引擎，但仍然连接失败！请联系技术人员进行检查。", true, MessageBoxIcon.Error);
// return false;
// }
// }
// else
// {
// restoreInfo("无法启动数据库引擎，错误消息：\r\n" + msg.Message, true, MessageBoxIcon.Error);
// return false;
// }
//}
//else
//{
// restoreInfo("数据库引擎连接失败，无法创建默认数据库", true, MessageBoxIcon.Error);
// return false;
//}
#endregion
}
restoreInfo("数据库引擎连接OK", false, MessageBoxIcon.Warning);
#endregion
#region 获取系统目录信息
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("四、获取目录信息", false, MessageBoxIcon.Warning);
string dataRootPath = "";
if (!string.IsNullOrEmpty(dataFilePath))
{
dataRootPath = dataFilePath;
I3DirectoryUtil.CreateDirctory(dataRootPath);
}
else
{
using (DataTable tmp = new DataTable("sysdatabases"))
{
string sqlStr = "select * from sysdatabases where name = 'master'";
I3MsgInfo msg = masterCon.FillTable(tmp, true, sqlStr, null, null);
if (!msg.State)
{
restoreInfo("取不到系统目录，" + runMode + "操作中止！错误消息：\r\n" + msg.Message, true, MessageBoxIcon.Error);
return false;
}
try
{
if (tmp.Rows.Count == 0)
{
restoreInfo("取不到系统目录，" + runMode + "操作中止！", true, MessageBoxIcon.Error);
return false;
}
dataRootPath = Path.GetDirectoryName(tmp.Rows[0]["filename"].ToString());
}
finally
{
masterCon.DisposeDataTable(tmp);
}
restoreInfo("系统目录获取成功：" + dataRootPath, false, MessageBoxIcon.Warning);
}
}
#endregion
#region 校验备份文件
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("五、校验数据库备份文件", false, MessageBoxIcon.Warning);
string moveStr = "";
if (!File.Exists(bakFileName))
{
restoreInfo("数据库备份文件不存在！\r\n文件路径：\r\n" + bakFileName, true, MessageBoxIcon.Error);
return false;
}
List<string> toDeleteFile = new List<string>();
using (DataTable tmp = new DataTable("FILELISTONLY"))
{
string sqlStr = " RESTORE FILELISTONLY from disk=" + I3StringUtil.QuotedStr(bakFileName);
I3MsgInfo msg = masterCon.FillTable(tmp, true, sqlStr, null, null);
if (!msg.State)
{
restoreInfo("该备份文件不是数据库备份文件!\r\n文件路径：\r\n" + bakFileName + "\r\n错误消息：\r\n" + msg.Message, true, MessageBoxIcon.Error);
return false;
}
try
{
foreach (DataRow row in tmp.Rows)
{
if (row["Type"].ToString() == "D")
{
string fileName = Path.Combine(dataRootPath, dataBasename.Trim() + "_Dat.MDF");
toDeleteFile.Add(fileName);
moveStr = moveStr + " Move " + I3DBUtil.QuotedStr(row["logicalname"].ToString())
+ " To " + I3DBUtil.QuotedStr(fileName) + ",";
}
else
{
string fileName = Path.Combine(dataRootPath, dataBasename.Trim() + "_Log.LDF");
toDeleteFile.Add(fileName);
moveStr = moveStr + " Move " + I3DBUtil.QuotedStr(row["logicalname"].ToString())
+ " To " + I3DBUtil.QuotedStr(fileName) + ",";
}
}
moveStr = I3StringUtil.SubString(moveStr, 0, moveStr.Length - 1);
//MessageBox.Show(moveStr);
}
finally
{
masterCon.DisposeDataTable(tmp);
}
}
restoreInfo("数据库备份文件校验成功", false, MessageBoxIcon.Warning);
#endregion
#region 删除现有数据库
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("六、检查数据库" + dataBasename + "是否已经存在", false, MessageBoxIcon.Warning);
bool has;
#region 获取数据库是否已经存在
using (DataTable tmp = new DataTable("sysdatabases"))
{
string sqlStr = " select * from sysdatabases where name = " + I3DBUtil.QuotedStr(dataBasename);
I3MsgInfo msg = masterCon.FillTable(tmp, true, sqlStr, null, null);
if (!msg.State)
{
restoreInfo("出现错误，操作中止！错误消息：\r\n" + msg.Message, true, MessageBoxIcon.Error);
return false;
}
try
{
has = tmp.Rows.Count > 0;
}
finally
{
masterCon.DisposeDataTable(tmp);
}
}
#endregion
if (has)
{
if (restoreMode == I3SQLServerRestoreMode.rmCreate)
{
restoreInfo("数据库" + dataBasename + "已经存在，不能创建数据库！", true, MessageBoxIcon.Error);
return false;
}
#region 删除数据库
if (MessageBox.Show("数据库" + dataBasename + "已经存在，是否删除？\r\n警告：进行删除操作前，请确认之前已经备份数据库，否则所有数据将丢失！", "删除数据库", MessageBoxButtons.YesNo, MessageBoxIcon.Warning) == DialogResult.Yes)
{
restoreInfo("开始删除数据库" + dataBasename, false, MessageBoxIcon.Warning);
string sqlStr = " DROP DATABASE " + dataBasename;
I3MsgInfo msg = masterCon.Execute(sqlStr, null);
if (msg.State)
{
restoreInfo("数据库" + dataBasename + "删除成功", false, MessageBoxIcon.Warning);
}
else
{
restoreInfo("数据库" + dataBasename + "删除失败，错误消息：" + msg.Message, true, MessageBoxIcon.Error);
return false;
}
}
else
{
restoreInfo("数据库" + dataBasename + "已经存在，用户放弃删除，不能" + runMode + "数据库！", true, MessageBoxIcon.Error);
return false;
}
#endregion
}
#region 再次删除数据库文件 注：确保清空文件
foreach (string s in toDeleteFile)
{
I3FileUtil.CheckFileNotExists(s);
}
#endregion
restoreInfo("现有数据库删除成功，可以进行下一步操作", false, MessageBoxIcon.Warning);
#endregion
#region 恢复
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("七、开始" + runMode + "数据库，请耐心等待", false, MessageBoxIcon.Warning);
string restoreStr = " RESTORE DATABASE " + dataBasename.Trim()
+ " From Disk = " + I3DBUtil.QuotedStr(bakFileName)
+ " with " + moveStr;
//MessageBox.Show(restoreStr);
SqlCommand com = new SqlCommand(restoreStr, (SqlConnection)masterCon.GetConnection());
try
{
com.CommandTimeout = 600;
com.ExecuteNonQuery();
}
catch (Exception ex)
{
restoreInfo("数据库" + dataBasename + runMode + "不成功！错误消息：\r\n" + ex.Message, true, MessageBoxIcon.Error);
return false;
}
finally
{
com.Dispose();
}
restoreInfo("数据库" + runMode + "成功！", true, MessageBoxIcon.Warning);
restoreInfo("", false, MessageBoxIcon.Warning);
restoreInfo("", false, MessageBoxIcon.Warning);
#endregion
return true;
}
finally
{
if (masterCon != null && masterCon.Active)
{
masterCon.Close();
}
}
}