From 4c7296d45efe849dc70a3b2e2240c905481a91c9 Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期一, 29 九月 2025 17:44:46 +0800
Subject: [PATCH] 工控网涉密网联调
---
blade-service/blade-mdm/src/main/java/org/springblade/mdm/utils/FileContentUtil.java | 94 +++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 91 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 315080f..b3fa140 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,16 +104,40 @@
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);
}
//绁炲鐨勬椂鍒诲氨鍦ㄨ繖涓柟娉曚簡锛岃繑鍥炲瓧绗﹂泦缂栫爜銆�
return detector.getDetectedCharset();
+ }
+
+ /**
+ * 璇诲彇鏂囦欢鏌愯
+ * @param inputStream
+ * @param lineIndex 琛岀储寮� 0based
+ * @return 琛屾枃鏈�
+ * @throws IOException
+ */
+ public static String readLineAt(InputStream inputStream,int lineIndex) throws IOException {
+ byte[] bytes = inputStream.readAllBytes();
+ ByteArrayInputStream byteStream = new ByteArrayInputStream(bytes);
+ Charset charset = Charsets.charset(detectFromInputStream(byteStream));
+ byteStream.reset();//閲嶇疆锛屼娇缁х画鍙敤
+
+ String lineText = "";
+ try (BufferedReader reader = new BufferedReader(new InputStreamReader(byteStream,charset))) {
+ // 璺宠繃绗竴琛�
+ int current = 0;
+ while(current<=lineIndex && lineText!=null) {
+ lineText = reader.readLine();
+ current++;
+ }
+ }
+ return lineText;
}
/**
@@ -118,4 +154,56 @@
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());
+ }
+
+ public static List<String> readFirstNLines(InputStream inputStream, int n) throws IOException {
+ List<String> lines = new ArrayList<>();
+
+ try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) {
+ String line;
+ int count = 0;
+
+ while ((line = reader.readLine()) != null && count < n) {
+ lines.add(line);
+ count++;
+ }
+ }
+
+ return lines;
+ }
}
--
Gitblit v1.9.3