forked from fabric8io/kubernetes-client
-
Notifications
You must be signed in to change notification settings - Fork 0
/
PatchHandler.java
123 lines (109 loc) · 5.03 KB
/
PatchHandler.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
115
116
117
118
119
120
121
122
123
/**
* Copyright (C) 2015 Red Hat, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.fabric8.kubernetes.client.server.mock.crud;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import io.fabric8.kubernetes.client.dsl.base.PatchType;
import io.fabric8.kubernetes.client.utils.Serialization;
import io.fabric8.mockwebserver.crud.AttributeSet;
import io.fabric8.zjsonpatch.JsonPatch;
import okhttp3.MediaType;
import okhttp3.mockwebserver.MockResponse;
import java.net.HttpURLConnection;
import java.util.Iterator;
import java.util.Map;
import java.util.Objects;
import static io.fabric8.kubernetes.client.server.mock.crud.KubernetesCrudDispatcherHandler.isStatusPath;
import static io.fabric8.kubernetes.client.server.mock.crud.KubernetesCrudDispatcherHandler.setStatus;
import static java.net.HttpURLConnection.HTTP_ACCEPTED;
import static java.net.HttpURLConnection.HTTP_UNSUPPORTED_TYPE;
public class PatchHandler implements KubernetesCrudDispatcherHandler {
private static final String PATH = "path";
private final KubernetesCrudPersistence persistence;
public PatchHandler(KubernetesCrudPersistence persistence) {
this.persistence = persistence;
}
@Override
public MockResponse handle(String path, String contentType, String requestBody) throws KubernetesCrudDispatcherException {
final AttributeSet query = persistence.getKey(path);
final Map.Entry<AttributeSet, String> currentResourceEntry = persistence.findResource(query);
if (currentResourceEntry == null) {
return new MockResponse().setResponseCode(HttpURLConnection.HTTP_NOT_FOUND);
}
final JsonNode currentResource = persistence.asNode(currentResourceEntry);
final JsonNode patch = persistence.asNode(requestBody);
// Read the patch and create a complete resource (either from the body or by applying the PATCH operations)
final JsonNode fullPatch;
if (getMergeType(contentType) == PatchType.JSON) {
fullPatch = JsonPatch.apply(patch, initPaths(currentResource.deepCopy(), patch));
} else {
fullPatch = persistence.merge(currentResource, requestBody);
}
validatePath(query, fullPatch);
final JsonNode updatedResource;
if (isStatusPath(path)) {
updatedResource = currentResource.deepCopy();
setStatus(updatedResource, fullPatch.get(STATUS));
} else {
updatedResource = fullPatch;
// preserve original status (PATCH requests to the custom resource ignore changes to the status stanza)
if (persistence.isStatusSubresourceEnabledForResource(path)) {
setStatus(updatedResource, currentResource.path(STATUS));
}
}
persistence.preserveMetadata(currentResource, updatedResource);
if (!isStatusPath(path)) {
persistence.touchGeneration(currentResource, updatedResource);
}
persistence.touchResourceVersion(currentResource, updatedResource);
final String updatedAsString = Serialization.asJson(updatedResource);
validateRequestBody(updatedAsString);
persistence.processEvent(path, query, currentResourceEntry.getKey(), updatedAsString);
return new MockResponse().setResponseCode(HTTP_ACCEPTED).setBody(updatedAsString);
}
private PatchType getMergeType(String contentType) throws KubernetesCrudDispatcherException {
final PatchType mergeType;
if (contentType == null) {
mergeType = PatchType.JSON;
} else {
final String subtype = Objects.requireNonNull(MediaType.parse(contentType)).subtype();
if (subtype.equals(MediaType.get(PatchType.JSON.getContentType()).subtype())) {
mergeType = PatchType.JSON;
} else if (subtype.equals(MediaType.get(PatchType.JSON_MERGE.getContentType()).subtype())) {
mergeType = PatchType.JSON_MERGE;
} else {
throw new KubernetesCrudDispatcherException("Unsupported Media Type", HTTP_UNSUPPORTED_TYPE);
}
}
return mergeType;
}
// Ensure resource contains all paths affected by the provided patch
private JsonNode initPaths(JsonNode resource, JsonNode patch) {
for (Iterator<JsonNode> it = patch.elements(); it.hasNext();) {
final String fullPath = it.next().get(PATH).asText();
final String[] paths = fullPath.replaceAll("^/", "").split("/");
JsonNode node = resource;
for (int p = 0; p < paths.length - 1; p++) {
final String path = paths[p];
if (node.get(path) == null && node.isObject()) {
((ObjectNode) node).set(path, ((ObjectNode) node).objectNode());
}
node = node.path(path);
}
}
return resource;
}
}