blob: 2b77db119c8698c106ea9caa6dd340160aece93c (
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.model.application.provider;
import com.yahoo.config.FileReference;
import com.yahoo.config.application.api.FileRegistry;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.Reader;
import java.util.*;
import java.util.regex.Pattern;
/**
* Registry of files added earlier (i.e. during deployment)
*
* @author tonytv
*/
public class PreGeneratedFileRegistry implements FileRegistry {
private final String fileSourceHost;
private final Map<String, String> path2Hash = new LinkedHashMap<>();
private static String entryDelimiter = "\t";
private static Pattern entryDelimiterPattern = Pattern.compile(entryDelimiter, Pattern.LITERAL);
private PreGeneratedFileRegistry(Reader readerArg) {
BufferedReader reader = new BufferedReader(readerArg);
try {
fileSourceHost = reader.readLine();
if (fileSourceHost == null)
throw new RuntimeException("Error while reading pre generated file registry");
String line;
while ((line = reader.readLine()) != null) {
addFromLine(line);
}
} catch(IOException e) {
throw new RuntimeException("Error while reading pre generated file registry", e);
} finally {
try {
reader.close();
} catch(IOException e) {}
}
}
private void addFromLine(String line) {
String[] parts = entryDelimiterPattern.split(line);
addEntry(parts[0], parts[1]);
}
private void addEntry(String relativePath, String hash) {
path2Hash.put(relativePath, hash);
}
public static String exportRegistry(FileRegistry registry) {
List<FileRegistry.Entry> entries = registry.export();
StringBuilder builder = new StringBuilder();
builder.append(registry.fileSourceHost()).append('\n');
for (FileRegistry.Entry entry : entries) {
builder.append(entry.relativePath).append(entryDelimiter).append(entry.reference.value()).
append('\n');
}
return builder.toString();
}
public static PreGeneratedFileRegistry importRegistry(Reader reader) {
return new PreGeneratedFileRegistry(reader);
}
public FileReference addFile(String relativePath) {
return new FileReference(path2Hash.get(relativePath));
}
@Override
public FileReference addUri(String uri) {
throw new IllegalArgumentException("FileReference addUri(String uri) is not implemented for " + getClass().getCanonicalName());
}
@Override
public String fileSourceHost() {
return fileSourceHost;
}
public Set<String> getPaths() {
return path2Hash.keySet();
}
@Override
public List<Entry> export() {
List<Entry> entries = new ArrayList<>();
for (Map.Entry<String, String> entry : path2Hash.entrySet()) {
entries.add(new Entry(entry.getKey(), new FileReference(entry.getValue())));
}
return entries;
}
}
|
