diff --git a/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/config/MilvusPropertiesConfiguration.java b/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/config/MilvusPropertiesConfiguration.java index 135a57a..da9ef06 100644 --- a/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/config/MilvusPropertiesConfiguration.java +++ b/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/config/MilvusPropertiesConfiguration.java @@ -21,4 +21,5 @@ public class MilvusPropertiesConfiguration { private String token; private List packages; private boolean openLog; + private String logLevel; } \ No newline at end of file diff --git a/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java b/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java index 6b10358..8abae15 100644 --- a/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java +++ b/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java @@ -24,7 +24,11 @@ public class MilvusInit extends AbstractMilvusClientBuilder { @PostConstruct public void initialize() { printBanner(); - LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus", milvusPropertiesConfiguration.isOpenLog()); + + LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus", + milvusPropertiesConfiguration.isOpenLog(), + milvusPropertiesConfiguration.getLogLevel()); + MilvusProperties milvusProperties = new MilvusProperties(); BeanUtils.copyProperties(milvusPropertiesConfiguration, milvusProperties); super.setProperties(milvusProperties); diff --git a/milvus-plus-core/src/main/java/org/dromara/milvus/plus/log/LogLevelController.java b/milvus-plus-core/src/main/java/org/dromara/milvus/plus/log/LogLevelController.java index be467c2..1de085e 100644 --- a/milvus-plus-core/src/main/java/org/dromara/milvus/plus/log/LogLevelController.java +++ b/milvus-plus-core/src/main/java/org/dromara/milvus/plus/log/LogLevelController.java @@ -11,8 +11,9 @@ public class LogLevelController { /** * 设置特定包下所有类的日志级别。 + * * @param packageName 包名 - * @param level 日志级别 + * @param level 日志级别 */ public static void setLogLevelForPackage(String packageName, Level level) { for (ch.qos.logback.classic.Logger logger : loggerContext.getLoggerList()) { @@ -25,11 +26,16 @@ public class LogLevelController { /** * 动态设置日志开关。 * 当设置为Level.OFF时,等同于关闭日志。 + * * @param packageName 包名 - * @param enabled 是否启用日志 + * @param enabled 是否启用日志 */ - public static void setLoggingEnabledForPackage(String packageName, boolean enabled) { - Level level = enabled ? Level.DEBUG : Level.OFF; // 可以根据需要设置为INFO, WARN, ERROR等 - setLogLevelForPackage(packageName, level); + public static void setLoggingEnabledForPackage(String packageName, boolean enabled, String level) { + // 开启日志 默认 -> debug + if (enabled) { + setLogLevelForPackage(packageName, Level.toLevel(level, Level.DEBUG)); + } else { + setLogLevelForPackage(packageName, Level.OFF); + } } } \ No newline at end of file diff --git a/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/entity/MilvusPropertiesConfiguration.java b/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/entity/MilvusPropertiesConfiguration.java index a678c73..0c44129 100644 --- a/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/entity/MilvusPropertiesConfiguration.java +++ b/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/entity/MilvusPropertiesConfiguration.java @@ -21,4 +21,5 @@ public class MilvusPropertiesConfiguration { private String token; private List packages; private boolean openLog; + private String logLevel; } \ No newline at end of file diff --git a/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/service/MilvusInit.java b/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/service/MilvusInit.java index b12aa55..2062722 100644 --- a/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/service/MilvusInit.java +++ b/milvus-plus-solon-plugin/src/main/java/org/dromara/solon/service/MilvusInit.java @@ -17,7 +17,9 @@ public class MilvusInit extends AbstractMilvusClientBuilder implements Lifecycle @Bean public MilvusClientV2 init(MilvusPropertiesConfiguration milvusPropertiesConfiguration) { printBanner(); - LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus", milvusPropertiesConfiguration.isOpenLog()); + LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus", + milvusPropertiesConfiguration.isOpenLog(), + milvusPropertiesConfiguration.getLogLevel()); MilvusProperties milvusProperties = new MilvusProperties(); BeanUtils.copyProperties(milvusPropertiesConfiguration, milvusProperties); super.setProperties(milvusProperties);