admin/MilvusPlus
1
0
Fork 0
mirror of https://gitee.com/dromara/MilvusPlus.git synced 2025-12-06 17:08:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 支持配置日志级别

Browse Source
This commit is contained in:
code2tan 2024-06-07 00:38:37 +08:00
parent 19bf4c6811
commit afca393473
5 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/config/MilvusPropertiesConfiguration.java
View File

@ -21,4 +21,5 @@ public class MilvusPropertiesConfiguration {
private String token;
private List<String> packages;
private boolean openLog;
private String logLevel;
}

6
milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java
View File

@ -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);

12
milvus-plus-core/src/main/java/org/dromara/milvus/plus/log/LogLevelController.java
View File

@ -11,6 +11,7 @@ public class LogLevelController {
/**
* 设置特定包下所有类的日志级别。
*
* @param packageName 包名
* @param level 日志级别
*/
@ -25,11 +26,16 @@ public class LogLevelController {
/**
* 动态设置日志开关。
* 当设置为Level.OFF时，等同于关闭日志。
*
* @param packageName 包名
* @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);
}
}
}

1
milvus-plus-solon-plugin/src/main/java/org/dromara/solon/entity/MilvusPropertiesConfiguration.java
View File

@ -21,4 +21,5 @@ public class MilvusPropertiesConfiguration {
private String token;
private List<String> packages;
private boolean openLog;
private String logLevel;
}

4
milvus-plus-solon-plugin/src/main/java/org/dromara/solon/service/MilvusInit.java
View File

@ -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);