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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.standalone;
import com.yahoo.config.FileReference;
import com.yahoo.config.application.api.FileRegistry;
import com.yahoo.filedistribution.fileacquirer.FileAcquirer;
import net.jpountz.lz4.LZ4FrameOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.file.Path;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
/**
* FileAcquirer and FileRegistry working on a local directory.
*
* @author Tony Vaagenes
* @author ollivir
*/
public class LocalFileDb implements FileAcquirer, FileRegistry {
private final Map<FileReference, File> fileReferenceToFile = new HashMap<>();
private final Path appPath;
public LocalFileDb(Path appPath) {
this.appPath = appPath;
}
/* FileAcquirer overrides */
@Override
public File waitFor(FileReference reference, long l, TimeUnit timeUnit) {
synchronized (this) {
File file = fileReferenceToFile.get(reference);
if (file == null) {
return new File(reference.value()); // Downloaded file reference: Will (hopefully) be resolved client side
}
return file;
}
}
@Override
public void shutdown() {
}
/* FileRegistry overrides */
@Override
public FileReference addFile(String relativePath) {
File file = appPath.resolve(relativePath).toFile();
Path relative = appPath.relativize(file.toPath()).normalize();
if (relative.isAbsolute() || relative.startsWith(".."))
throw new IllegalArgumentException(file + " is not a descendant of " + appPath);
if (!file.exists()) {
throw new RuntimeException("The file does not exist: " + file.getPath());
}
FileReference fileReference = new FileReference("LocalFileDb:" + relativePath);
fileReferenceToFile.put(fileReference, file);
return fileReference;
}
@Override
public List<Entry> export() {
return fileReferenceToFile.entrySet().stream().map(entry -> new Entry(entry.getValue().getPath(), entry.getKey()))
.toList();
}
@Override
public FileReference addUri(String uri) {
throw new RuntimeException("addUri(String uri) is not implemented here.");
}
@Override
public FileReference addBlob(String name, ByteBuffer blob) {
writeBlob(blob, name);
File file = appPath.resolve(name).toFile();
FileReference fileReference = new FileReference("LocalFileDb:" + name);
fileReferenceToFile.put(fileReference, file);
return fileReference;
}
private void writeBlob(ByteBuffer blob, String relativePath) {
try (FileOutputStream fos = new FileOutputStream(new File(appPath.toFile(), relativePath))) {
if (relativePath.endsWith(".lz4")) {
LZ4FrameOutputStream lz4 = new LZ4FrameOutputStream(fos);
lz4.write(blob.array(), blob.arrayOffset(), blob.remaining());
lz4.close();
} else {
fos.write(blob.array(), blob.arrayOffset(), blob.remaining());
}
} catch (IOException e) {
throw new IllegalArgumentException("Failed writing temp file", e);
}
}
}
|
