From 72e6bac02526caebc3797a7c2934d12896805708 Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期二, 23 九月 2025 00:47:18 +0800
Subject: [PATCH] 修复文件状态读取问题
---
blade-service/blade-mdm/src/main/java/org/springblade/mdm/utils/FileContentUtil.java | 53 ++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 50 insertions(+), 3 deletions(-)
diff --git a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/utils/FileContentUtil.java b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/utils/FileContentUtil.java
index f29bbdd..93b49a6 100644
--- a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/utils/FileContentUtil.java
+++ b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/utils/FileContentUtil.java
@@ -1,12 +1,22 @@
package org.springblade.mdm.utils;
import org.apache.tika.Tika;
+import org.apache.tika.config.TikaConfig;
+import org.apache.tika.detect.CompositeDetector;
+import org.apache.tika.detect.DefaultDetector;
+import org.apache.tika.detect.Detector;
+import org.apache.tika.detect.TextDetector;
+import org.apache.tika.mime.MimeType;
+import org.apache.tika.mime.MimeTypeException;
+import org.apache.tika.mime.MimeTypes;
import org.mozilla.universalchardet.UniversalDetector;
import org.springblade.core.tool.utils.Charsets;
import java.io.*;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
@@ -18,7 +28,9 @@
* @return
*/
public static boolean isTextFile(InputStream inputStream) {
- Tika tika = new Tika();
+ MimeTypes mimeTypes = MimeTypes.getDefaultMimeTypes();
+ Tika tika = new Tika(mimeTypes);
+
try {
String mimeType = tika.detect(inputStream);
//String mimeType = tika.detect(file);
@@ -92,11 +104,10 @@
byte[] buffer = new byte[2048];
try {
int actRead = inputStream.read(buffer);
-
detector.handleData(buffer, 0, actRead);
-
//璇嗗埆缁撴潫蹇呴』璋冪敤杩欎釜鏂规硶
detector.dataEnd();
+ detector.reset();
} catch (IOException e) {
throw new RuntimeException(e);
}
@@ -143,4 +154,40 @@
return new String(bytes, charset);
}
+
+ /**
+ * 鏇挎崲鏂囦欢涓殑鏌愪竴琛�
+ * @param ins 杈撳叆娴�
+ * @param replaceWith 鏇挎崲鐨勬枃鏈�
+ * @param lineIndex 琛岀储寮� 0 based
+ */
+ public static InputStream replaceAtLine(InputStream ins, int lineIndex,String replaceWith) throws IOException {
+ byte[] bytes = ins.readAllBytes();
+ ByteArrayInputStream byteStream = new ByteArrayInputStream(bytes);
+ Charset charset = Charsets.charset(detectFromInputStream(byteStream));
+ byteStream.reset();
+
+ Path tempFile = Files.createTempFile("temppro"+System.currentTimeMillis(), ".tmp");
+
+ try (InputStreamReader isr = new InputStreamReader(byteStream, charset);
+ BufferedReader reader = new BufferedReader(isr);
+ BufferedWriter writer = Files.newBufferedWriter(tempFile,charset)) {//
+
+ int currentLine = 0;
+ String line;
+
+ while ((line = reader.readLine()) != null) {
+ // 濡傛灉鏄洰鏍囪锛屽啓鍏ユ柊鍐呭
+ if (currentLine == lineIndex) {
+ writer.write(replaceWith);
+ } else {
+ writer.write(line);
+ }
+ writer.newLine();
+ currentLine++;
+ }
+
+ }
+ return new ByteArrayInputStream(Files.newInputStream(tempFile).readAllBytes());
+ }
}
--
Gitblit v1.9.3