MySQL代码备份

Posted 2021-02-10 moonsoft

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("
");
            }   
            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");
    }

}

servlet

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>");        
                
            }
            
        }
        
    }

}