-
Notifications
You must be signed in to change notification settings - Fork 119
/
FingerprintingCopyMethod.java
114 lines (101 loc) · 3.73 KB
/
FingerprintingCopyMethod.java
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package hudson.plugins.copyartifact;
import hudson.Extension;
import hudson.FilePath;
import hudson.Util;
import hudson.model.AbstractBuild;
import hudson.model.Fingerprint;
import hudson.model.FingerprintMap;
import hudson.model.Run;
import hudson.tasks.Fingerprinter.FingerprintAction;
import hudson.util.IOException2;
import jenkins.model.Jenkins;
import java.io.IOException;
import java.security.DigestOutputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.Map;
/**
* Performs fingerprinting during the copy.
*
* This minimizes the cost of the fingerprinting as the I/O bound nature of the copy operation
* masks the cost of digest computation.
*
* @author Kohsuke Kawaguchi
*/
@Extension(ordinal=-100)
public class FingerprintingCopyMethod extends Copier {
/**
* Null if the source of the copy operation isn't {@link AbstractBuild} but some other Run type.
*/
private AbstractBuild<?,?> src;
private AbstractBuild<?,?> dst;
private final MessageDigest md5 = newMD5();
private final Map<String,String> fingerprints = new HashMap<String, String>();
@Override
public void init(Run src, AbstractBuild<?, ?> dst, FilePath srcDir, FilePath baseTargetDir) throws IOException, InterruptedException {
this.src = src instanceof AbstractBuild ? (AbstractBuild)src : null;
this.dst = dst;
fingerprints.clear();
}
private MessageDigest newMD5() {
try {
return MessageDigest.getInstance("MD5");
} catch (NoSuchAlgorithmException e) {
throw new AssertionError(e); // impossible
}
}
@Override
public int copyAll(FilePath srcDir, String filter, FilePath targetDir, boolean fingerprintArtifacts) throws IOException, InterruptedException {
targetDir.mkdirs(); // Create target if needed
FilePath[] list = srcDir.list(filter);
for (FilePath file : list) {
String tail = file.getRemote().substring(srcDir.getRemote().length());
if (tail.startsWith("\\") || tail.startsWith("/"))
tail = tail.substring(1);
copyOne(file, new FilePath(targetDir, tail), fingerprintArtifacts);
}
return list.length;
}
@Override
public void copyOne(FilePath s, FilePath d, boolean fingerprintArtifacts) throws IOException, InterruptedException {
try {
md5.reset();
DigestOutputStream out =new DigestOutputStream(d.write(),md5);
try {
s.copyTo(out);
} finally {
out.close();
}
d.chmod(s.mode());
d.touch(s.lastModified());
String digest = Util.toHexString(md5.digest());
if (fingerprintArtifacts) {
FingerprintMap map = Jenkins.getInstance().getFingerprintMap();
Fingerprint f = map.getOrCreate(src, s.getName(), digest);
if (src!=null) {
f.add((AbstractBuild)src);
}
f.add(dst);
fingerprints.put(s.getName(), digest);
}
} catch (IOException e) {
throw new IOException2("Failed to copy "+s+" to "+d,e);
}
}
@Override
public void end() {
// add action
for (AbstractBuild r : new AbstractBuild[]{src,dst}) {
if (r == null)
continue;
FingerprintAction fa = r.getAction(FingerprintAction.class);
if (fa != null) fa.add(fingerprints);
else r.getActions().add(new FingerprintAction(r, fingerprints));
}
}
@Override
public Copier clone() {
return new FingerprintingCopyMethod();
}
}