Java 调用bat执行的备份Oracle数据库 类


package com.buckupDB;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.swing.filechooser.FileSystemView;
import org.apache.tomcat.dbcp.dbcp.BasicDataSource;
import com.pm360.mda.platform.db.DBConn;
import com.pm360.mda.platform.util.DBUtil;
import com.pm360.mda.platform.util.Logger;
import com.pm360.mda.platform.util.StringUtil;


/**
 *
 * 数据备份  impdp / expdp
 * @author ZQD32
 *
 */
public class BuckupdpDB {

 /**
  * //获取桌面路径
  * @return
  */
  private static String getDesktopPath()
  {
   FileSystemView fsv = FileSystemView.getFileSystemView();
   File file= fsv.getHomeDirectory();
       return file.getPath();
  }
 
 
  /**
   * 获取时间字符
   * @return
   */
  private static String getDateTime()
  {
   Date date=new Date();
   String str=date.toLocaleString();
   str=str.replaceAll(":", "-");
   str=str.replaceAll(" ", "-");
      return str;
  }
 
  /**开启指定进程
   * @param map
   */
  private static void runingProcess(Map<String, Object> map)
  {
   String strPath = StringUtil.formatDbColumn(map.get("Paths"));
   strPath = strPath.replace(" ", "\" \"");
   try
   {
    StringBuilder sb=new StringBuilder("cmd /c start "+strPath+"\\temp.bat");
    Process p = Runtime.getRuntime().exec(sb.toString());
  }
   catch (IOException e)
   {
   e.printStackTrace();
  }
  }
 
  /**
   * 执行备份文件
   * @return
   * @throws IOException
   */
  public static String expingProcess(Map<String, Object> map)
  {  
   if(productionExpBat(map).equalsIgnoreCase("good"))
   {
    runingProcess(map);
   }
   else
   {
    return "false";
   }
    return "true";
  }
 
  /**
   * 执行导入文件
   * @return
   * @throws IOException
   */
  public static String impingProcess(Map<String, Object> map)
  {
   if(productionImpBat(map).equalsIgnoreCase("good"))
   {
    runingProcess(map);
   }
   else
   {
    return "false";
   }
    return "true";
  }
 
  /**
   * 获取 连接信息
   * @param sid
   * @param user
   */
  private static Map<String, String> getConnInfo()
  {
   Map<String,String> map =new HashMap<String, String>();
  
   try {
    BasicDataSource init = (BasicDataSource)new InitialContext().lookup("java:comp/env/jdbc/**");
    String url =init.getUrl();
    //获取Oracle数据库实例名
    map.put("sid", url.substring(url.lastIndexOf(":")+1));
    //获取oracle数据库登录名
    map.put("user",init.getUsername());
    //获取oracle数据库登密码
    map.put("pwd", init.getPassword());
   } catch (NamingException e) {
    e.printStackTrace();
   }
   return map;
  }
 
 
 /**
  *
  * 生成 导出配置Bat文件
  */
  private static String productionExpBat(Map<String, Object> map)
  {
      //得到数据库登录的信息
   Map<String,String> ConnInfoMap =getConnInfo();
   //生成数据泵配置文件
   productionExpdpPar(map);
   //设置一个标志
   String flag="";
   //创建个缓存 数据集
   List<String> tmpBat=new ArrayList<String>();
   //获取模板 文件
   File Afile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\processBat\\dp\\expdpDB.bat");
   //生成空白的bat临时文件
   File Bfile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\temp.bat");
   //申明读取缓冲器
   BufferedReader br=null;
   try {
    //设置读取缓冲器 文件指向
    br = new BufferedReader(new InputStreamReader(new FileInputStream(Afile)));
   } catch (FileNotFoundException e1) {
    e1.printStackTrace();
   }
   //获取   rar 备份 文件的名字
   String filename=StringUtil.formatDbColumn(map.get("FILENAME"));
   if(filename==null||filename.equals("")){
    //默认为 DBProjBuckup
    filename="DBProjBuckup";
   }
   filename=filename+"_"+getDateTime();

   try {
    try {
     while(br.ready())//判断是否还有可读信息
     {
      //读取一行数据
      String str=br.readLine().toString();
      //设置数据库的实例
      if(str.startsWith("set sid")){
       str="set sid="+ConnInfoMap.get("sid");
      }
      //设置数据登录名
      else if(str.startsWith("set user")){
       str="set user="+ConnInfoMap.get("user");
      }
      //设置数据登密码
      else if(str.startsWith("set pwd")){
       str="set pwd="+ConnInfoMap.get("pwd");;
      }
      //设置备份项目id过滤
      else if(str.startsWith("set proj_id")){
       str="set proj_id="+StringUtil.formatDbColumn(map.get("PROJ_ID"));
      }
      //设置备份文件名
      else if(str.startsWith("set filename")){
       str="set filename="+filename;
      }
      //桌面路径
      else if(str.startsWith("set Desktop")){
       str="set Desktop="+getDesktopPath();
      }
      //dmp文件存放路径**************该路径需要查取的
      else if(str.startsWith("set savePth")){
       str="set savePth="+getDumpPath();
      }
      tmpBat.add(str);
     }
     //关闭缓冲器
     br.close();
    } catch (IOException e) {
     try {
      br.close();
     } catch (IOException e1) {
      e1.printStackTrace();
     }
     e.printStackTrace();
    }
   
    //设定写数据缓冲器   文件指向
    PrintWriter pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(Bfile)),true);
    for(String str : tmpBat)
    {
     //按行写入
     pw.println(str);  
    }
    //pw.println("del /q BuckupParfile.par"); 
    //pw.println("del /q temp.bat"); 
   
