dom/mdsal-dom-inmemory-datastore/src/main/java/org/opendaylight/mdsal/dom/store/inmemory/ShardRootModificationContext.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8
   9 package org.opendaylight.mdsal.dom.store.inmemory;
  10
  11 import static java.util.Objects.requireNonNull;
  12
  13 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
  14 import org.opendaylight.yangtools.concepts.Identifiable;
  15 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  16 import org.opendaylight.yangtools.yang.data.api.schema.tree.CursorAwareDataTreeModification;
  17 import org.opendaylight.yangtools.yang.data.api.schema.tree.CursorAwareDataTreeSnapshot;
  18 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
  19 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModificationCursor;
  20
  21 // Non-final for mocking
  22 class ShardRootModificationContext implements Identifiable<DOMDataTreeIdentifier> {
  23
  24     private final DOMDataTreeIdentifier identifier;
  25     private final CursorAwareDataTreeSnapshot snapshot;
  26
  27     private CursorAwareDataTreeModification modification = null;
  28     private DataTreeModificationCursorAdaptor cursor = null;
  29
  30     ShardRootModificationContext(final DOMDataTreeIdentifier identifier,
  31             final CursorAwareDataTreeSnapshot snapshot) {
  32         this.identifier = requireNonNull(identifier);
  33         this.snapshot = requireNonNull(snapshot);
  34     }
  35
  36     @Override
  37     public DOMDataTreeIdentifier getIdentifier() {
  38         return identifier;
  39     }
  40
  41     DataTreeModificationCursorAdaptor cursor() {
  42         if (cursor == null) {
  43             if (modification == null) {
  44                 modification = snapshot.newModification();
  45             }
  46
  47             // FIXME: Should there be non-root path?
  48             DataTreeModificationCursor dataTreeCursor = modification.openCursor(YangInstanceIdentifier.empty()).get();
  49             cursor = DataTreeModificationCursorAdaptor.of(dataTreeCursor);
  50         }
  51         return cursor;
  52     }
  53
  54     boolean isModified() {
  55         return modification != null;
  56     }
  57
  58     DataTreeModification ready() {
  59         if (cursor != null) {
  60             cursor.close();
  61             cursor = null;
  62         }
  63         DataTreeModification ret = null;
  64         if (modification != null) {
  65             modification.ready();
  66             ret = modification;
  67             modification = null;
  68         }
  69
  70         return ret;
  71     }
  72
  73     void closeCursor() {
  74         cursor.close();
  75         cursor = null;
  76     }
  77 }