| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367 |
- package com.goafanti.common.utils;
- import java.io.File;
- import java.io.FileInputStream;
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.OutputStream;
- import java.io.PrintWriter;
- import java.io.UnsupportedEncodingException;
- import java.net.URLEncoder;
- import java.util.ArrayList;
- import java.util.List;
- import java.util.regex.Pattern;
- import javax.servlet.http.HttpServletResponse;
- import org.apache.poi.hssf.usermodel.HSSFWorkbook;
- import org.apache.poi.ss.usermodel.Cell;
- import org.apache.poi.ss.usermodel.CellType;
- import org.apache.poi.ss.usermodel.Row;
- import org.apache.poi.ss.usermodel.Sheet;
- import org.apache.poi.ss.usermodel.Workbook;
- import org.apache.poi.xssf.usermodel.XSSFWorkbook;
- import org.springframework.beans.factory.annotation.Value;
- import org.springframework.web.multipart.MultipartFile;
- import com.goafanti.achievement.bo.InputAchievement;
- import com.goafanti.common.bo.Error;
- import com.goafanti.common.error.BusinessException;
- public class FileUtils {
- @Value(value = "${upload.private.path}")
- private String uploadPrivatePath = null;
- private static final Pattern SLASH_PATTERN = Pattern.compile("^/|/$");
- private static final String SLASH = "/";
- private static final String UNDERLINE = "_";
- /**
- * response 输出JSON
- *
- * @param hresponse
- * @param resultMap
- * @throws IOException
- */
- public static void out(HttpServletResponse response, String jsonStr) {
- PrintWriter out = null;
- try {
- response.setHeader("Content-Type", "application/json");
- response.setCharacterEncoding("UTF-8");
- out = response.getWriter();
- out.println(jsonStr);
- } catch (Exception e) {
- LoggerUtils.fmtError(FileUtils.class, e, "输出数据失败");
- } finally {
- if (null != out) {
- out.flush();
- out.close();
- }
- }
- }
- /**
- * Excel报表导出
- *
- * @param response
- * @param fileName
- * @param workbook
- */
- public static void downloadExcel(HttpServletResponse response, String fileName, HSSFWorkbook workbook) {
- String fn = null;
- try {
- fn = URLEncoder.encode(fileName, "UTF-8");
- } catch (UnsupportedEncodingException e1) {
- fn = fileName;
- }
- response.setContentType("application/x-download;charset=UTF-8");
- response.setHeader("Content-Disposition", "attachment; filename=\"" + fn + "\";");
- OutputStream out = null;
- try {
- out = response.getOutputStream();
- workbook.write(out);
- } catch (IOException e) {
- LoggerUtils.fmtError(FileUtils.class, e, "输出Excel失败");
- } finally {
- if (null != out) {
- try {
- out.flush();
- out.close();
- } catch (IOException e) {
- out = null;
- }
- }
- }
- }
- /**
- * 根据上传文件 拼接fileName
- *
- * @param mf
- * @param isPrivate
- * @param sign
- * @param path
- * @return
- */
- public static String mosaicFileName(MultipartFile mf, boolean isPrivate, String sign, String path, String uid) {
- String suffix = mf.getOriginalFilename().substring(mf.getOriginalFilename().lastIndexOf("."));
- boolean uniq = false;
- if (sign.indexOf("demand_picture") != -1|| sign.indexOf("achievement_picture") != -1 || sign.indexOf("user_picture")!=-1
- || sign.indexOf("jt_project_picture")!=-1 || sign.indexOf("jt_business_picture")!=-1 || sign.indexOf("honor_picture")!=-1
- || sign.indexOf("banners")!=-1|| sign.indexOf("head_portrait")!=-1 || sign.indexOf("video")!=-1 ||sign.indexOf("eventPlanning")!=-1 ) {
- uniq = true;
- }
- String fileName = "";
- if (isPrivate || StringUtils.isNotBlank(sign)) {
- if (uniq) {
- fileName = path + uid + "/" + System.nanoTime() + "_" + sign + suffix;
- } else {
- fileName = path + uid + "/" + sign + suffix;
- }
- } else {
- fileName = path + uid + "/" + System.nanoTime() + suffix;
- }
- return fileName;
- }
- public static String mosaicFileName( String fn, boolean isPrivate, String path, String uid) {
- String fileName = "";
- if (isPrivate) {
- fileName = path + uid + "/" + fn;
- } else {
- fileName = path + uid + "/" + "logo" + fn ;
- }
- return fileName;
- }
- /**
- * 获取下载文件名称
- *
- * @param path
- * @return
- */
- public static String getDownloadFileName(String path) {
- if (StringUtils.isBlank(path) || path.lastIndexOf(".") < 0) {
- return null;
- }
- String prefix = "附件";
- String suffix = path.substring(path.lastIndexOf("."));
- return prefix + suffix;
- }
- public static String getSuffix(String path) {
- if (StringUtils.isBlank(path) || path.lastIndexOf(".") < 0) {
- return "";
- }
- return path.substring(path.lastIndexOf("."));
- }
- /**
- * 拼接文件路径
- *
- * @param root
- * @param params
- * @return
- */
- public static String buildFilePath(String... params) {
- StringBuilder sb = new StringBuilder();
- if (params != null) {
- for (String s : params) {
- sb.append(SLASH);
- sb.append(SLASH_PATTERN.matcher(s).replaceAll(""));
- }
- }
- return sb.toString();
- }
- /**
- * 拼接文件名字
- *
- * @param root
- * @param params
- * @return
- */
- public static String buildFileName(String... params) {
- return StringUtils.join(params, UNDERLINE);
- }
- // year
- private static String subStr(String s) {
- return "企业" + s.substring(s.lastIndexOf(UNDERLINE) + 1, s.lastIndexOf("."));
- }
- public static boolean deleteFile(String realFilePath) {
- return new File(realFilePath).delete();
- }
- /**
- *
- * @param response
- * @param fileName
- * @param realFilePath
- */
- public static void downloadFile(HttpServletResponse response, String fileName, String realFilePath) {
- InputStream in = null;
- OutputStream out = null;
- byte[] buffer = new byte[8 * 1024];
- try {
- File file = new File(realFilePath);
- in = new FileInputStream(file);
- out = response.getOutputStream();
- // 设置文件MIME类型
- response.setContentType("application/octet-stream");
- fileName = java.net.URLEncoder.encode(fileName, "UTF-8");
- response.setHeader("Content-Disposition", "attachment; filename=" + fileName);
- response.setHeader("Content-Length", String.valueOf(file.length()));
- for (;;) {
- int bytes = in.read(buffer);
- if (bytes == -1) {
- break;
- }
- out.write(buffer, 0, bytes);
- }
- } catch (IOException e) {
- LoggerUtils.fmtError(FileUtils.class, e, "IO错误:%s", e.getMessage());
- } finally {
- try {
- if (in != null) {
- in.close();
- }
- } catch (IOException e) {
- in = null;
- LoggerUtils.fmtError(FileUtils.class, e, "IO错误:%s", e.getMessage());
- }
- try {
- if (out != null) {
- out.close();
- }
- } catch (IOException e) {
- out = null;
- LoggerUtils.fmtError(FileUtils.class, e, "IO错误:%s", e.getMessage());
- }
- }
- }
- public static Workbook getWorkBook(MultipartFile file) {
- //创建Workbook工作薄对象,表示整个excel
- Workbook workbook = null;
- try {
- //获取excel文件的io流
- InputStream is = file.getInputStream();
- //如果是xls,使用HSSFWorkbook;如果是xlsx,使用XSSFWorkbook
- if(file.getOriginalFilename().matches("^.+\\.(?i)(xls)$"))workbook = new HSSFWorkbook(is);
- if(file.getOriginalFilename().matches("^.+\\.(?i)(xlsx)$"))workbook = new XSSFWorkbook(is);
- } catch (IOException e) {
- }
- return workbook;
- }
- public static Sheet pushReadFile(MultipartFile file) {
- //获得Workbook工作薄对象
- Workbook workbook = getWorkBook(file);
- Sheet sheet=null;
- if(workbook != null){
- //获得当前sheet工作表
- sheet = workbook.getSheetAt(0);
- if(sheet == null){
- throw new BusinessException(new Error("未找到正确的参数。"));
- }
- }
- return sheet;
- }
- public static int rowDispose(Sheet sheet, int firstRowNum) {
- int lastRowNum=sheet.getLastRowNum();
- //处理空白行
- for (int i = firstRowNum; i <= sheet.getLastRowNum();i++) {
- Row r = sheet.getRow(i);
- if(r == null){
- // 如果是空行(即没有任何数据、格式),直接把它以下的数据往上移动
- sheet.shiftRows(i+1, sheet.getLastRowNum(),-1);
- continue;
- }
- boolean flag = false;
- for(Cell c:r){
- if(c.getCellTypeEnum() != CellType.BLANK){
- flag = true;
- break;
- }
- }
- if(flag){
- i++;
- continue;
- } else{
- //如果是空白行(即可能没有数据,但是有一定格式)
- if(i == sheet.getLastRowNum())//如果到了最后一行,直接将那一行remove掉
- sheet.removeRow(r);
- else//如果还没到最后一行,则数据往上移一行
- sheet.shiftRows(i+1, sheet.getLastRowNum(),-1);
- }
- }
- if (lastRowNum<2) {
- throw new BusinessException(new Error("未找到正确的参数。"));
- }
- return sheet.getLastRowNum();
- }
- public static int rowDisposeNotBlank(Sheet sheet, int firstRowNum) {
- int lastRowNum=sheet.getLastRowNum();
- //行数从0开始 默认使用2行做头部,则设置1
- int i=1;
- for(int rowNum = firstRowNum+2;rowNum <= lastRowNum;rowNum++){
- Row row = sheet.getRow(rowNum);
- if(row == null){
- continue;
- }
- Cell cell = row.getCell(0);
- String str =FileUtils.getCellValue(cell);
- if (StringUtils.isNotBlank(str)) {
- i++;
- }
- }
- if (i<2) {
- throw new BusinessException(new Error("未找到正确的参数。"));
- }
- return i;
- }
- public static String getCellValue(Cell cell){
- String cellValue = "";
- if(cell == null){
- return cellValue;
- }
- //把数字当成String来读,避免出现1读成1.0的情况
- if(cell.getCellTypeEnum() == CellType.NUMERIC){
- cell.setCellType(CellType.STRING);
- }
- //判断数据的类型
- switch (cell.getCellTypeEnum()){
- case NUMERIC: //数字
- cellValue = String.valueOf(cell.getNumericCellValue());
- break;
- case STRING: //字符串
- cellValue = String.valueOf(cell.getStringCellValue());
- break;
- case BOOLEAN: //Boolean
- cellValue = String.valueOf(cell.getBooleanCellValue());
- break;
- case FORMULA: //公式
- cellValue = String.valueOf(cell.getCellFormula());
- break;
- case BLANK: //空值
- cellValue = "";
- break;
- case ERROR: //故障
- cellValue = "非法字符";
- break;
- default:
- cellValue = "未知类型";
- break;
- }
- return cellValue;
- }
- }
|