com.vip.saturn.job.console.utils.SaturnConsoleUtils Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2016 vip.com.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 *  the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 * 
 **/

package com.vip.saturn.job.console.utils;

import com.vip.saturn.job.console.domain.JobType;
import com.vip.saturn.job.console.exception.SaturnJobConsoleException;
import com.vip.saturn.job.console.repository.zookeeper.CuratorRepository.CuratorFrameworkOp;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.joda.time.DateTime;
import org.joda.time.DateTimeZone;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.Random;
import java.util.TimeZone;

/**
 * Utils for saturn console.
 *
 * @author kfchu
 */
public class SaturnConsoleUtils {

	private static DateTimeFormatter dtf = DateTimeFormat.forPattern("yyyy-MM-dd HH:mm:ss");

	private static Random random = new Random();

	public static Logger getAuditLogger() {
		return LoggerFactory.getLogger("AUDITLOG");
	}

	public static String parseMillisecond2DisplayTime(String longInStr) {
		return parseMillisecond2DisplayTime(longInStr, null);
	}

	public static String parseMillisecond2DisplayTime(String longInStr, TimeZone timezone) {
		if (StringUtils.isBlank(longInStr)) {
			return null;
		}

		return dtf.print(new DateTime(Long.parseLong(longInStr), DateTimeZone.forTimeZone(timezone)));
	}

	/**
	 * 如果存在/config/enabledReport节点，则返回节点的内容； 如果不存在/config/enabledReport节点，如果作业类型是Java或者Shell，则返回true；否则，返回false；
	 */
	public static boolean checkIfJobIsEnabledReport(String jobName, CuratorFrameworkOp curatorFrameworkOp) {
		String enabledReportNodePath = JobNodePath.getEnabledReportNodePath(jobName);

		if (curatorFrameworkOp.checkExists(enabledReportNodePath)) {
			return Boolean.parseBoolean(curatorFrameworkOp.getData(enabledReportNodePath));
		}

		String jobTypeNodePath = JobNodePath.getConfigNodePath(jobName, "jobType");

		// if enabledReportNodePath不存在, 如果作业类型是cron或passive，默认上报
		JobType jobType = JobType.getJobType(curatorFrameworkOp.getData(jobTypeNodePath));

		return JobType.isCron(jobType) || JobType.isPassive(jobType);
	}

	public static File createTmpFile() throws SaturnJobConsoleException, IOException {
		int loopTimes = 5;

		int i = 0;
		File tmp = null;
		for (; i < loopTimes; i++) {
			tmp = new File(SaturnConstants.CACHES_FILE_PATH, genTmpFileName());
			if (!tmp.exists()) {
				break;
			}
		}

		if (i == loopTimes) {
			throw new SaturnJobConsoleException("fail to create temp file.");
		}

		FileUtils.forceMkdir(tmp.getParentFile());
		if (tmp.createNewFile()) {
			return tmp;
		} else {
			throw new SaturnJobConsoleException("fail to create temp file.");
		}
	}

	private static String genTmpFileName() {
		return "tmp_exportFile_" + System.currentTimeMillis() + "_" + random.nextInt(1000) + ".xls";
	}

	public static void exportFile(HttpServletResponse response, File srcFile, String exportFileName,
			boolean deleteTmpFile) throws SaturnJobConsoleException {
		try (InputStream inputStream = new FileInputStream(srcFile)) {
			exportFile(response, inputStream, exportFileName);
		} catch (IOException e) {
			throw new SaturnJobConsoleException("file not found:" + srcFile.getName(), e);
		} finally {
			if (deleteTmpFile && srcFile != null) {
				srcFile.delete();
			}
		}
	}

	public static void exportFile(HttpServletResponse response, InputStream inputStream, String exportFileName)
			throws SaturnJobConsoleException {
		try {
			response.setContentType("application/octet-stream");
			response.setHeader("Content-disposition",
					"attachment; filename=" + new String(exportFileName.getBytes("UTF-8"), "ISO8859-1"));

			try (BufferedInputStream bis = new BufferedInputStream(inputStream);
					BufferedOutputStream bos = new BufferedOutputStream(response.getOutputStream())) {
				byte[] buff = new byte[2048];
				int bytesRead;
				while (-1 != (bytesRead = bis.read(buff, 0, buff.length))) {
					bos.write(buff, 0, bytesRead);
				}
			}
		} catch (Exception e) {
			throw new SaturnJobConsoleException(e);
		}
	}

}