Java代碼調用Shell指令碼並傳入參數實現DB2資料庫表匯出到檔案

標籤：java   db2   shell   匯出   export   

本文通過Java代碼調用Shell指令碼並傳入參數實現DB2資料庫表匯出到檔案，代碼如下：

import java.io.File;import java.io.IOException;import java.io.InputStreamReader;import java.io.LineNumberReader;import java.util.HashMap;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import com.*.dmp.bean.AgentConfigInfo;import com.*.dmp.bean.MapKeys;import com.*.dmp.bean.RunStatus;import com.*.dmp.common.SpringUtils;public class ExportDataServiceDB2 {AgentConfigInfo agentConfigInfo = SpringUtils.getContext().getBean(AgentConfigInfo.class);private Logger LOG = LoggerFactory.getLogger(ExportDataServiceDB2.class);private StringBuffer resultMsg = new StringBuffer();String isOK = "0";private String exportShell = agentConfigInfo.getEXPORT_SHELL();//private String exportCMD = agentConfigInfo.getEXPORT_CMD();private StringBuffer exportFilePath = agentConfigInfo.getEXPORT_FILE_PATH();/** * @Title: ExportData  * @Description:  調用Shell指令碼實現db2資料的匯出 * @param dataMap * @throws IOException 對方法的參數進行描述 * @return HashMap<String,String> 傳回型別 */public HashMap<String, String> ExportData(HashMap<String, String> dataMap) throws IOException {String dbSchema = dataMap.get("db_schema");String dbUser = dataMap.get("db_user");String dbPassword = dataMap.get("db_password");String tableName = dataMap.get("table_name");String interFile = dataMap.get("inter_file");String delimiter = dataMap.get("delimiter");String exportLimit = dataMap.get("export_limit");String filePath = mkDirectory(exportFilePath, interFile);dataMap.put("file_abs_path", filePath);String cmdPara = createExportShellParams(dbSchema, dbUser,dbPassword, tableName, filePath, delimiter, exportLimit);LOG.info("Export Parameters: " + cmdPara);resultMsg.append("Export Parameters: " + cmdPara + "\n");String cmd = exportShell + " " + cmdPara;Process ps = null;InputStreamReader isr = null;LineNumberReader input = null;String line = null;try {LOG.info("Run Command:   " + cmd );resultMsg.append("Run Command:   " + cmd +  "\n");ps = Runtime.getRuntime().exec(cmd);isr = new InputStreamReader(ps.getInputStream()); // 使用Reader進行輸入讀取和列印input = new LineNumberReader(isr);while (null != (line = input.readLine())) {LOG.info(line);resultMsg.append(line);if (line.contains("failed") || line.contains("Failed") || line.contains("FAILED") || line.contains("錯誤")) {isOK = RunStatus.EXPORT_FAIL;dataMap.put("export_status", isOK);dataMap.put("proc_log", resultMsg.toString());//dataMap = packageResult(isOK, resultMsg.toString());// 組裝返回的訊息return dataMap;} else {isOK = RunStatus.PROC_RUN_SUCCESS;}}//if (0 != ps.waitFor()) {//isOK = RunStatus.EXPORT_FAIL;//} else {//isOK = RunStatus.PROC_RUN_SUCCESS;//}} catch (IOException e) {LOG.error("Run the Command Exception: " + cmd + ": " + e.getMessage());resultMsg.append("Run the Command Exception: " + cmd  + ": " + e.getMessage() + "\n");isOK = RunStatus.EXPORT_FAIL;} finally {if (null != input) {input.close();}if (null != isr) {isr.close();}if (null != ps) {ps.destroy();ps = null;}}dataMap.put("export_status", isOK);dataMap.put("proc_log", resultMsg.toString());//dataMap = packageResult(isOK, resultMsg.toString());// 組裝返回的訊息return dataMap;}/** * @Title: createExportShellParams  * @Description:  組裝參數 * @param msgId * @param dbSchema * @param dbUser * @param dbPassword * @param tableName * @param filePath * @param delimiter * @param exportLimit * @return String 傳回型別 * @throws */private String createExportShellParams(String dbSchema, String dbUser, String dbPassword, String tableName,String filePath, String delimiter, String exportLimit) {StringBuilder params = new StringBuilder();params.append(dbSchema + " ").append(dbUser + " ").append(dbPassword + " ").append(tableName + " ").append(filePath + " ").append(delimiter + " ").append(exportLimit);return params.toString();}/** * @Title: mkDirectory  * @Description:  根據配置的路徑和檔案名稱，判斷檔案路徑是否存在，若不存在，則先建立，拼接匯出檔案絕對路徑。 * @param filePath * @param interFile * @return 對方法的參數進行描述 * @return String 傳回型別 * @throws */private String mkDirectory(StringBuffer filePath, String interFile) {File file = new File(filePath.toString());if ( file.isDirectory() ) {if (filePath.toString().endsWith("/")) {filePath.append(interFile);} else {filePath.append("/").append(interFile);}} else {LOG.info("The file path is not exists, need to be created now. ");file.mkdir();if (filePath.toString().endsWith("/")) {filePath.append(interFile);} else {filePath.append("/").append(interFile);}}return filePath.toString();}/** 返回訊息組裝結果 */private HashMap<String, String> packageResult(String isOK, String resultMsg) {HashMap<String, String> hsmap = new HashMap<String, String>();hsmap.put(MapKeys.PROC_STATUS, isOK);hsmap.put(MapKeys.PROC_LOG, resultMsg);return hsmap;}}

傳入的執行參數放入一個Map（HashMap<String, String> dataMap）中

/**  EXPORT TEST  */map.put("db_schema", "md");map.put("db_user", "root");map.put("db_password", "root");map.put("table_name", "inter_log");map.put("inter_file", "inter_log_20140915.avl");map.put("delimiter", "|");map.put("export_limit", "");

代碼執行之後，將執行日誌以及執行結果也存入該Map中一起返回

dataMap.put("export_status", isOK);dataMap.put("proc_log", resultMsg.toString());return dataMap;

執行結果介面：

650) this.width=650;" src="http://s3.51cto.com/wyfs02/M02/49/AA/wKioL1QYCYfzuZzpAAIgt-iNz48024.jpg" title="QQ20140916174454.png" alt="wKioL1QYCYfzuZzpAAIgt-iNz48024.jpg" />

本文出自 “Forever_Love_ING” 部落格，請務必保留此出處http://dwf07223.blog.51cto.com/8712758/1553422

Java代碼調用Shell指令碼並傳入參數實現DB2資料庫表匯出到檔案