歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux編程 >> SHELL編程

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

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