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(ParserIdentifier.parserPatchTarget(path, 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 // save data defined in value node for next (later) processing, because target needs to be read
252 // always first and there is no ordering in Json input
253 deferredValue = readValueNode(in);
255 // We have a target schema node, reuse this reader without buffering the value.
256 edit.setData(readEditData(in, edit.getTargetSchemaNode(), path));
260 // FIXME: this does not look right, as it can wreck our logic
267 if (deferredValue != null) {
268 // read saved data to normalized node when target schema is already known
269 edit.setData(readEditData(new JsonReader(new StringReader(deferredValue)), edit.getTargetSchemaNode(),
275 * Parse data defined in value node and saves it to buffer.
276 * @param sb Buffer to read value node
277 * @param in JsonReader reader
278 * @throws IOException if operation fails
280 private String readValueNode(final @NonNull JsonReader in) throws IOException {
282 final StringBuilder sb = new StringBuilder().append("{\"").append(in.nextName()).append("\":");
289 while (in.hasNext()) {
290 if (in.peek() == JsonToken.STRING) {
291 sb.append('"').append(in.nextString()).append('"');
293 readValueObject(sb, in);
295 if (in.peek() != JsonToken.END_ARRAY) {
304 readValueObject(sb, in);
309 return sb.append('}').toString();
313 * Parse one value object of data and saves it to buffer.
314 * @param sb Buffer to read value object
315 * @param in JsonReader reader
316 * @throws IOException if operation fails
318 private void readValueObject(final @NonNull StringBuilder sb, final @NonNull JsonReader in) throws IOException {
319 // read simple leaf value
320 if (in.peek() == JsonToken.STRING) {
321 sb.append('"').append(in.nextString()).append('"');
328 while (in.hasNext()) {
329 sb.append('"').append(in.nextName()).append("\":");
333 sb.append('"').append(in.nextString()).append('"');
339 while (in.hasNext()) {
340 if (in.peek() == JsonToken.STRING) {
341 sb.append('"').append(in.nextString()).append('"');
343 readValueObject(sb, in);
346 if (in.peek() != JsonToken.END_ARRAY) {
355 readValueObject(sb, in);
358 if (in.peek() != JsonToken.END_OBJECT) {
368 * Read patch edit data defined in value node to NormalizedNode.
369 * @param in reader JsonReader reader
370 * @return NormalizedNode representing data
372 private static NormalizedNode readEditData(final @NonNull JsonReader in,
373 final @NonNull SchemaNode targetSchemaNode, final @NonNull InstanceIdentifierContext<?> path) {
374 final NormalizedNodeResult resultHolder = new NormalizedNodeResult();
375 final NormalizedNodeStreamWriter writer = ImmutableNormalizedNodeStreamWriter.from(resultHolder);
376 JsonParserStream.create(writer, JSONCodecFactorySupplier.RFC7951.getShared(path.getSchemaContext()),
377 SchemaInferenceStack.ofInstantiatedPath(path.getSchemaContext(), targetSchemaNode.getPath()).toInference())
380 return resultHolder.getResult();
384 * Prepare PatchEntity from PatchEdit instance when it satisfies conditions, otherwise throws exception.
385 * @param edit Instance of PatchEdit
386 * @return PatchEntity Patch entity
388 private static PatchEntity prepareEditOperation(final @NonNull PatchEdit edit) {
389 if (edit.getOperation() != null && edit.getTargetSchemaNode() != null
390 && checkDataPresence(edit.getOperation(), edit.getData() != null)) {
391 if (!edit.getOperation().isWithValue()) {
392 return new PatchEntity(edit.getId(), edit.getOperation(), edit.getTarget());
395 // for lists allow to manipulate with list items through their parent
396 final YangInstanceIdentifier targetNode;
397 if (edit.getTarget().getLastPathArgument() instanceof NodeIdentifierWithPredicates) {
398 targetNode = edit.getTarget().getParent();
400 targetNode = edit.getTarget();
403 return new PatchEntity(edit.getId(), edit.getOperation(), targetNode, edit.getData());
406 throw new RestconfDocumentedException("Error parsing input", ErrorType.PROTOCOL, ErrorTag.MALFORMED_MESSAGE);
410 * Check if data is present when operation requires it and not present when operation data is not allowed.
411 * @param operation Name of operation
412 * @param hasData Data in edit are present/not present
413 * @return true if data is present when operation requires it or if there are no data when operation does not
414 * allow it, false otherwise
416 private static boolean checkDataPresence(final @NonNull PatchEditOperation operation, final boolean hasData) {
417 return operation.isWithValue() == hasData;
421 * Helper class representing one patch edit.
423 private static final class PatchEdit {
425 private PatchEditOperation operation;
426 private YangInstanceIdentifier target;
427 private SchemaNode targetSchemaNode;
428 private NormalizedNode data;
434 void setId(final String id) {
435 this.id = requireNonNull(id);
438 PatchEditOperation getOperation() {
442 void setOperation(final PatchEditOperation operation) {
443 this.operation = requireNonNull(operation);
446 YangInstanceIdentifier getTarget() {
450 void setTarget(final YangInstanceIdentifier target) {
451 this.target = requireNonNull(target);
454 SchemaNode getTargetSchemaNode() {
455 return targetSchemaNode;
458 void setTargetSchemaNode(final SchemaNode targetSchemaNode) {
459 this.targetSchemaNode = requireNonNull(targetSchemaNode);
462 NormalizedNode getData() {
466 void setData(final NormalizedNode data) {
467 this.data = requireNonNull(data);
474 targetSchemaNode = null;