?using System;
using System.Data.SqlClient;
using System.Data;

namespace System.Data.Xiao
{

    /// <summary>

    /// DbOper類,主要應用SQLDMO實現對Microsoft SQL Server數據庫的備份和恢復

    /// </summary>

    public class DbOper
    {

        private string server;

        private string uid;

        private string pwd;

        private string database;

        private string conn;

        /// <summary>

        /// DbOper類的構造函數

        /// </summary>

        public DbOper(string strServer, string strUser, string strPass, string strDataBase)
        {

            server = strServer;

            uid = strUser;

            pwd = strPass;

            database = strDataBase;

        }

        public string cut(string str, string bg, string ed)
        {

            string sub;

            sub = str.Substring(str.IndexOf(bg) + bg.Length);

            if (sub.IndexOf(ed) != -1)
                sub = sub.Substring(0, sub.IndexOf(ed));

            return sub;

        }

        private string CreatePath(string createpath)
        {
            string CurrTime = System.DateTime.Now.ToString("yyyyMMddHHmmss");
            CurrTime = CurrTime.Substring(0, 12);
            string spath = createpath;
            spath += database;
            spath += "_db_";
            spath += CurrTime;
            spath += ".bak";
            return spath;
        }

        /// <summary>

        /// 數據庫備份

        /// </summary>

        public bool DbBackup(string url)
        {
            string path = CreatePath(url);

            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 e)
            {
                return false;
                throw e;
            }

            finally
            {

                oSQLServer.DisConnect();

            }

        }



        /// <summary>

        /// 數據庫恢復

        /// </summary>

        public string DbRestore(string url)
        {

            if (exepro() != true)//執行存儲過程
            {

                return "操作失敗";

            }

            else
            {

                SQLDMO.Restore oRestore = new SQLDMO.RestoreClass();

                SQLDMO.SQLServer oSQLServer = new SQLDMO.SQLServerClass();

                try
                {

                    oSQLServer.LoginSecure = false;

                    oSQLServer.Connect(server, uid, pwd);

                    oRestore.Action = SQLDMO.SQLDMO_RESTORE_TYPE.SQLDMORestore_Database;

                    oRestore.Database = database;

                    oRestore.Files = url;//@"d:\Northwind.bak";

                    oRestore.FileNumber = 1;

                    oRestore.ReplaceDatabase = true;

                    oRestore.SQLRestore(oSQLServer);

                    return "成功恢復數據庫。";

                }

                catch (Exception e)
                {

                    return "恢復數據庫失敗";

                    throw e;

                }

                finally
                {

                    oSQLServer.DisConnect();

                }

            }

        }

        public 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.AddWithValue("@dbname", database);

            try
            {

                conn1.Open();

                cmd.ExecuteNonQuery();

                return true;

            }

            catch (Exception ex)
            {

                return false;

            }
            finally
            {
                conn1.Close();
            }
        }
    }
}