From f5b083c6f9a3eee7aa98db44d1cbf9507bcca04d Mon Sep 17 00:00:00 2001 From: jaysunxiao Date: Sun, 11 Jul 2021 13:21:20 +0800 Subject: [PATCH] =?UTF-8?q?perf[protocol]:=20=E4=BC=98=E5=8C=96=E7=94=9F?= =?UTF-8?q?=E4=BA=A7=E5=8D=8F=E8=AE=AE=E7=9A=84=E6=97=B6=E5=80=99=E6=89=AB?= =?UTF-8?q?=E6=8F=8F=E4=BA=86=E5=A4=AA=E5=A4=9A=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../protocol/generate/GenerateProtocolDocument.java | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolDocument.java b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolDocument.java index 192d9433..d40f1f7e 100644 --- a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolDocument.java +++ b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolDocument.java @@ -76,12 +76,16 @@ public abstract class GenerateProtocolDocument { public static void initProtocolDocument(List protocolRegistrations) { AssertionUtils.notNull(tempProtocolDocumentMap, "[{}]已经初始完成，初始化完成过后不能调用initProtocolDocument", GenerateProtocolDocument.class.getSimpleName()); + // 文件的注释生成 + var proAbsFile = new File(FileUtils.getProAbsPath()); + var list = FileUtils.getAllReadableFiles(proAbsFile.getParentFile() == null ? proAbsFile : proAbsFile.getParentFile()) + .stream() + .filter(it -> it.getName().endsWith(".java")) + .collect(Collectors.toList()); + for (var protocolRegistration : protocolRegistrations) { var protocolClazzName = protocolRegistration.protocolConstructor().getDeclaringClass().getSimpleName(); - // 文件的注释生成 - var proAbsFile = new File(FileUtils.getProAbsPath()); - var list = FileUtils.getAllReadableFiles(proAbsFile.getParentFile() == null ? proAbsFile : proAbsFile.getParentFile()); var protocolFile = list.stream() .filter(it -> it.getName().equals(StringUtils.format("{}.java", protocolClazzName))) .findFirst();