blob: 5fa3959cdd12583c82716aceef3d9c0a18d2efd8 (
plain) (
blame)
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package io.github.jshipit;
import me.tongfei.progressbar.ProgressBar;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.*;
public class BlobDownloader extends Thread {
private final String url;
private final String digest;
private final String[][] headers;
private final String tmpdir;
public BlobDownloader(String url, String digest, String[][] headers, String tmpdir) {
this.url = url;
this.digest = digest;
this.headers = headers;
this.tmpdir = tmpdir;
}
public void run() throws RuntimeException {
URL url_obj = null;
try {
url_obj = new URI(this.url).toURL();
System.out.println(url_obj.toString());
} catch(URISyntaxException | MalformedURLException e) {
System.out.println("URISyntaxException | MalformedURLException");
e.printStackTrace();
}
assert url_obj != null;
HttpURLConnection con;
try {
con = (HttpURLConnection) url_obj.openConnection();
con.setRequestMethod("GET");
for (String[] header : this.headers) {
con.setRequestProperty(header[0], header[1]);
}
con.connect();
if (con.getResponseCode() != 200) {
System.out.println("Error: " + con.getResponseCode());
throw new RuntimeException("Failed : HTTP error code : "
+ con.getResponseCode());
} else {
System.out.println("Success: " + con.getResponseCode());
}
} catch (IOException e) {
System.out.println("Failed to connect");
e.printStackTrace();
return;
}
int fileSize = con.getContentLength();
try (InputStream in = con.getInputStream();
OutputStream out = new FileOutputStream(tmpdir + "/" + digest.replace("sha256:", "") + "layer.tar")) {
byte[] buffer = new byte[4096];
int bytesRead;
try (ProgressBar pb = new ProgressBar("Download blob "+digest.replace("sha256:", ""), fileSize)) {
while ((bytesRead = in.read(buffer)) != -1) {
out.write(buffer, 0, bytesRead);
pb.stepBy(bytesRead);
}
}
} catch (IOException e) {
System.out.println("Failed to download blob "+digest);
e.printStackTrace();
}
}
}
|
