Spring短URL服务：将返回String改为返回List<String>并修复代码问题

1. 服务接口定义 #

public interface ShortUrlService {
    List<String> batchGenerateShortUrls(List<String> longUrls);
}

2. 实现类代码 #

import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@Service
public class ShortUrlServiceImpl implements ShortUrlService {

    private final RestTemplate restTemplate;
    // 本地缓存模拟，生产环境建议替换为Redis等分布式缓存
    private final Map<String, String> urlCache = new ConcurrentHashMap<>();

    // 构造注入RestTemplate，需确保容器中已配置该Bean
    public ShortUrlServiceImpl(RestTemplate restTemplate) {
        this.restTemplate = restTemplate;
    }

    @Override
    public List<String> batchGenerateShortUrls(List<String> longUrls) {
        List<String> shortUrls = new ArrayList<>();
        for (String longUrl : longUrls) {
            // 先查缓存，避免重复生成
            if (urlCache.containsKey(longUrl)) {
                shortUrls.add(urlCache.get(longUrl));
                continue;
            }
            // 生成单个短URL并缓存
            String shortUrl = generateSingleShortUrl(longUrl);
            urlCache.put(longUrl, shortUrl);
            urlCache.put(shortUrl, longUrl); // 反向映射用于后续跳转
            shortUrls.add(shortUrl);
        }
        return shortUrls;
    }

    // 单个URL的短链生成逻辑，替换原HttpClient请求
    private String generateSingleShortUrl(String longUrl) {
        String encodedUrl = URLEncoder.encode(longUrl, StandardCharsets.UTF_8);
        String requestUrl = buildRequestUrl(encodedUrl);
        requestUrl = addCacheParameter(requestUrl);
        // 使用RestTemplate发送GET请求获取短链
        return restTemplate.getForObject(requestUrl, String.class);
    }

    // 构造第三方短链服务请求地址
    private String buildRequestUrl(String encodedUrl) {
        return "https://your-short-url-service.com/api/shorten?url=" + encodedUrl;
    }

    // 添加缓存控制参数（原方法适配单个URL，批量时循环调用即可）
    private String addCacheParameter(String requestUrl) {
        return requestUrl + "&cache=1";
    }
}

3. RestTemplate配置类（必填） #

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;

@Configuration
public class RestTemplateConfig {
    @Bean
    public RestTemplate restTemplate() {
        return new RestTemplate();
    }
}

• List无法调用replace方法：原代码错误对List整体调用replace，实际要遍历List中的每个String元素，单独对单个URL执行字符替换/编码操作（如用URLEncoder.encode做标准URL编码，替代手动replace）。
• RestTemplate使用不当：不能直接对List调用RestTemplate方法，需循环处理每个长URL，单独构造请求地址，调用getForObject获取单个短链后收集到结果List中。
• 类型赋值不匹配：确保方法返回值为List<String>，遍历过程中逐个将生成的短链添加到ArrayList，避免单个String与List类型的赋值错误。
• addCacheParameter等方法适配List：无需修改这类原本处理单个URL的方法，批量场景下循环调用即可，保持方法职责单一。
• 替换HttpClient为RestTemplate：用restTemplate.getForObject替代原HttpClient的getRequest逻辑，注意先对长URL做标准编码，避免特殊字符导致请求失败。

• 异常处理：生产环境要给RestTemplate请求添加try-catch块，处理网络异常、第三方服务报错等情况，可选择跳过失败的URL或抛出自定义异常，避免单个URL失败导致整个批量请求中断。
• 缓存优化：如果是高并发场景，建议用Redis替代本地ConcurrentHashMap，保证多实例环境下的缓存一致性。
• 性能优化：可通过并行流处理批量URL，但要注意控制并发数，避免压垮第三方短链服务。

内容的提问来源于stack exchange，提问作者sticky fingers