    //关闭缓冲器
    pw.close(); 
   
    flag="good";
   } catch (FileNotFoundException e) {
    flag="error";
    e.printStackTrace();
   }  
   return flag;
  }
 
  /**
   * 获取数据泵数据存储路径
   * @return
   */
  private static String getDumpPath()
  {
  //select OS_PATH from sys.dir$ where  OS_PATH like '%\dpdump\'
   String dumpPath=null;
   Connection conn=null;
   try {
   conn=DBConn.getConnection("***");//该方法自己写
   dumpPath = (String)DBUtil.getResultFieldValue(conn, "select OS_PATH from sys.dir$ where  OS_PATH like '%\\dpdump\\'");
   } catch (SQLException e) {
   Logger.error(e);
  }finally{
   try
   {
    conn.close();
   } catch (SQLException e)
   {
    Logger.error(e);
   }
  }
   return dumpPath;
  }
 
  /**
   * 配置数据泵的参数
   * @param map
   * @return
   */
  private static String productionExpdpPar(Map<String, Object> map)
  {
      String  proj_id = StringUtil.formatDbColumn(map.get("PROJ_ID"));
   //获取模板 文件
   File Afile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\processBat\\dp\\BuckupParfile.par");
   //生成空白的bat临时文件
   File Bfile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\BuckupParfile.par");
   try {
    //申明读取缓冲器  //设置读取缓冲器 文件指向
    BufferedReader br = new BufferedReader(new InputStreamReader(new FileInputStream(Afile)));

    //设定写数据缓冲器   文件指向
    PrintWriter pw= new PrintWriter(new OutputStreamWriter(new FileOutputStream(Bfile)),true);
    
    try {
     while(br.ready())//判断是否还有可读信息
     {
      //读取一行数据
      String str=br.readLine().toString();
      str=str.replace("%proj_id%", proj_id);
      pw.println(str);  
     }
     br.close();
     pw.close();
    } catch (IOException e) {
     e.printStackTrace();
    }
   } catch (FileNotFoundException e) {
   e.printStackTrace();
   }
   return "";
  }
 
  /**
   * 生成导入配置bat文件
   * @param map
   * @return
   */
  private static String productionImpBat(Map<String, Object> map)
  {
      //获取   rar 备份 文件的名字
   String FilePath=StringUtil.formatDbColumn(map.get("FilePath"));
   String FileName = FilePath.substring(FilePath.lastIndexOf('\\')+1);
      
      //生成 导入前清理 sql 文件 
      productionClearSql(map);
     
      Map<String,String> ConnInfoMap =getConnInfo();
  
  
  
   //设置一个标志
   String flag="";
   //创建个缓存 数据集
   List<String> tmpBat=new ArrayList<String>();
   //获取模板 文件
   File Afile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\processBat\\dp\\impdpDB.bat");
   //生成空白的bat临时文件
   File Bfile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\temp.bat");
   //申明读取缓冲器
   BufferedReader br=null;
   try {
    //设置读取缓冲器 文件指向
    br = new BufferedReader(new InputStreamReader(new FileInputStream(Afile)));
   } catch (FileNotFoundException e1) {
    e1.printStackTrace();
   }
  
   try {
    try {
     while(br.ready())//判断是否还有可读信息
     {
      //读取一行数据
      String str=br.readLine().toString();
      //设置数据库的实例
      if(str.startsWith("set sid")){
       str="set sid="+ConnInfoMap.get("sid");
      }
      //设置数据登录名
      else if(str.startsWith("set user")){
       str="set user="+ConnInfoMap.get("user");
      }
      //设置数据登密码
      else if(str.startsWith("set pwd")){
       str="set pwd="+ConnInfoMap.get("pwd");
      }
      //设置备份文件 路径
      else if(str.startsWith("set FilePath")){
       str="set FilePath="+FilePath;
      }
      //dump路径
      else if(str.startsWith("set Dumpfile")){
       str="set Dumpfile="+getDumpPath();
      }
      //文件名
      else if(str.startsWith("set FileName")){
       str="set FileName="+FileName;
      }
      tmpBat.add(str);
     }
     //关闭缓冲器
     br.close();
    } catch (IOException e) {
     try {
      br.close();
     } catch (IOException e1) {
      e1.printStackTrace();
     }
     e.printStackTrace();
    }
   
    //设定写数据缓冲器   文件指向
    PrintWriter pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(Bfile)),true);
    for(String str : tmpBat)
    {
     //按行写入
     pw.println(str);  
    }
    pw.println("ECHO watting …………………………");
    pw.println("sqlplus %user%/%pwd%@%sid%  @UpdateParentId.sql ");
    pw.println("del /q UpdateParentId.sql");
    pw.println("del /q impClearing.sql");
    pw.println("del /q temp.bat");
    //关闭缓冲器
    pw.close(); 
    flag="good";
   } catch (FileNotFoundException e) {
    flag="error";
    e.printStackTrace();
   }  
  
   return flag;
  }

  /**
   * 生成导入前sql清理文件
   * @param map
   * @return
   */
  private static String productionClearSql(Map<String, Object> map)
  {
    //获取项目id
    //StringUtil.formatDbColumn(map.get("PROJ_ID"))  这个是即将导入  项目的   新的父id
      String proj_id =getProjId(map);
   //创建个缓存 数据集
   List<String> tmpBat=new ArrayList<String>();
   //获取模板 文件
   File Afile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\processBat\\dp\\impClearing.sql");
   //生成空白的bat临时文件
   File Bfile=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\impClearing.sql");
   //申明读取缓冲器
   BufferedReader br=null;
   try {
    //设置读取缓冲器 文件指向
    br = new BufferedReader(new InputStreamReader(new FileInputStream(Afile)));
   } catch (FileNotFoundException e1) {
    e1.printStackTrace();
   }
   try {
    try {
     while(br.ready())//判断是否还有可读信息
     {
      //读取一行数据
      String str=br.readLine().toString();
      //设置备份文件名  %proj_id%
      str = str.replace("%proj_id%", proj_id);
      tmpBat.add(str);
     }
     //关闭缓冲器
     br.close();
    } catch (IOException e) {
     try {
      br.close();
     } catch (IOException e1) {
      e1.printStackTrace();
     }
     e.printStackTrace();
    }
   
    //设定写数据缓冲器   文件指向
    PrintWriter pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(Bfile)),true);
    for(String str : tmpBat)
    {
     //按行写入
     pw.println(str);  
    }
    pw.println("exit");  
    //关闭缓冲器
    pw.close(); 
   } catch (FileNotFoundException e) {
    e.printStackTrace();
   }  
  
   return "";
  }

 
  /**
   * 生成导入   项目的id  和   更新语句
   * @param map
   * @return
   */
  private static String getProjId(Map<String, Object> map)
  {
   String FilePath=StringUtil.formatDbColumn(map.get("FilePath"));
   FilePath = FilePath.substring(FilePath.lastIndexOf("-")+1, FilePath.lastIndexOf('.'));
  
   File UpParentId=new File(StringUtil.formatDbColumn(map.get("Paths"))+"\\UpdateParentId.sql");
   PrintWriter pw;
  try {
   pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(UpParentId)),true);
   pw.println("update cm_proj t  set t.parent_proj_id='"+StringUtil.formatDbColumn(map.get("PROJ_ID"))+"',t.parent_path=('"
     +getProjPath(StringUtil.formatDbColumn(map.get("PROJ_ID")))+"'||';'||'"+FilePath+"') where t.proj_id='"+FilePath+"';"); 
   pw.println("commit;");
   pw.println("exit");
   //关闭缓冲器
   pw.close();
  } catch (FileNotFoundException e) {
   e.printStackTrace();
  }
   return FilePath;
  }
 
  /**
   * 获取父级路径
   * @param map
   * @return
   */
  private static String getProjPath(String proj_id)
  {
   String projPath=null;
   Connection conn=null;
   try {
   conn=DBConn.getConnection("***");//该方法自己写
   projPath = (String)DBUtil.getResultFieldValue(conn, "select  parent_path  from cm_proj  where proj_id ='"+proj_id+"'");
   conn.close();
  } catch (SQLException e) {
   Logger.error(e);
  }finally{
   try
   {
    conn.close();
   } catch (SQLException e)
   {
    Logger.error(e);
   }
  }
   return projPath;
  }

}

相关内容