You've already forked FrameTour-BE
默认存储方式
This commit is contained in:
@@ -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);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@@ -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;
|
||||
}
|
||||
|
@@ -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;
|
||||
}
|
||||
|
Reference in New Issue
Block a user