dboperate.cs

学生成绩管理系统

using System;
using System.Collections;
using System.Data;
using System.Data.SqlClient;
using TeachHelper.DataAccessLayer;


namespace TeachHelper.Comm
{

    /// <summary>
    /// 数据库的备份和还原
    /// </summary>
    public class DbOperate
    {
        /// <summary>
        /// 服务器
        /// </summary>
        private string server;

        /// <summary>
        /// 登录名
        /// </summary>
        private string uid;

        /// <summary>
        /// 登录密码
        /// </summary>
        private string pwd;

        /// <summary>
        /// 要操作的数据库
        /// </summary>
        private string database;

        /// <summary>
        /// 数据库连接字符串
        /// </summary>
        private string conn;

        /// <summary>
        /// DbOperate类的构造函数
        /// 在这里进行字符串的切割，获取服务器，登录名，密码，数据库
        /// </summary>
        public DbOperate()
        {
            this.conn = DataAccess.ConnectionString;
            server = StringCut(conn, "Data Source=", ";");
            uid = StringCut(conn, "User ID=", ";");
            pwd = StringCut(conn, "Password=", ";");
            database = StringCut(conn, "Initial Catalog=", ";");
        }

        /// <summary>
        /// 切割字符串
        /// </summary>
        /// <param name="str"></param>
        /// <param name="bg"></param>
        /// <param name="ed"></param>
        /// <returns></returns>
        public string StringCut(string str, string bg, string ed)
        {
            string sub;
            sub = str.Substring(str.IndexOf(bg) + bg.Length);
            sub = sub.Substring(0, sub.IndexOf(";"));
            return sub;
        }


        /// <summary>
        /// 构造文件名
        /// </summary>
        /// <param name="dirpath">存放目录路径,如:@"C:\aa\"</param>
        /// <returns>文件名</returns>
        private string CreatePath(string dirpath)
        {
            string CurrTime = System.DateTime.Now.ToString();
            CurrTime = CurrTime.Replace("-", "");
            CurrTime = CurrTime.Replace(":", "");
            CurrTime = CurrTime.Replace(" ", "");
            //CurrTime = CurrTime.Substring(0, 12);
            string path = dirpath;
            path += database;
            path += "_db_";
            path += CurrTime;
            path += ".BAK";
            return path;
        }

        /// <summary>
        /// 数据库备份
        /// </summary>
        /// <param name="dirpath">存放目录路径,如:@"C:\aa\"</param>
        /// <returns>备份是否成功</returns>
        public bool DbBackup(string dirpath)
        {
            string path = CreatePath(dirpath);
            SQLDMO.Backup oBackup = new SQLDMO.BackupClass();
            SQLDMO.SQLServer oSQLServer = new SQLDMO.SQLServerClass();
            try
            {
                oSQLServer.LoginSecure = false;
                oSQLServer.Connect(server, uid, pwd);
                oBackup.Action = SQLDMO.SQLDMO_BACKUP_TYPE.SQLDMOBackup_Database;
                oBackup.Database = database;
                oBackup.Files = path;
                oBackup.BackupSetName = database;
                oBackup.BackupSetDescription = "数据库备份";
                oBackup.Initialize = true;
                oBackup.SQLBackup(oSQLServer);

                return true;
            }
            catch (Exception ex)
            {
                return false;
                throw ex;
            }
            finally
            {
                oSQLServer.DisConnect();
            }
        }

        /// <summary>
        /// 数据库恢复
        /// <param name="path">存放文件路径,如:@"C:\aa\bb.bak"</param>
        /// </summary>
        public string DbRestore(string path)
        {
            if (exepro() != true)//执行存储过程
            {
                return "操作失败";
            }
            else
            {
                SQLDMO.Restore oRestore = new SQLDMO.RestoreClass();
                SQLDMO.SQLServer oSQLServer = new SQLDMO.SQLServerClass();
                try
                {
                    exepro();
                    oSQLServer.LoginSecure = false;
                    oSQLServer.Connect(server, uid, pwd);
                    oRestore.Action = SQLDMO.SQLDMO_RESTORE_TYPE.SQLDMORestore_Database;
                    oRestore.Database = database;
                    ///自行修改
                    oRestore.Files = path;
                    oRestore.FileNumber = 1;
                    oRestore.ReplaceDatabase = true;
                    oRestore.SQLRestore(oSQLServer);

                    return "ok";
                }
                catch (Exception e)
                {
                    return "恢复数据库失败";
                    throw e;
                }
                finally
                {
                    oSQLServer.DisConnect();
                }
            }
        }

        /// <summary>
        /// 杀死当前库的所有进程
        /// </summary>
        /// <returns></returns>
        private bool exepro()
        {

            SqlConnection conn1 = new SqlConnection("server=" + server + ";uid=" + uid + ";pwd=" + pwd + ";database=master");
            SqlCommand cmd = new SqlCommand("killspid", conn1);
            cmd.CommandType = CommandType.StoredProcedure;
            cmd.Parameters.Add("@dbname", "JESSUP");
            try
            {
                conn1.Open();
                cmd.ExecuteNonQuery();
                return true;
            }
            catch (Exception ex)
            {
                return false;
            }
            finally
            {
                conn1.Close();
            }
        }

    }

}