|
|
@ -7,7 +7,6 @@ import org.springframework.core.env.PropertiesPropertySource; |
|
|
|
import org.springframework.core.env.PropertySource; |
|
|
|
import org.springframework.core.env.PropertySource; |
|
|
|
import org.springframework.core.io.ClassPathResource; |
|
|
|
import org.springframework.core.io.ClassPathResource; |
|
|
|
import org.springframework.core.io.Resource; |
|
|
|
import org.springframework.core.io.Resource; |
|
|
|
import org.springframework.util.unit.DataSize; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import java.io.IOException; |
|
|
|
import java.io.IOException; |
|
|
|
import java.util.Properties; |
|
|
|
import java.util.Properties; |
|
|
@ -32,21 +31,10 @@ public class SopGatewayEnvironmentPostProcessor implements EnvironmentPostProces |
|
|
|
} |
|
|
|
} |
|
|
|
try { |
|
|
|
try { |
|
|
|
properties.load(resource.getInputStream()); |
|
|
|
properties.load(resource.getInputStream()); |
|
|
|
this.initMemSize(properties); |
|
|
|
|
|
|
|
return new PropertiesPropertySource(resource.getFilename(), properties); |
|
|
|
return new PropertiesPropertySource(resource.getFilename(), properties); |
|
|
|
} catch (IOException ex) { |
|
|
|
} catch (IOException ex) { |
|
|
|
throw new IllegalStateException("加载配置文件失败" + resource, ex); |
|
|
|
throw new IllegalStateException("加载配置文件失败" + resource, ex); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
private void initMemSize(Properties properties) { |
|
|
|
|
|
|
|
String maxFileSizeKey = "spring.servlet.multipart.max-file-size"; |
|
|
|
|
|
|
|
String maxMemSizeKey = "spring.codec.max-in-memory-size"; |
|
|
|
|
|
|
|
String defaultSize = "10MB"; |
|
|
|
|
|
|
|
String maxUploadSize = properties.getProperty(maxFileSizeKey, defaultSize); |
|
|
|
|
|
|
|
String maxBodySize = properties.getProperty(maxMemSizeKey, defaultSize); |
|
|
|
|
|
|
|
long size = Math.max(DataSize.parse(maxUploadSize).toBytes(), DataSize.parse(maxBodySize).toBytes()); |
|
|
|
|
|
|
|
properties.put(maxMemSizeKey, size); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|