diff --git a/src/Generator.java b/src/Generator.java new file mode 100644 index 0000000..200797f --- /dev/null +++ b/src/Generator.java @@ -0,0 +1,34 @@ +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()); + } + } +} + +