blob: 72bfa2380946c136716151976b0cdee609513c37 (
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.deployment;
import com.yahoo.config.application.api.ValidationId;
import com.yahoo.config.provision.Environment;
import com.yahoo.vespa.hosted.controller.application.ApplicationPackage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.time.Duration;
import java.time.Instant;
import java.util.Arrays;
import java.util.Date;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
* A builder that builds application packages for testing purposes.
*
* @author mpolden
*/
public class ApplicationPackageBuilder {
private String upgradePolicy = null;
private Environment environment = Environment.prod;
private final StringBuilder environmentBody = new StringBuilder();
private final StringBuilder validationOverridesBody = new StringBuilder();
private final StringBuilder blockChange = new StringBuilder();
private String searchDefinition = "search test { }";
public ApplicationPackageBuilder upgradePolicy(String upgradePolicy) {
this.upgradePolicy = upgradePolicy;
return this;
}
public ApplicationPackageBuilder environment(Environment environment) {
this.environment = environment;
return this;
}
public ApplicationPackageBuilder region(String regionName) {
environmentBody.append(" <region active='true'>");
environmentBody.append(regionName);
environmentBody.append("</region>\n");
return this;
}
public ApplicationPackageBuilder parallel(String... regionName) {
environmentBody.append(" <parallel>\n");
Arrays.stream(regionName).forEach(this::region);
environmentBody.append(" </parallel>\n");
return this;
}
public ApplicationPackageBuilder delay(Duration delay) {
environmentBody.append(" <delay seconds='");
environmentBody.append(delay.getSeconds());
environmentBody.append("'/>\n");
return this;
}
public ApplicationPackageBuilder blockChange(boolean revision, boolean version,
String daySpec, String hourSpec, String zoneSpec) {
blockChange.append(" <block-change");
blockChange.append(" revision='").append(revision).append("'");
blockChange.append(" version='").append(version).append("'");
blockChange.append(" days='").append(daySpec).append("'");
blockChange.append(" hours='").append(hourSpec).append("'");
blockChange.append(" time-zone='").append(zoneSpec).append("'");
blockChange.append("/>\n");
return this;
}
public ApplicationPackageBuilder allow(ValidationId validationId) {
validationOverridesBody.append(" <allow until='");
validationOverridesBody.append(asIso8601Date(Instant.now().plus(Duration.ofDays(29))));
validationOverridesBody.append("'>");
validationOverridesBody.append(validationId.value());
validationOverridesBody.append("</allow>\n");
return this;
}
/** Sets the content of the search definition test.sd */
public ApplicationPackageBuilder searchDefinition(String testSearchDefinition) {
this.searchDefinition = testSearchDefinition;
return this;
}
private byte[] deploymentSpec() {
StringBuilder xml = new StringBuilder("<deployment version='1.0'>\n");
if (upgradePolicy != null) {
xml.append("<upgrade policy='");
xml.append(upgradePolicy);
xml.append("'/>\n");
}
xml.append(blockChange);
xml.append(" <");
xml.append(environment.value());
xml.append(">\n");
xml.append(environmentBody);
xml.append(" </");
xml.append(environment.value());
xml.append(">\n</deployment>");
return xml.toString().getBytes(StandardCharsets.UTF_8);
}
private byte[] validationOverrides() {
String xml = "<validation-overrides version='1.0'>\n" +
validationOverridesBody +
"</validation-overrides>\n";
return xml.getBytes(StandardCharsets.UTF_8);
}
private byte[] searchDefinition() {
return searchDefinition.getBytes(StandardCharsets.UTF_8);
}
public ApplicationPackage build() {
ByteArrayOutputStream zip = new ByteArrayOutputStream();
ZipOutputStream out = new ZipOutputStream(zip);
try {
out.putNextEntry(new ZipEntry("deployment.xml"));
out.write(deploymentSpec());
out.closeEntry();
out.putNextEntry(new ZipEntry("validation-overrides.xml"));
out.write(validationOverrides());
out.closeEntry();
out.putNextEntry(new ZipEntry("search-definitions/test.sd"));
out.write(searchDefinition());
out.closeEntry();
} catch (IOException e) {
throw new UncheckedIOException(e);
} finally {
try {
out.close();
} catch (IOException ignored) {}
}
return new ApplicationPackage(zip.toByteArray());
}
private static String asIso8601Date(Instant instant) {
return new SimpleDateFormat("yyyy-MM-dd").format(Date.from(instant));
}
}
|
