package com.dus.utils;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.io.OutputStream;

import java.io.OutputStreamWriter;

//数据库备份

public class MySQLDump {

    public static boolean sqlDump(String cmd,String filePath){

        boolean falg = false;

        try {

            Runtime run = Runtime.getRuntime();

            //cmd 命令："C:/Program Files/MySQL/MySQL Server 5.1/bin/mysqldump -uroot -proot  email"

            Process p = run.exec(cmd);

            InputStream is =     p.getInputStream();// 控制台的输出信息作为输入流

            InputStreamReader isr = new InputStreamReader(is,"UTF-8");//设置输入流编码格式

            BufferedReader br = new BufferedReader(isr);

            //将控制台输入信息写入到文件输出流中

            FileOutputStream fos = new FileOutputStream(filePath);

            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(fos,"UTF-8"));

            String temp = null;

            while( (temp = br.readLine()) !=null){

                bw.write(temp);

                bw.newLine();

            }

            bw.flush();

            bw.close();

            br.close();

            falg = true;

            System.out.println("/* Dump  SQL File "+filePath+" OK! */");

        } catch (IOException e) {

            throw new RuntimeException("请将mysql命令添加到path中！",e);

        }

        return falg;

    }

    //恢复数据库

    /**

     * 导入

     *

     */

    public static void sqlLoad(String cmd,String sqlPath) {

        try {

            Runtime rt = Runtime.getRuntime();   

            // 调用 mysql 的 cmd: C:/Program Files/MySQL/MySQL Server 5.1/bin/mysql.exe -uroot -proot email

            Process child = rt.exec(cmd);

            OutputStream out = child.getOutputStream();//控制台的输入信息作为输出流

            //输入流

            BufferedReader br = new BufferedReader(new InputStreamReader(

                    new FileInputStream(sqlPath), "utf8"));

            //输出流

            OutputStreamWriter writer = new OutputStreamWriter(out, "utf8");

            String inStr;

            while ((inStr = br.readLine()) != null) {

                writer.write(inStr);

                writer.write("\r\n");

            }

            writer.flush();

            // 别忘记关闭输入输出流

            out.close();

            br.close();

            writer.close();

            System.out.println("/* Load  SQL File "+sqlPath+" OK! */");

        } catch (Exception e) {

            e.printStackTrace();

        }   

    }

    public static void main(String[] args) {

        sqlDump("D:/Program Files/MySQL/MySQL Server 5.0/bin/mysqldump -uroot -proot  weddingdb","e:/DBBack/weddingdb.sql");

        //sqlLoad("C:/Program Files/MySQL/MySQL Server 5.1/bin/mysql.exe -uroot -proot email","c:/email.sql");

    }

}

package com.dus.web;

import java.io.IOException;

import java.io.PrintWriter;

import java.util.Date;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

import com.dus.model.Log;

import com.dus.service.LogService;

import com.dus.utils.AppException;

import com.dus.utils.MySQLDump;

/**

 * 备份数据库 servlet

 */

public class MySQLDumpServlet extends HttpServlet {

    @Override

    protected void doGet(HttpServletRequest request, HttpServletResponse response)

            throws ServletException, IOException {

        doPost(request, response);

    }

    @Override

    protected void doPost(HttpServletRequest request, HttpServletResponse response)

            throws ServletException, IOException {

        response.setContentType("text/html;charset=UTF-8");

        request.setCharacterEncoding("UTF-8");

        response.setCharacterEncoding("UTF-8");        

        PrintWriter pw = response.getWriter();

        boolean flag = false;

        // 声明session

        HttpSession session = null;

        // 使用request对象取得session

        session = request.getSession();

        Integer userId = (Integer)session.getAttribute("userId");

        // 若用户没有登录，则跳转到登录页面

        if (userId == null) {

            response.sendRedirect("toLogin");

        } else {

            flag = MySQLDump.sqlDump("C:/Program Files/MySQL/MySQL Server 5.5/bin/mysqldump -uroot -proot dusdb", "e:/DBBack/dusdb.sql");

            if(flag) {            

                pw.print("<script type='text/javascript'> alert('备份数据库成功!  SQL脚本路径：e:/DBBack/dusdb.sql'); </script>");            

                // 添加操作日志

                Log log = new Log();

                log.setUser_id(userId);

                log.setContent("数据库备份");

                log.setTime(new Date());

                LogService logService = new LogService();

                try {

                    logService.save(log);

                } catch (AppException e) {

                    e.printStackTrace();

                }

            } else {

                pw.print("<script type='text/javascript'> alert('备份数据库失败!'); </script>");        

            }

        }

    }

}