|
8 | 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | 9 | */ |
10 | 10 |
|
11 | | - |
12 | 11 | package com.altinity.ice.cli.internal.http; |
13 | 12 |
|
14 | 13 | import java.net.URI; |
|
17 | 16 | import org.json.*; |
18 | 17 |
|
19 | 18 | public final class MinioWildcard { |
20 | | - private static final HttpClient client = HttpClient.newHttpClient(); |
21 | | - |
22 | | - public static List<String> listHTTPWildcard(String urlWithStar) throws Exception { |
23 | | - URI uri = URI.create(urlWithStar); |
24 | | - String[] parts = uri.getPath().split("/", 3); |
25 | | - if (parts.length < 2) |
26 | | - throw new IllegalArgumentException("Bad MinIO URL: " + urlWithStar); |
27 | | - |
28 | | - String bucket = parts[1]; |
29 | | - String prefix = parts.length == 3 ? parts[2].replaceAll("\\*", "") : ""; |
30 | | - String endpoint = uri.getScheme() + "://" + uri.getHost() + ":" + uri.getPort(); |
31 | | - |
32 | | - String listUrl = endpoint + "/" + bucket + "?list-type=2&prefix=" + prefix; |
33 | | - HttpRequest req = HttpRequest.newBuilder(URI.create(listUrl)).GET().build(); |
34 | | - HttpResponse<String> resp = client.send(req, HttpResponse.BodyHandlers.ofString()); |
35 | | - if (resp.statusCode() != 200) |
36 | | - throw new RuntimeException("MinIO listObjects error: " + resp.statusCode()); |
37 | | - |
38 | | - JSONObject xml = org.json.XML.toJSONObject(resp.body()); |
39 | | - List<String> files = new ArrayList<>(); |
40 | | - var result = xml.getJSONObject("ListBucketResult"); |
41 | | - var contents = result.optJSONArray("Contents"); |
42 | | - if (contents != null) { |
43 | | - for (int i = 0; i < contents.length(); i++) { |
44 | | - String key = contents.getJSONObject(i).getString("Key"); |
45 | | - if (key.endsWith(".parquet")) |
46 | | - files.add(endpoint + "/" + bucket + "/" + key); |
47 | | - } |
48 | | - } else { |
49 | | - var single = result.optJSONObject("Contents"); |
50 | | - if (single != null) |
51 | | - files.add(endpoint + "/" + bucket + "/" + single.getString("Key")); |
52 | | - } |
53 | | - return files; |
| 19 | + private static final HttpClient client = HttpClient.newHttpClient(); |
| 20 | + |
| 21 | + public static List<String> listHTTPWildcard(String urlWithStar) throws Exception { |
| 22 | + URI uri = URI.create(urlWithStar); |
| 23 | + String[] parts = uri.getPath().split("/", 3); |
| 24 | + if (parts.length < 2) throw new IllegalArgumentException("Bad MinIO URL: " + urlWithStar); |
| 25 | + |
| 26 | + String bucket = parts[1]; |
| 27 | + String prefix = parts.length == 3 ? parts[2].replaceAll("\\*", "") : ""; |
| 28 | + String endpoint = uri.getScheme() + "://" + uri.getHost() + ":" + uri.getPort(); |
| 29 | + |
| 30 | + String listUrl = endpoint + "/" + bucket + "?list-type=2&prefix=" + prefix; |
| 31 | + HttpRequest req = HttpRequest.newBuilder(URI.create(listUrl)).GET().build(); |
| 32 | + HttpResponse<String> resp = client.send(req, HttpResponse.BodyHandlers.ofString()); |
| 33 | + if (resp.statusCode() != 200) |
| 34 | + throw new RuntimeException("MinIO listObjects error: " + resp.statusCode()); |
| 35 | + |
| 36 | + JSONObject xml = org.json.XML.toJSONObject(resp.body()); |
| 37 | + List<String> files = new ArrayList<>(); |
| 38 | + var result = xml.getJSONObject("ListBucketResult"); |
| 39 | + var contents = result.optJSONArray("Contents"); |
| 40 | + if (contents != null) { |
| 41 | + for (int i = 0; i < contents.length(); i++) { |
| 42 | + String key = contents.getJSONObject(i).getString("Key"); |
| 43 | + if (key.endsWith(".parquet")) files.add(endpoint + "/" + bucket + "/" + key); |
| 44 | + } |
| 45 | + } else { |
| 46 | + var single = result.optJSONObject("Contents"); |
| 47 | + if (single != null) files.add(endpoint + "/" + bucket + "/" + single.getString("Key")); |
54 | 48 | } |
| 49 | + return files; |
| 50 | + } |
55 | 51 | } |
0 commit comments