commit

许鹏程

2024-05-28 b44bfff58de3adc572fd4643ecb3ec9b5586bea3

commit

 src/main/java/com/product/lucene/util/FileUtils.java

@@ -8,6 +8,7 @@
import java.io.InputStream;
import java.util.List;

import com.product.common.lang.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
@@ -36,20 +37,71 @@
     */
    public static String FileToString(File file) {
        String file_content = "";
        if ((file != null) && (file.isFile())) {
            if (file.getName().toLowerCase().endsWith(".txt")) {
                file_content = txtToString(file);
            } else if ((file.getName().toLowerCase().endsWith(".doc"))) {
                file_content = docToString(file);
            } else if ((file.getName().toLowerCase().endsWith(".docx"))) {
                file_content = docxToString(file);
            } else if (file.getName().toLowerCase().endsWith(".xls")) {
                file_content = readXls(file);
            } else if (file.getName().toLowerCase().endsWith(".xlsx")) {
                file_content = readXlsx(file);
        String fileType = "unknown";
        //获取文件后缀 保留.号
        String fileSuffix = file.getName().substring(file.getName().lastIndexOf("."));
        if (StringUtils.equalsAny(fileSuffix, ".doc", ".docx", ".xls", ".xlsx")) {
            //读取文件
            fileType = checkDocType(file);
        }
        try {
            if ((file != null) && (file.isFile())) {


                if ("doc".equals(fileType)) {
                    file_content = docToString(file);
                } else if ("docx".equals(fileType)) {
                    file_content = docxToString(file);
                } else if ("xls".equals(fileType)) {
                    file_content = readXls(file);
                } else if ("xlsx".equals(fileType)) {
                    file_content = readXlsx(file);
                } else if (file.getName().toLowerCase().endsWith(".txt")) {
                    file_content = txtToString(file);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return file_content;
    }


    public static String checkDocType(File file) {
        try (FileInputStream fis = new FileInputStream(file)) {
            byte[] bytes = new byte[8];
            fis.read(bytes, 0, 8);

            String hex = bytesToHex(bytes);

            if (hex.contains("504B0304") && (file.getName().endsWith(".doc") || file.getName().endsWith(".docx"))) {
                return "docx";
            } else if (hex.contains("D0CF11E0") && (file.getName().endsWith(".doc") || file.getName().endsWith(".docx"))) {
                //因为doc文件的头部也是D0CF11E0，所以需要判断文件后缀
                return "doc";
            }
            //增加xls 和 xlsx的判断
            else if (hex.contains("504B0304") && (file.getName().endsWith(".xls") || file.getName().endsWith(".xlsx"))) {
                return "xlsx";
            } else if (hex.contains("D0CF11E0") && (file.getName().endsWith(".xls") || file.getName().endsWith(".xlsx"))) {
                //因为xls文件的头部也是D0CF11E0，所以需要判断文件后缀
                return "xls";
            } else {
                return "unknown";
            }
        } catch (
                IOException e) {
            e.printStackTrace();
            return "unknown";
        }
    }

    private static String bytesToHex(byte[] bytes) {
        StringBuilder hex = new StringBuilder();
        for (byte b : bytes) {
            hex.append(String.format("%02X", b));
        }
        return hex.toString();
    }

    /**
@@ -58,7 +110,7 @@
     * @param file
     * @return
     */
    public static String txtToString(File file) {
    public static String txtToString(File file) throws IOException {
        String result = "";
        try (BufferedReader br = new BufferedReader(new FileReader(file))) {// 构造一个BufferedReader类来读取文件
            String s = null;
@@ -67,7 +119,7 @@
                result = result + "\n" + s;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
        return result;
    }
@@ -78,7 +130,7 @@
     * @param file
     * @return
     */
    public static String docToString(File file) {
    public static String docToString(File file) throws IOException {
        String result = "";
        try (
                FileInputStream fileInputStream = new FileInputStream(file);
@@ -87,7 +139,7 @@
            Range rang = doc.getRange();
            result += rang.text();
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
        return result;
    }
@@ -99,7 +151,8 @@
     * @return
     * @throws IOException
     */
    public static String docxToString(File file) {
    public static String docxToString(File file) throws IOException {

        StringBuffer s = new StringBuffer();
        try (
                InputStream inputStream = new FileInputStream(file);
@@ -129,7 +182,7 @@
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
        return s.toString();
    }
@@ -141,7 +194,7 @@
     * @return
     * @throws IOException
     */
    public static String readXls(File file) {
    public static String readXls(File file) throws IOException {
        StringBuffer info = new StringBuffer();
        try (
                InputStream inputStream = new FileInputStream(file);
@@ -166,7 +219,7 @@
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
        return info.toString();
    }
@@ -178,7 +231,7 @@
     * @return
     * @throws IOException
     */
    public static String readXlsx(File file) {
    public static String readXlsx(File file) throws IOException {
        StringBuffer info = new StringBuffer();
        try (
                InputStream is = new FileInputStream(file);
@@ -203,7 +256,7 @@
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
        return info.toString();
    }
@@ -215,15 +268,24 @@
     * @return
     */
    private static String getValue(HSSFCell hssfCell) {
        if (hssfCell == null)
        if (hssfCell == null) {
            return "";
        }
        CellType cellType = hssfCell.getCellType();
        if (CellType.BOOLEAN.equals(cellType)) {
            return String.valueOf(hssfCell.getBooleanCellValue());
        } else if (CellType.NUMERIC.equals(cellType)) {
            return String.valueOf(hssfCell.getNumericCellValue());
        } else {
            return String.valueOf(hssfCell.getStringCellValue());
        //如果不是字符类型则返回空
        if (!CellType.BOOLEAN.equals(cellType) && !CellType.NUMERIC.equals(cellType) && !CellType.STRING.equals(cellType)) {
            return "";
        }
        try {
            if (CellType.BOOLEAN.equals(cellType)) {
                return String.valueOf(hssfCell.getBooleanCellValue());
            } else if (CellType.NUMERIC.equals(cellType)) {
                return String.valueOf(hssfCell.getNumericCellValue());
            } else {
                return String.valueOf(hssfCell.getStringCellValue());
            }
        } catch (Exception e) {
            throw e;
        }
    }

@@ -237,6 +299,10 @@
        if (xssfRow == null)
            return "";
        CellType cellType = xssfRow.getCellType();
        //如果不是字符类型则返回空
        if (!CellType.BOOLEAN.equals(cellType) && !CellType.NUMERIC.equals(cellType) && !CellType.STRING.equals(cellType)) {
            return "";
        }
        if (CellType.BOOLEAN.equals(cellType)) {
            return String.valueOf(xssfRow.getBooleanCellValue());
        } else if (CellType.NUMERIC.equals(cellType)) {
@@ -246,7 +312,7 @@
        }
    }

    public static void main(String[] args) {
    public static void main(String[] args) throws Exception {
        File f1 = new File("D://ceshi/a.txt");
        File f2 = new File("D://ceshi/b.doc");
        File f3 = new File("D://ceshi/c.docx");