| | |
| | | import java.util.Date; |
| | | import java.util.HashMap; |
| | | import java.util.Map; |
| | | import java.util.concurrent.ConcurrentHashMap; |
| | | |
| | | import com.fasterxml.jackson.databind.ObjectMapper; |
| | | |
| | | @Service |
| | |
| | | } |
| | | } |
| | | |
| | | public String doGet(String pid){ |
| | | // 从配置中获取 API URL,并拼接 key |
| | | String apiUrl = PropertiesUtil.getProperty("JS_IN_HTTP_URL") + "stock?key=" + PropertiesUtil.getProperty("JS_IN_KEY"); |
| | | String result = null; |
| | | private static class CacheEntry { |
| | | String result; |
| | | long lastExecuteTime; |
| | | } |
| | | |
| | | private static final Map<String, CacheEntry> cacheMap = new ConcurrentHashMap<>(); |
| | | private static final long MIN_INTERVAL_MS = 3000; |
| | | private static final Object globalLock = new Object(); |
| | | |
| | | public static String doGet(String pid) { |
| | | long currentTime = System.currentTimeMillis(); |
| | | |
| | | // 1. 先检查该pid是否有缓存 |
| | | CacheEntry entry = cacheMap.get(pid); |
| | | if (entry != null && (currentTime - entry.lastExecuteTime) < MIN_INTERVAL_MS) { |
| | | return entry.result; |
| | | } |
| | | |
| | | // 2. 使用双重检查锁定更新缓存 |
| | | synchronized (getLockForPid(pid)) { |
| | | currentTime = System.currentTimeMillis(); |
| | | entry = cacheMap.get(pid); |
| | | |
| | | if (entry != null && (currentTime - entry.lastExecuteTime) < MIN_INTERVAL_MS) { |
| | | return entry.result; |
| | | } |
| | | |
| | | // 3. 执行实际请求 |
| | | String newResult = doActualPost(pid); |
| | | |
| | | // 4. 更新该pid的缓存 |
| | | CacheEntry newEntry = new CacheEntry(); |
| | | newEntry.result = newResult; |
| | | newEntry.lastExecuteTime = System.currentTimeMillis(); |
| | | cacheMap.put(pid, newEntry); |
| | | |
| | | return newResult; |
| | | } |
| | | } |
| | | |
| | | // 为每个pid提供独立的锁对象,避免不同pid间的阻塞 |
| | | private static final Map<String, Object> pidLocks = new ConcurrentHashMap<>(); |
| | | |
| | | private static Object getLockForPid(String pid) { |
| | | return pidLocks.computeIfAbsent(pid, k -> new Object()); |
| | | } |
| | | |
| | | |
| | | private static String doActualPost(String pid) { |
| | | String apiUrl = PropertiesUtil.getProperty("JS_IN_HTTP_URL") + |
| | | "stock?key=" + PropertiesUtil.getProperty("JS_IN_KEY"); |
| | | |
| | | try { |
| | | URL url = new URL(apiUrl); |
| | | HttpURLConnection connection = (HttpURLConnection) url.openConnection(); |
| | | |
| | | // 设置请求方法为 POST |
| | | connection.setRequestMethod("POST"); |
| | | // 设置请求头 |
| | | connection.setRequestProperty("Content-Type", "application/x-www-form-urlencoded"); |
| | | connection.setDoOutput(true); // 允许向连接输出 |
| | | connection.setDoOutput(true); |
| | | connection.setConnectTimeout(5000); |
| | | connection.setReadTimeout(10000); |
| | | |
| | | // 构建 POST 数据 |
| | | String postData = "pid=" + pid; |
| | | |
| | | // 发送 POST 请求 |
| | | try (OutputStream os = connection.getOutputStream()) { |
| | | byte[] input = postData.getBytes("utf-8"); |
| | | os.write(input, 0, input.length); |
| | | } |
| | | |
| | | // 读取响应 |
| | | BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream())); |
| | | BufferedReader in = new BufferedReader( |
| | | new InputStreamReader(connection.getInputStream())); |
| | | String inputLine; |
| | | StringBuffer response = new StringBuffer(); |
| | | StringBuilder response = new StringBuilder(); |
| | | |
| | | while ((inputLine = in.readLine()) != null) { |
| | | response.append(inputLine); |
| | | } |
| | | in.close(); |
| | | result = response.toString(); |
| | | |
| | | return response.toString(); |
| | | |
| | | } catch (Exception e) { |
| | | e.printStackTrace(); |
| | | // 如果请求失败,返回该pid的旧缓存(如果有) |
| | | CacheEntry oldEntry = cacheMap.get(pid); |
| | | if (oldEntry != null && oldEntry.result != null) { |
| | | return oldEntry.result; |
| | | } |
| | | return "{\"error\":\"请求失败:" + e.getMessage() + "\"}"; |
| | | } |
| | | return result; |
| | | } |
| | | |
| | | @Override |