com.clustercontrol.agent.job.CheckSumThread.java Source code

Introduction

Here is the source code for com.clustercontrol.agent.job.CheckSumThread.java

Source

/*
    
Copyright (C) 2011 NTT DATA Corporation
    
This program is free software; you can redistribute it and/or
Modify it under the terms of the GNU General Public License
as published by the Free Software Foundation, version 2.
    
This program is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied
warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE.  See the GNU General Public License for more details.
    
 */

package com.clustercontrol.agent.job;

import java.io.FileInputStream;
import java.io.IOException;
import java.security.MessageDigest;
import java.util.Date;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.clustercontrol.agent.SendQueue;
import com.clustercontrol.jobmanagement.bean.CommandConstant;
import com.clustercontrol.jobmanagement.bean.RunStatusConstant;
import com.clustercontrol.util.HinemosTime;
import com.clustercontrol.ws.jobmanagement.RunInstructionInfo;
import com.clustercontrol.ws.jobmanagement.RunResultInfo;

/**
 * ??<BR>
 * 
 */
public class CheckSumThread extends AgentThread {

    //
    private static Log m_log = LogFactory.getLog(CheckSumThread.class);

    protected static final String ALGORITHM = "MD5";

    /**
     * 
     * 
     * @param props
     */
    public CheckSumThread(RunInstructionInfo info, SendQueue sendQueue) {
        super(info, sendQueue);
    }

    /* (non-Javadoc)
     * @see java.lang.Runnable#run()
     */
    @Override
    public void run() {
        m_log.debug("run start");

        Date startDate = HinemosTime.getDateInstance();

        //?
        RunHistoryUtil.addRunHistory(m_info, RunHistoryUtil.dummyProcess());

        //---------------------------
        //-- ?
        //---------------------------

        //?
        RunResultInfo info = new RunResultInfo();
        info.setSessionId(m_info.getSessionId());
        info.setJobunitId(m_info.getJobunitId());
        info.setJobId(m_info.getJobId());
        info.setFacilityId(m_info.getFacilityId());
        info.setCommand(m_info.getCommand());
        info.setCommandType(m_info.getCommandType());
        info.setStopType(m_info.getStopType());
        info.setStatus(RunStatusConstant.START);
        info.setTime(startDate.getTime());

        m_log.info("run SessionID=" + m_info.getSessionId() + ", JobID=" + m_info.getJobId());

        //?
        m_sendQueue.put(info);

        if (m_info.getCommand().equals(CommandConstant.GET_CHECKSUM)) {
            String checksum = getCheckSum(m_info.getFilePath());
            if (checksum != null) {
                info.setStatus(RunStatusConstant.END);
                info.setCheckSum(checksum);
                info.setTime(HinemosTime.getDateInstance().getTime());
                info.setErrorMessage("");
                info.setMessage("");
                info.setEndValue(0);
            } else {
                String message = "GET_CHECKSUM is failure";
                m_log.warn(message);
                info.setStatus(RunStatusConstant.ERROR);
                info.setTime(HinemosTime.getDateInstance().getTime());
                info.setErrorMessage("");
                info.setMessage(message);
                info.setEndValue(-1);
            }
        } else if (m_info.getCommand().equals(CommandConstant.CHECK_CHECKSUM)) {
            String checksum = getCheckSum(m_info.getFilePath());
            if (checksum.equals(m_info.getCheckSum())) {
                info.setStatus(RunStatusConstant.END);
                info.setTime(HinemosTime.getDateInstance().getTime());
                info.setErrorMessage("");
                info.setMessage("");
                info.setEndValue(0);
            } else {
                String message = "CHECK_CHECKSUM is failure." + " from=" + m_info.getCheckSum() + ", to="
                        + checksum;
                m_log.warn(message);
                info.setStatus(RunStatusConstant.ERROR);
                info.setTime(HinemosTime.getDateInstance().getTime());
                info.setErrorMessage("");
                info.setMessage(message);
                info.setEndValue(-1);
            }
        }

        //?
        m_sendQueue.put(info);

        //?
        RunHistoryUtil.delRunHistory(m_info);

        m_log.debug("run end");
    }

    /**
     * ??
     * 
     * @param path
     * @return
     */
    private String getCheckSum(String path) {
        m_log.debug("get checksum start");

        String checksum = null;
        FileInputStream inputStream = null;
        try {
            inputStream = new FileInputStream(path);
            MessageDigest md = MessageDigest.getInstance(ALGORITHM);

            byte[] readData = new byte[256];
            int len;
            while ((len = inputStream.read(readData)) >= 0) {
                md.update(readData, 0, len);
            }
            inputStream.close();

            checksum = changeString(md.digest());

        } catch (Exception e) {
            m_log.warn("getCheckSum error. " + e.getMessage(), e);
        } finally {
            if (inputStream != null)
                try {
                    inputStream.close();
                } catch (IOException e) {
                    throw new InternalError("failed to close the FileInputStream");
                }
        }

        m_log.debug("get checksum end. path=" + path + ", md5=" + checksum);
        return checksum;
    }

    /**
     * ????
     * 
     * @param digest
     * @return
     */
    private String changeString(byte[] digest) {
        StringBuilder hashString = new StringBuilder();
        for (int i = 0; i < digest.length; i++) {
            int d = digest[i];
            if (d < 0) {//?
                d += 256;
            }
            if (d < 16) {//1???2???
                hashString.append("0");
            }
            hashString.append(Integer.toString(d, 16));//162????
        }
        return hashString.toString();
    }
}