歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux編程 >> SHELL編程 >> Java代碼調用Shell腳本並傳入參數實現DB2數據庫表導出到文件

Java代碼調用Shell腳本並傳入參數實現DB2數據庫表導出到文件

日期:2017/3/1 9:38:43   编辑:SHELL編程

本文通過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;

執行結果界面:

大話設計模式(帶目錄完整版) PDF+源代碼 http://www.linuxidc.com/Linux/2014-08/105152.htm

Java中介者設計模式 http://www.linuxidc.com/Linux/2014-07/104319.htm

Java 設計模式之模板方法開發中應用 http://www.linuxidc.com/Linux/2014-07/104318.htm

設計模式之 Java 中的單例模式(Singleton) http://www.linuxidc.com/Linux/2014-06/103542.htm

Copyright © Linux教程網 All Rights Reserved