Update Generator.java

src/Generator.java

@@ -1,34 +1,32 @@
-import java.io.FileWriter;
+import java.io.FileWriter;
-import java.io.IOException;
+import java.io.IOException;
-import java.time.LocalDateTime;
+import java.time.LocalDateTime;
-import java.time.format.DateTimeFormatter;
+import java.time.format.DateTimeFormatter;
-
+
-public class Generator {
+public class Generator {
-  public static void generate(Users users, int count, ProblemsGenerator problemsGenerator) {
+  public static void generate(Users users, int count, ProblemsGenerator problemsGenerator) {
-    StringBuilder builder = new StringBuilder();
+    StringBuilder builder = new StringBuilder();
-    for (int i = 1; i <= count; i++) {
+    for (int i = 1; i <= count; i++) {
-      builder.append(i).append(". ").append(problemsGenerator.generate()).append("\n\n");
+      builder.append(i).append(". ").append(problemsGenerator.generate()).append("\n\n");
-    }
+    }
-    save(users, builder.toString());
+    save(users, builder.toString());
-    System.out.println("题目生成完成。");
+    System.out.println("题目生成完成。");
-  }
+  }
-  
+  
-  private static void save(Users users, String content) {
+  private static void save(Users users, String content) {
-    try {
+    try {
-      String folder = "doc/" + users.getType();
+      String folder = "doc/" + users.getName();
-      java.io.File dir = new java.io.File(folder);
+      java.io.File dir = new java.io.File(folder);
-      if (!dir.exists()) {
+      if (!dir.exists()) {
-        dir.mkdirs();
+        dir.mkdirs();
-      }
+      }
-      String filename = LocalDateTime.now()
+      String filename = LocalDateTime.now()
-          .format(DateTimeFormatter.ofPattern("yyyy-MM-dd-HH-mm-ss")) + ".txt";
+          .format(DateTimeFormatter.ofPattern("yyyy-MM-dd-HH-mm-ss")) + ".txt";
-      FileWriter writer = new FileWriter(folder + "/" + filename);
+      FileWriter writer = new FileWriter(folder + "/" + filename);
-      writer.write(content);
+      writer.write(content);
-      writer.close();
+      writer.close();
-    } catch (IOException e) {
+    } catch (IOException e) {
-      System.out.println("保存失败：" + e.getMessage());
+      System.out.println("保存失败：" + e.getMessage());
-    }
+    }
-  }
+  }
-}
+}
-
-