diff --git a/src/main/java/com/power/doc/builder/DocBuilderTemplate.java b/src/main/java/com/power/doc/builder/DocBuilderTemplate.java index 19a5b67..1b9922b 100644 --- a/src/main/java/com/power/doc/builder/DocBuilderTemplate.java +++ b/src/main/java/com/power/doc/builder/DocBuilderTemplate.java @@ -100,14 +100,18 @@ public class DocBuilderTemplate { String outPath = config.getOutPath(); String strTime = DateTimeUtil.long2Str(now, DateTimeUtil.DATE_FORMAT_SECOND); FileUtil.mkdirs(outPath); + List errorCodeList = config.getErrorCodes(); + if (CollectionUtil.isEmpty(errorCodeList)) { + errorCodeList = errorCodeDictToList(config); + } Template tpl = BeetlTemplateUtil.getByName(template); tpl.binding(TemplateVariable.API_DOC_LIST.getVariable(), apiDocList); - tpl.binding(TemplateVariable.ERROR_CODE_LIST.getVariable(), config.getErrorCodes()); + tpl.binding(TemplateVariable.ERROR_CODE_LIST.getVariable(), errorCodeList); tpl.binding(TemplateVariable.VERSION_LIST.getVariable(), config.getRevisionLogs()); tpl.binding(TemplateVariable.VERSION.getVariable(), now); tpl.binding(TemplateVariable.CREATE_TIME.getVariable(), strTime); tpl.binding(TemplateVariable.PROJECT_NAME.getVariable(), config.getProjectName()); - if (CollectionUtil.isEmpty(config.getErrorCodes())) { + if (CollectionUtil.isEmpty(errorCodeList)) { tpl.binding(TemplateVariable.DICT_ORDER.getVariable(), apiDocList.size() + 1); } else { tpl.binding(TemplateVariable.DICT_ORDER.getVariable(), apiDocList.size() + 2); diff --git a/src/main/java/com/power/doc/model/ApiErrorCodeDictionary.java b/src/main/java/com/power/doc/model/ApiErrorCodeDictionary.java index c4c631d..6a29d7c 100644 --- a/src/main/java/com/power/doc/model/ApiErrorCodeDictionary.java +++ b/src/main/java/com/power/doc/model/ApiErrorCodeDictionary.java @@ -28,6 +28,10 @@ public class ApiErrorCodeDictionary { return enumClass; } + public static ApiErrorCodeDictionary dict(){ + return new ApiErrorCodeDictionary(); + } + public ApiErrorCodeDictionary setEnumClass(Class enumClass) { this.enumClass = enumClass; return this;