liujie
2026-08-04 e9e42b621afdf6ac94f5327ee843b32b10a401b4
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package com.tourguide.config;
 
import com.tourguide.config.properties.StorageProperties;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
 
import java.nio.file.Paths;
 
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
    private final StorageProperties storageProperties;
 
    public WebMvcConfig(StorageProperties storageProperties) {
        this.storageProperties = storageProperties;
    }
 
    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        registerResourceHandler(registry, storageProperties.getAvatarPublicPath(),
                storageProperties.getAvatarDirectory(), "/uploads/avatars");
        registerResourceHandler(registry, storageProperties.getGuideApplicationPublicPath(),
                storageProperties.getGuideApplicationDirectory(), "/uploads/guide-applications");
    }
 
    private void registerResourceHandler(ResourceHandlerRegistry registry, String publicPath,
                                         String directory, String defaultPublicPath) {
        String normalizedPublicPath = normalizePublicPath(publicPath, defaultPublicPath);
        String resourceLocation = Paths.get(directory).toAbsolutePath().normalize().toUri().toString();
        registry.addResourceHandler(normalizedPublicPath + "/**")
                .addResourceLocations(resourceLocation.endsWith("/") ? resourceLocation : resourceLocation + "/");
    }
 
    private String normalizePublicPath(String publicPath, String defaultPublicPath) {
        String normalized = publicPath == null ? defaultPublicPath : publicPath.trim();
        if (!normalized.startsWith("/")) {
            normalized = "/" + normalized;
        }
        while (normalized.endsWith("/")) {
            normalized = normalized.substring(0, normalized.length() - 1);
        }
        return normalized;
    }
}