默认存储方式

This commit is contained in:
2024-12-27 11:28:19 +08:00
parent 201a6617ac
commit fdf35224dd
8 changed files with 33 additions and 24 deletions

View File

@@ -3,28 +3,30 @@ package com.ycwl.basic.storage.starter;
import com.ycwl.basic.storage.StorageFactory;
import com.ycwl.basic.storage.adapters.IStorageAdapter;
import com.ycwl.basic.storage.starter.config.OverallStorageConfig;
import com.ycwl.basic.storage.starter.config.StorageConfigItem;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.annotation.Configuration;
import java.util.Map;
@Configuration
public class StorageAutoConfiguration {
private final OverallStorageConfig config;
public StorageAutoConfiguration(OverallStorageConfig config) {
this.config = config;
if (config != null) {
loadConfig();
if (config.getConfigs() != null) {
loadConfig();
}
if (StringUtils.isNotBlank(config.getDefaultUse())) {
StorageFactory.setDefault(config.getDefaultUse());
}
}
}
private void loadConfig() {
config.getConfigs().forEach(item -> {
String name = item.getName();
String type = item.getType();
IStorageAdapter adapter = StorageFactory.get(type);
Map<String, String> config = item.getConfig();
adapter.loadConfig(config);
StorageFactory.register(name, adapter);
});
for (StorageConfigItem item : config.getConfigs()) {
IStorageAdapter adapter = StorageFactory.get(item.getType());
adapter.loadConfig(item.getConfig());
StorageFactory.register(item.getName(), adapter);
}
}
}

View File

@@ -5,11 +5,11 @@ import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.List;
import java.util.Map;
@Component
@ConfigurationProperties(prefix = "storage")
@Data
public class OverallStorageConfig {
private String defaultUse;
private List<StorageConfigItem> configs;
}

View File

@@ -1,5 +1,6 @@
package com.ycwl.basic.storage.starter.config;
import com.ycwl.basic.storage.enums.StorageType;
import lombok.Data;
import java.util.Map;
@@ -7,6 +8,6 @@ import java.util.Map;
@Data
public class StorageConfigItem {
private String name;
private String type;
private StorageType type;
private Map<String, String> config;
}