2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.restconf.nb.rfc8040.jersey.providers.patch;
10 import static com.google.common.base.Preconditions.checkArgument;
11 import static com.google.common.base.Verify.verify;
12 import static java.util.Objects.requireNonNull;
14 import com.google.common.base.Throwables;
15 import com.google.common.collect.ImmutableList;
16 import com.google.gson.stream.JsonReader;
17 import com.google.gson.stream.JsonToken;
18 import java.io.IOException;
19 import java.io.InputStream;
20 import java.io.InputStreamReader;
21 import java.io.StringReader;
22 import java.nio.charset.StandardCharsets;
23 import java.util.ArrayList;
24 import java.util.List;
25 import java.util.Locale;
26 import java.util.Optional;
27 import java.util.concurrent.atomic.AtomicReference;
28 import javax.ws.rs.Consumes;
29 import javax.ws.rs.WebApplicationException;
30 import javax.ws.rs.ext.Provider;
31 import org.eclipse.jdt.annotation.NonNull;
32 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
33 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
34 import org.opendaylight.restconf.common.errors.RestconfDocumentedException;
35 import org.opendaylight.restconf.common.errors.RestconfError.ErrorTag;
36 import org.opendaylight.restconf.common.patch.PatchContext;
37 import org.opendaylight.restconf.common.patch.PatchEditOperation;
38 import org.opendaylight.restconf.common.patch.PatchEntity;
39 import org.opendaylight.restconf.nb.rfc8040.MediaTypes;
40 import org.opendaylight.restconf.nb.rfc8040.codecs.StringModuleInstanceIdentifierCodec;
41 import org.opendaylight.restconf.nb.rfc8040.handlers.SchemaContextHandler;
42 import org.opendaylight.restconf.nb.rfc8040.utils.parser.ParserIdentifier;
43 import org.opendaylight.yangtools.yang.common.ErrorType;
44 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
45 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
46 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
47 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
48 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactorySupplier;
49 import org.opendaylight.yangtools.yang.data.codec.gson.JsonParserStream;
50 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
51 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
52 import org.opendaylight.yangtools.yang.data.impl.schema.ResultAlreadySetException;
53 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
54 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
55 import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
56 import org.slf4j.Logger;
57 import org.slf4j.LoggerFactory;
60 @Consumes(MediaTypes.APPLICATION_YANG_PATCH_JSON)
61 public class JsonPatchBodyReader extends AbstractPatchBodyReader {
62 private static final Logger LOG = LoggerFactory.getLogger(JsonPatchBodyReader.class);
64 public JsonPatchBodyReader(final SchemaContextHandler schemaContextHandler,
65 final DOMMountPointService mountPointService) {
66 super(schemaContextHandler, mountPointService);
69 @SuppressWarnings("checkstyle:IllegalCatch")
71 protected PatchContext readBody(final InstanceIdentifierContext<?> path, final InputStream entityStream)
72 throws WebApplicationException {
74 return readFrom(path, entityStream);
75 } catch (final Exception e) {
76 throw propagateExceptionAs(e);
80 private PatchContext readFrom(final InstanceIdentifierContext<?> path, final InputStream entityStream)
82 final JsonReader jsonReader = new JsonReader(new InputStreamReader(entityStream, StandardCharsets.UTF_8));
83 AtomicReference<String> patchId = new AtomicReference<>();
84 final List<PatchEntity> resultList = read(jsonReader, path, patchId);
87 return new PatchContext(path, resultList, patchId.get());
90 @SuppressWarnings("checkstyle:IllegalCatch")
91 public PatchContext readFrom(final String uriPath, final InputStream entityStream) throws
92 RestconfDocumentedException {
95 ParserIdentifier.toInstanceIdentifier(uriPath, getSchemaContext(),
96 Optional.ofNullable(getMountPointService())), entityStream);
97 } catch (final Exception e) {
98 propagateExceptionAs(e);
103 private static RuntimeException propagateExceptionAs(final Exception exception) throws RestconfDocumentedException {
104 Throwables.throwIfInstanceOf(exception, RestconfDocumentedException.class);
105 LOG.debug("Error parsing json input", exception);
107 if (exception instanceof ResultAlreadySetException) {
108 throw new RestconfDocumentedException("Error parsing json input: Failed to create new parse result data. ");
111 RestconfDocumentedException.throwIfYangError(exception);
112 throw new RestconfDocumentedException("Error parsing json input: " + exception.getMessage(), ErrorType.PROTOCOL,
113 ErrorTag.MALFORMED_MESSAGE, exception);
116 private List<PatchEntity> read(final JsonReader in, final InstanceIdentifierContext<?> path,
117 final AtomicReference<String> patchId) throws IOException {
118 final List<PatchEntity> resultCollection = new ArrayList<>();
119 final StringModuleInstanceIdentifierCodec codec = new StringModuleInstanceIdentifierCodec(
120 path.getSchemaContext());
121 final JsonPatchBodyReader.PatchEdit edit = new JsonPatchBodyReader.PatchEdit();
123 while (in.hasNext()) {
130 Boolean.toString(in.nextBoolean());
144 parseByName(in.nextName(), edit, in, path, codec, resultCollection, patchId);
158 return ImmutableList.copyOf(resultCollection);
162 * Switch value of parsed JsonToken.NAME and read edit definition or patch id.
164 * @param name value of token
165 * @param edit PatchEdit instance
166 * @param in JsonReader reader
167 * @param path InstanceIdentifierContext context
168 * @param codec Draft11StringModuleInstanceIdentifierCodec codec
169 * @param resultCollection collection of parsed edits
170 * @throws IOException if operation fails
172 private void parseByName(final @NonNull String name, final @NonNull PatchEdit edit,
173 final @NonNull JsonReader in, final @NonNull InstanceIdentifierContext<?> path,
174 final @NonNull StringModuleInstanceIdentifierCodec codec,
175 final @NonNull List<PatchEntity> resultCollection,
176 final @NonNull AtomicReference<String> patchId) throws IOException {
179 if (in.peek() == JsonToken.BEGIN_ARRAY) {
182 while (in.hasNext()) {
183 readEditDefinition(edit, in, path, codec);
184 resultCollection.add(prepareEditOperation(edit));
190 readEditDefinition(edit, in, path, codec);
191 resultCollection.add(prepareEditOperation(edit));
197 patchId.set(in.nextString());
205 * Read one patch edit object from Json input.
207 * @param edit PatchEdit instance to be filled with read data
208 * @param in JsonReader reader
209 * @param path InstanceIdentifierContext path context
210 * @param codec Draft11StringModuleInstanceIdentifierCodec codec
211 * @throws IOException if operation fails
213 private void readEditDefinition(final @NonNull PatchEdit edit, final @NonNull JsonReader in,
214 final @NonNull InstanceIdentifierContext<?> path,
215 final @NonNull StringModuleInstanceIdentifierCodec codec) throws IOException {
216 String deferredValue = null;
219 while (in.hasNext()) {
220 final String editDefinition = in.nextName();
221 switch (editDefinition) {
223 edit.setId(in.nextString());
226 edit.setOperation(PatchEditOperation.valueOf(in.nextString().toUpperCase(Locale.ROOT)));
229 // target can be specified completely in request URI
230 final String target = in.nextString();
231 if (target.equals("/")) {
232 edit.setTarget(path.getInstanceIdentifier());
233 edit.setTargetSchemaNode(path.getSchemaContext());
235 edit.setTarget(codec.deserialize(codec.serialize(path.getInstanceIdentifier()).concat(target)));
237 final EffectiveStatement<?, ?> parentStmt = SchemaInferenceStack.ofInstantiatedPath(
238 path.getSchemaContext(),
239 codec.getDataContextTree().findChild(edit.getTarget()).orElseThrow().getDataSchemaNode()
240 .getPath().getParent())
242 verify(parentStmt instanceof SchemaNode, "Unexpected parent %s", parentStmt);
243 edit.setTargetSchemaNode((SchemaNode) parentStmt);
248 checkArgument(edit.getData() == null && deferredValue == null, "Multiple value entries found");
250 if (edit.getTargetSchemaNode() == null) {
251 final StringBuilder sb = new StringBuilder();
253 // save data defined in value node for next (later) processing, because target needs to be read
254 // always first and there is no ordering in Json input
255 readValueNode(sb, in);
256 deferredValue = sb.toString();
258 // We have a target schema node, reuse this reader without buffering the value.
259 edit.setData(readEditData(in, edit.getTargetSchemaNode(), path));
263 // FIXME: this does not look right, as it can wreck our logic
270 if (deferredValue != null) {
271 // read saved data to normalized node when target schema is already known
272 edit.setData(readEditData(new JsonReader(new StringReader(deferredValue)), edit.getTargetSchemaNode(),
278 * Parse data defined in value node and saves it to buffer.
279 * @param sb Buffer to read value node
280 * @param in JsonReader reader
281 * @throws IOException if operation fails
283 private void readValueNode(final @NonNull StringBuilder sb, final @NonNull JsonReader in) throws IOException {
286 sb.append("{\"").append(in.nextName()).append("\":");
293 while (in.hasNext()) {
294 if (in.peek() == JsonToken.STRING) {
295 sb.append('"').append(in.nextString()).append('"');
297 readValueObject(sb, in);
299 if (in.peek() != JsonToken.END_ARRAY) {
308 readValueObject(sb, in);
317 * Parse one value object of data and saves it to buffer.
318 * @param sb Buffer to read value object
319 * @param in JsonReader reader
320 * @throws IOException if operation fails
322 private void readValueObject(final @NonNull StringBuilder sb, final @NonNull JsonReader in) throws IOException {
323 // read simple leaf value
324 if (in.peek() == JsonToken.STRING) {
325 sb.append('"').append(in.nextString()).append('"');
332 while (in.hasNext()) {
333 sb.append('"').append(in.nextName()).append("\":");
337 sb.append('"').append(in.nextString()).append('"');
343 while (in.hasNext()) {
344 if (in.peek() == JsonToken.STRING) {
345 sb.append('"').append(in.nextString()).append('"');
347 readValueObject(sb, in);
350 if (in.peek() != JsonToken.END_ARRAY) {
359 readValueObject(sb, in);
362 if (in.peek() != JsonToken.END_OBJECT) {
372 * Read patch edit data defined in value node to NormalizedNode.
373 * @param in reader JsonReader reader
374 * @return NormalizedNode representing data
376 private static NormalizedNode readEditData(final @NonNull JsonReader in,
377 final @NonNull SchemaNode targetSchemaNode, final @NonNull InstanceIdentifierContext<?> path) {
378 final NormalizedNodeResult resultHolder = new NormalizedNodeResult();
379 final NormalizedNodeStreamWriter writer = ImmutableNormalizedNodeStreamWriter.from(resultHolder);
380 JsonParserStream.create(writer, JSONCodecFactorySupplier.RFC7951.getShared(path.getSchemaContext()),
381 SchemaInferenceStack.ofInstantiatedPath(path.getSchemaContext(), targetSchemaNode.getPath()).toInference())
384 return resultHolder.getResult();
388 * Prepare PatchEntity from PatchEdit instance when it satisfies conditions, otherwise throws exception.
389 * @param edit Instance of PatchEdit
390 * @return PatchEntity Patch entity
392 private static PatchEntity prepareEditOperation(final @NonNull PatchEdit edit) {
393 if (edit.getOperation() != null && edit.getTargetSchemaNode() != null
394 && checkDataPresence(edit.getOperation(), edit.getData() != null)) {
395 if (!edit.getOperation().isWithValue()) {
396 return new PatchEntity(edit.getId(), edit.getOperation(), edit.getTarget());
399 // for lists allow to manipulate with list items through their parent
400 final YangInstanceIdentifier targetNode;
401 if (edit.getTarget().getLastPathArgument() instanceof NodeIdentifierWithPredicates) {
402 targetNode = edit.getTarget().getParent();
404 targetNode = edit.getTarget();
407 return new PatchEntity(edit.getId(), edit.getOperation(), targetNode, edit.getData());
410 throw new RestconfDocumentedException("Error parsing input", ErrorType.PROTOCOL, ErrorTag.MALFORMED_MESSAGE);
414 * Check if data is present when operation requires it and not present when operation data is not allowed.
415 * @param operation Name of operation
416 * @param hasData Data in edit are present/not present
417 * @return true if data is present when operation requires it or if there are no data when operation does not
418 * allow it, false otherwise
420 private static boolean checkDataPresence(final @NonNull PatchEditOperation operation, final boolean hasData) {
421 return operation.isWithValue() == hasData;
425 * Helper class representing one patch edit.
427 private static final class PatchEdit {
429 private PatchEditOperation operation;
430 private YangInstanceIdentifier target;
431 private SchemaNode targetSchemaNode;
432 private NormalizedNode data;
438 void setId(final String id) {
439 this.id = requireNonNull(id);
442 PatchEditOperation getOperation() {
446 void setOperation(final PatchEditOperation operation) {
447 this.operation = requireNonNull(operation);
450 YangInstanceIdentifier getTarget() {
454 void setTarget(final YangInstanceIdentifier target) {
455 this.target = requireNonNull(target);
458 SchemaNode getTargetSchemaNode() {
459 return targetSchemaNode;
462 void setTargetSchemaNode(final SchemaNode targetSchemaNode) {
463 this.targetSchemaNode = requireNonNull(targetSchemaNode);
466 NormalizedNode getData() {
470 void setData(final NormalizedNode data) {
471 this.data = requireNonNull(data);
478 targetSchemaNode = null;