Update Generator.java

src/Generator.java

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