重构读取文件内容送审代码、输出通过内容到指定文件代码

王言钊
1 parent 07b32855
Showing 1 changed file with 64 additions and 20 deletions
src/main/java/com/cnlive/shenhe/job/TextJob.java → src/main/java/com/cnlive/shenhe/utils/TextDetectionUtil.java
-package com.cnlive.shenhe.job;
+package com.cnlive.shenhe.utils;
  
 import com.alibaba.fastjson.JSONObject;
-import com.cnlive.shenhe.utils.YiDunAntiFraudUtil;
 import lombok.extern.slf4j.Slf4j;
  
 import java.io.*;
@@ -11,45 +10,59 @@ import java.io.*;
  * @DateTime: 2022-06-27 10:30
  */
 @Slf4j
-public class TextJob {
-
-
+public class TextDetectionUtil {
+    /**
+     * 文本内容送审方法
+     *
+     * @param args
+     */
     public static void main(String[] args) {
-        BufferedReader bufferedReader = null;
-        BufferedWriter bufferedWriter = null;
-        String type = null;
-        String desc = null;
+        // 输入文件地址
+        String fileReaderUrl = "C:\\Users\\Administrator\\Desktop\\待检测数据.txt";
+        // 输出文件地址
+        String fileWriterUrl = "C:\\Users\\Administrator\\Desktop\\通过数据.txt";
+        // 调用输入流方法
+        BufferedReader bufferedReader = FileReaderUrl(fileReaderUrl);
+        // 调用输出流方法
+        BufferedWriter bufferedWriter = FileWriterUrl(fileWriterUrl);
         try {
-            bufferedReader = new BufferedReader(new FileReader(new File("C:\\Users\\Administrator\\Desktop\\123.txt")));
-            bufferedWriter = new BufferedWriter(new BufferedWriter(new FileWriter(new File("C:\\Users\\Administrator\\Desktop\\通过数据.txt"))));
-            boolean falg = true;
-            while (falg) {
-                if (falg = (bufferedReader.readLine()) != null ? true : false) {
-                    String txt = bufferedReader.readLine();
-                    JSONObject textFilter = YiDunAntiFraudUtil.getTextFilter(txt);
-                    type = textFilter.getString("type");
-                    desc = textFilter.getString("desc");
+            boolean flag = true;
+            while (flag) {
+                String textContent = bufferedReader.readLine();
+                if (flag = textContent != null ? true : false) {
+                    // 文本内容送审
+                    JSONObject textFilter = YiDunAntiFraudUtil.getTextFilter(textContent);
+                    // 送审返回结果类型
+                    String type = textFilter.getString("type");
+                    // 送审返回结果内容
+                    String desc = textFilter.getString("desc");
+                    // 如果是通过的内容 输出到文件中
                     if (type.equals("PASS")) {
-                        bufferedWriter.write(txt);
+                        // 输出（写）的操作
+                        bufferedWriter.write(textContent);
+                        // 换行操作
                         bufferedWriter.newLine();
                     }
                     log.info("type:{}" + type);
                     log.info("desc:{}" + desc);
                 }
             }
-
         } catch (IOException e) {
             e.printStackTrace();
         } finally {
+            // 如果输入流不为空
             if (bufferedReader != null) {
                 try {
+                    // 关闭输入流
                     bufferedReader.close();
                 } catch (IOException e) {
                     e.printStackTrace();
                 }
             }
+            // 如果输出流不为空
             if (bufferedWriter != null) {
                 try {
+                    // 关闭输出流
                     bufferedWriter.close();
                 } catch (IOException e) {
                     e.printStackTrace();
@@ -58,4 +71,35 @@ public class TextJob {
         }
     }
  
+    /**
+     * 创建文件输入流
+     *
+     * @param fileReaderUrl
+     * @return
+     */
+    public static BufferedReader FileReaderUrl(String fileReaderUrl) {
+        BufferedReader bufferedReader = null;
+        try {
+            bufferedReader = new BufferedReader(new FileReader(new File(fileReaderUrl)));
+        } catch (FileNotFoundException e) {
+            e.printStackTrace();
+        }
+        return bufferedReader;
+    }
+
+    /**
+     * 创建文件输出流
+     *
+     * @param fileWriterUrl
+     * @return
+     */
+    public static BufferedWriter FileWriterUrl(String fileWriterUrl) {
+        BufferedWriter bufferedWriter = null;
+        try {
+            bufferedWriter = new BufferedWriter(new FileWriter(new File(fileWriterUrl)));
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        return bufferedWriter;
+    }
 }