Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2017-present Open Networking Laboratory |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | package org.onosproject.netconf.client.impl; |
| 18 | |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 19 | import com.google.common.annotations.Beta; |
| 20 | import org.apache.felix.scr.annotations.Activate; |
| 21 | import org.apache.felix.scr.annotations.Component; |
| 22 | import org.apache.felix.scr.annotations.Deactivate; |
| 23 | import org.apache.felix.scr.annotations.Reference; |
| 24 | import org.apache.felix.scr.annotations.ReferenceCardinality; |
Thomas Vachuska | 59d24eb | 2017-03-22 10:57:34 -0700 | [diff] [blame] | 25 | import org.apache.felix.scr.annotations.Service; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 26 | import org.onosproject.cluster.NodeId; |
| 27 | import org.onosproject.net.DeviceId; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 28 | import org.onosproject.netconf.NetconfController; |
| 29 | import org.onosproject.netconf.NetconfDevice; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 30 | import org.onosproject.netconf.NetconfException; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 31 | import org.onosproject.netconf.NetconfSession; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 32 | import org.onosproject.netconf.client.NetconfTranslator; |
| 33 | import org.onosproject.yang.model.DataNode; |
| 34 | import org.onosproject.yang.model.InnerNode; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 35 | import org.onosproject.yang.model.KeyLeaf; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 36 | import org.onosproject.yang.model.LeafListKey; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 37 | import org.onosproject.yang.model.LeafNode; |
| 38 | import org.onosproject.yang.model.ListKey; |
| 39 | import org.onosproject.yang.model.NodeKey; |
| 40 | import org.onosproject.yang.model.ResourceData; |
| 41 | import org.onosproject.yang.model.ResourceId; |
Aaron Kruglikov | d24dc98 | 2017-03-23 18:48:58 -0700 | [diff] [blame] | 42 | import org.onosproject.yang.model.SchemaContext; |
| 43 | import org.onosproject.yang.model.SchemaContextProvider; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 44 | import org.onosproject.yang.model.SchemaId; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 45 | import org.onosproject.yang.runtime.CompositeStream; |
| 46 | import org.onosproject.yang.runtime.DefaultAnnotatedNodeInfo; |
| 47 | import org.onosproject.yang.runtime.DefaultAnnotation; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 48 | import org.onosproject.yang.runtime.DefaultCompositeData; |
| 49 | import org.onosproject.yang.runtime.DefaultCompositeStream; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 50 | import org.onosproject.yang.runtime.DefaultResourceData; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 51 | import org.onosproject.yang.runtime.DefaultRuntimeContext; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 52 | import org.onosproject.yang.runtime.DefaultYangSerializerContext; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 53 | import org.onosproject.yang.runtime.YangRuntimeService; |
| 54 | import org.onosproject.yang.runtime.YangSerializerContext; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 55 | import org.onosproject.yang.runtime.helperutils.SerializerHelper; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 56 | import org.osgi.service.component.ComponentContext; |
| 57 | import org.slf4j.Logger; |
| 58 | import org.slf4j.LoggerFactory; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 59 | |
| 60 | import java.io.BufferedReader; |
| 61 | import java.io.ByteArrayInputStream; |
| 62 | import java.io.IOException; |
| 63 | import java.io.InputStream; |
| 64 | import java.io.InputStreamReader; |
| 65 | import java.nio.charset.StandardCharsets; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 66 | import java.util.Iterator; |
| 67 | import java.util.List; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 68 | import java.util.regex.Matcher; |
| 69 | import java.util.regex.Pattern; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 70 | |
| 71 | import static com.google.common.base.Preconditions.checkNotNull; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 72 | import static org.onosproject.yang.model.DataNode.Type.SINGLE_INSTANCE_LEAF_VALUE_NODE; |
| 73 | import static org.onosproject.yang.runtime.helperutils.SerializerHelper.addDataNode; |
| 74 | |
| 75 | /*FIXME these imports are not visible using OSGI*/ |
| 76 | |
| 77 | /*FIXME these imports are not visible using OSGI*/ |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 78 | |
| 79 | /*TODO once the API's are finalized this comment will be made more specified.*/ |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 80 | |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 81 | /** |
| 82 | * Translator which accepts data types defined for the DynamicConfigService and |
| 83 | * makes the appropriate calls to NETCONF devices before encoding and returning |
| 84 | * responses in formats suitable for the DynamicConfigService. |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 85 | * <p> |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 86 | * NOTE: This entity does not ensure you are the master of a device you attempt |
| 87 | * to contact. If you are not the master an error will be thrown because there |
| 88 | * will be no session available. |
| 89 | */ |
| 90 | @Beta |
Thomas Vachuska | 59d24eb | 2017-03-22 10:57:34 -0700 | [diff] [blame] | 91 | @Service |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 92 | @Component(immediate = true) |
| 93 | public class NetconfTranslatorImpl implements NetconfTranslator { |
| 94 | |
| 95 | private final Logger log = LoggerFactory |
| 96 | .getLogger(getClass()); |
| 97 | |
| 98 | private NodeId localNodeId; |
| 99 | |
| 100 | private static final String GET_CONFIG_MESSAGE_REGEX = |
| 101 | "<data>\n?\\s*(.*?)\n?\\s*</data>"; |
| 102 | private static final int GET_CONFIG_CORE_MESSAGE_GROUP = 1; |
| 103 | private static final Pattern GET_CONFIG_CORE_MESSAGE_PATTERN = |
| 104 | Pattern.compile(GET_CONFIG_MESSAGE_REGEX, Pattern.DOTALL); |
| 105 | private static final String GET_CORE_MESSAGE_REGEX = "<data>\n?\\s*(.*?)\n?\\s*</data>"; |
| 106 | private static final int GET_CORE_MESSAGE_GROUP = 1; |
| 107 | private static final Pattern GET_CORE_MESSAGE_PATTERN = |
| 108 | Pattern.compile(GET_CORE_MESSAGE_REGEX, Pattern.DOTALL); |
| 109 | |
| 110 | private static final String NETCONF_1_0_BASE_NAMESPACE = |
| 111 | "urn:ietf:params:xml:ns:netconf:base:1.0"; |
| 112 | |
| 113 | private static final String GET_URI = "urn:ietf:params:xml:ns:yang:" + |
| 114 | "yrt-ietf-network:networks/network/node"; |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 115 | private static final String XML_ENCODING_SPECIFIER = "xml"; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 116 | private static final String OP_SPECIFIER = "xc:operation"; |
| 117 | private static final String REPLACE_OP_SPECIFIER = "replace"; |
| 118 | private static final String DELETE_OP_SPECIFIER = "delete"; |
| 119 | private static final String XMLNS_XC_SPECIFIER = "xmlns:xc"; |
| 120 | private static final String XMLNS_SPECIFIER = "xmlns"; |
| 121 | |
| 122 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 123 | protected NetconfController netconfController; |
| 124 | |
| 125 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 126 | protected YangRuntimeService yangRuntimeService; |
| 127 | |
| 128 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
Aaron Kruglikov | d24dc98 | 2017-03-23 18:48:58 -0700 | [diff] [blame] | 129 | protected SchemaContextProvider schemaContextProvider; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 130 | |
| 131 | @Activate |
| 132 | public void activate(ComponentContext context) { |
| 133 | log.info("Started"); |
| 134 | } |
| 135 | |
| 136 | @Deactivate |
| 137 | public void deactivate() { |
| 138 | log.info("Stopped"); |
| 139 | } |
| 140 | |
| 141 | @Override |
| 142 | public ResourceData getDeviceConfig(DeviceId deviceId) throws IOException { |
| 143 | NetconfSession session = getNetconfSession(deviceId); |
| 144 | /*FIXME "running" will be replaced with an enum once netconf supports multiple datastores.*/ |
| 145 | String reply = session.getConfig("running"); |
| 146 | Matcher protocolStripper = GET_CONFIG_CORE_MESSAGE_PATTERN.matcher(reply); |
| 147 | reply = protocolStripper.group(GET_CONFIG_CORE_MESSAGE_GROUP); |
| 148 | return yangRuntimeService.decode( |
| 149 | new DefaultCompositeStream( |
| 150 | null, |
| 151 | /*FIXME is UTF_8 the appropriate encoding? */ |
| 152 | new ByteArrayInputStream(reply.getBytes(StandardCharsets.UTF_8))), |
| 153 | new DefaultRuntimeContext.Builder() |
| 154 | .setDataFormat(XML_ENCODING_SPECIFIER) |
| 155 | .addAnnotation( |
| 156 | new DefaultAnnotation(XMLNS_SPECIFIER, |
| 157 | NETCONF_1_0_BASE_NAMESPACE)) |
| 158 | .build()).resourceData(); |
| 159 | } |
| 160 | |
| 161 | @Override |
| 162 | public boolean editDeviceConfig(DeviceId deviceId, ResourceData resourceData, |
| 163 | NetconfTranslator.OperationType operationType) throws IOException { |
| 164 | NetconfSession session = getNetconfSession(deviceId); |
Aaron Kruglikov | d24dc98 | 2017-03-23 18:48:58 -0700 | [diff] [blame] | 165 | SchemaContext context = schemaContextProvider |
| 166 | .getSchemaContext(ResourceId.builder().addBranchPointSchema("/", null).build()); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 167 | ResourceData modifiedPathResourceData = getResourceData(resourceData.resourceId(), |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 168 | resourceData.dataNodes(), |
| 169 | new DefaultYangSerializerContext(context, null)); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 170 | DefaultCompositeData.Builder compositeDataBuilder = DefaultCompositeData |
| 171 | .builder() |
| 172 | .resourceData(modifiedPathResourceData); |
| 173 | for (DataNode node : resourceData.dataNodes()) { |
| 174 | ResourceId resourceId = resourceData.resourceId(); |
| 175 | if (operationType != OperationType.DELETE) { |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 176 | resourceId = getAnnotatedNodeResourceId( |
| 177 | resourceData.resourceId(), node); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 178 | } |
| 179 | if (resourceId != null) { |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 180 | DefaultAnnotatedNodeInfo.Builder annotatedNodeInfo = |
| 181 | DefaultAnnotatedNodeInfo.builder(); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 182 | annotatedNodeInfo.resourceId(resourceId); |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 183 | annotatedNodeInfo.addAnnotation( |
| 184 | new DefaultAnnotation( |
| 185 | OP_SPECIFIER, operationType == OperationType.DELETE ? |
| 186 | DELETE_OP_SPECIFIER : REPLACE_OP_SPECIFIER)); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 187 | compositeDataBuilder.addAnnotatedNodeInfo(annotatedNodeInfo.build()); |
| 188 | } |
| 189 | } |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 190 | CompositeStream config = yangRuntimeService.encode( |
| 191 | compositeDataBuilder.build(), |
| 192 | new DefaultRuntimeContext.Builder() |
| 193 | .setDataFormat(XML_ENCODING_SPECIFIER) |
| 194 | .addAnnotation(new DefaultAnnotation( |
| 195 | XMLNS_XC_SPECIFIER, NETCONF_1_0_BASE_NAMESPACE)) |
| 196 | .build()); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 197 | /* FIXME need to fix to string conversion. */ |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 198 | |
| 199 | try { |
| 200 | String reply = session.requestSync(Utils.editConfig(streamToString( |
| 201 | config.resourceData()))); |
| 202 | } catch (NetconfException e) { |
| 203 | log.error("failed to send a request sync", e); |
| 204 | return false; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 205 | } |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 206 | /* NOTE: a failure to edit is reflected as a NetconfException.*/ |
| 207 | return true; |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 208 | } |
| 209 | |
| 210 | @Override |
| 211 | public ResourceData getDeviceState(DeviceId deviceId) throws IOException { |
| 212 | NetconfSession session = getNetconfSession(deviceId); |
| 213 | /*TODO the first parameter will come into use if get is required to support filters.*/ |
| 214 | String reply = session.get(null, null); |
| 215 | Matcher protocolStripper = GET_CORE_MESSAGE_PATTERN.matcher(reply); |
| 216 | reply = protocolStripper.group(GET_CORE_MESSAGE_GROUP); |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 217 | return yangRuntimeService.decode( |
| 218 | new DefaultCompositeStream( |
| 219 | null, |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 220 | /*FIXME is UTF_8 the appropriate encoding? */ |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 221 | new ByteArrayInputStream(reply.getBytes(StandardCharsets.UTF_8))), |
| 222 | new DefaultRuntimeContext.Builder() |
| 223 | .setDataFormat(XML_ENCODING_SPECIFIER) |
| 224 | .addAnnotation( |
| 225 | new DefaultAnnotation( |
| 226 | XMLNS_SPECIFIER, |
| 227 | NETCONF_1_0_BASE_NAMESPACE)) |
| 228 | .build()).resourceData(); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 229 | /* NOTE: a failure to get is reflected as a NetconfException.*/ |
| 230 | } |
| 231 | |
| 232 | /** |
| 233 | * Returns a session for the specified deviceId if this node is its master, |
| 234 | * returns null otherwise. |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 235 | * |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 236 | * @param deviceId the id of node for witch we wish to retrieve a session |
| 237 | * @return a NetconfSession with the specified node or null |
| 238 | */ |
| 239 | private NetconfSession getNetconfSession(DeviceId deviceId) { |
| 240 | NetconfDevice device = netconfController.getNetconfDevice(deviceId); |
| 241 | checkNotNull(device, "The specified deviceId could not be found by the NETCONF controller."); |
| 242 | NetconfSession session = device.getSession(); |
| 243 | checkNotNull(session, "A session could not be retrieved for the specified deviceId."); |
| 244 | return session; |
| 245 | } |
| 246 | |
| 247 | /** |
| 248 | * Accepts a stream and converts it to a string. |
| 249 | * |
| 250 | * @param stream the stream to be converted |
| 251 | * @return a string with the same sequence of characters as the stream |
| 252 | * @throws IOException if reading from the stream fails |
| 253 | */ |
| 254 | private String streamToString(InputStream stream) throws IOException { |
| 255 | BufferedReader reader = new BufferedReader(new InputStreamReader(stream)); |
| 256 | StringBuilder builder = new StringBuilder(); |
| 257 | |
| 258 | String nextLine = reader.readLine(); |
| 259 | while (nextLine != null) { |
| 260 | builder.append(nextLine); |
| 261 | nextLine = reader.readLine(); |
| 262 | } |
| 263 | return builder.toString(); |
| 264 | } |
| 265 | |
| 266 | /** |
| 267 | * Returns resource data having resource id as "/" and data node tree |
| 268 | * starting from "/" by creating data nodes for given resource id(parent |
| 269 | * node for given list of nodes) and list of child nodes. |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 270 | * <p> |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 271 | * This api will be used in encode flow only. |
| 272 | * |
| 273 | * @param rid resource identifier till parent node |
| 274 | * @param nodes list of data nodes |
| 275 | * @param cont yang serializer context |
| 276 | * @return resource data. |
| 277 | */ |
| 278 | public static ResourceData getResourceData( |
| 279 | ResourceId rid, List<DataNode> nodes, YangSerializerContext cont) { |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 280 | if (rid == null) { |
| 281 | ResourceData.Builder resData = DefaultResourceData.builder(); |
| 282 | for (DataNode node : nodes) { |
| 283 | resData.addDataNode(node); |
| 284 | } |
| 285 | return resData.build(); |
| 286 | } |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 287 | List<NodeKey> keys = rid.nodeKeys(); |
| 288 | Iterator<NodeKey> it = keys.iterator(); |
| 289 | DataNode.Builder dbr = SerializerHelper.initializeDataNode(cont); |
| 290 | |
| 291 | // checking the resource id weather it is getting started from / or not |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 292 | |
| 293 | while (it.hasNext()) { |
| 294 | NodeKey nodekey = it.next(); |
| 295 | SchemaId sid = nodekey.schemaId(); |
| 296 | dbr = addDataNode(dbr, sid.name(), sid.namespace(), |
| 297 | null, null); |
| 298 | if (nodekey instanceof ListKey) { |
| 299 | for (KeyLeaf keyLeaf : ((ListKey) nodekey).keyLeafs()) { |
| 300 | String val; |
| 301 | if (keyLeaf.leafValue() == null) { |
| 302 | val = null; |
| 303 | } else { |
| 304 | val = keyLeaf.leafValAsString(); |
| 305 | } |
| 306 | dbr = addDataNode(dbr, keyLeaf.leafSchema().name(), |
| 307 | sid.namespace(), val, |
| 308 | SINGLE_INSTANCE_LEAF_VALUE_NODE); |
| 309 | } |
| 310 | } |
| 311 | } |
| 312 | |
| 313 | if (dbr instanceof LeafNode.Builder && |
| 314 | (nodes != null || !nodes.isEmpty())) { |
| 315 | //exception "leaf/leaf-list can not have child node" |
| 316 | } |
| 317 | |
| 318 | if (nodes != null && !nodes.isEmpty()) { |
| 319 | // adding the parent node for given list of nodes |
| 320 | for (DataNode node : nodes) { |
| 321 | dbr = ((InnerNode.Builder) dbr).addNode(node); |
| 322 | } |
| 323 | } |
Aaron Kruglikov | d24dc98 | 2017-03-23 18:48:58 -0700 | [diff] [blame] | 324 | /*FIXME this can be uncommented for use with versions of onos-yang-tools newer than 1.12.0-b6*/ |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 325 | // while (dbr.parent() != null) { |
| 326 | // dbr = SerializerHelper.exitDataNode(dbr); |
| 327 | // } |
| 328 | |
| 329 | ResourceData.Builder resData = DefaultResourceData.builder(); |
| 330 | |
| 331 | resData.addDataNode(dbr.build()); |
| 332 | resData.resourceId(null); |
| 333 | return resData.build(); |
| 334 | } |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 335 | |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 336 | /** |
| 337 | * Returns resource id for annotated data node by adding resource id of top |
| 338 | * level data node to given resource id. |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 339 | * <p> |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 340 | * Annotation will be added to node based on the updated resource id. |
| 341 | * This api will be used in encode flow only. |
| 342 | * |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 343 | * @param rid resource identifier till parent node |
| 344 | * @param node data node |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 345 | * @return updated resource id. |
| 346 | */ |
| 347 | public static ResourceId getAnnotatedNodeResourceId(ResourceId rid, |
| 348 | DataNode node) { |
| 349 | |
| 350 | String val; |
| 351 | ResourceId.Builder rIdBldr = ResourceId.builder(); |
Bharat saraswal | b0dbe6b | 2017-03-24 22:51:06 +0530 | [diff] [blame^] | 352 | if (rid != null) { |
| 353 | try { |
| 354 | rIdBldr = rid.copyBuilder(); |
| 355 | } catch (CloneNotSupportedException e) { |
| 356 | e.printStackTrace(); |
| 357 | } |
| 358 | } else { |
| 359 | rIdBldr.addBranchPointSchema("/", null); |
Aaron Kruglikov | bc26a52 | 2017-02-21 11:27:49 -0800 | [diff] [blame] | 360 | } |
| 361 | DataNode.Type type = node.type(); |
| 362 | NodeKey k = node.key(); |
| 363 | SchemaId sid = k.schemaId(); |
| 364 | |
| 365 | switch (type) { |
| 366 | |
| 367 | case MULTI_INSTANCE_LEAF_VALUE_NODE: |
| 368 | val = ((LeafListKey) k).value().toString(); |
| 369 | rIdBldr.addLeafListBranchPoint(sid.name(), sid.namespace(), val); |
| 370 | break; |
| 371 | |
| 372 | case MULTI_INSTANCE_NODE: |
| 373 | rIdBldr.addBranchPointSchema(sid.name(), sid.namespace()); |
| 374 | // Preparing the list of key values for multiInstanceNode |
| 375 | for (KeyLeaf keyLeaf : ((ListKey) node.key()).keyLeafs()) { |
| 376 | val = keyLeaf.leafValAsString(); |
| 377 | rIdBldr.addKeyLeaf(keyLeaf.leafSchema().name(), sid.namespace(), val); |
| 378 | } |
| 379 | break; |
| 380 | case SINGLE_INSTANCE_LEAF_VALUE_NODE: |
| 381 | case SINGLE_INSTANCE_NODE: |
| 382 | rIdBldr.addBranchPointSchema(sid.name(), sid.namespace()); |
| 383 | break; |
| 384 | |
| 385 | default: |
| 386 | throw new IllegalArgumentException(); |
| 387 | } |
| 388 | return rIdBldr.build(); |
| 389 | } |
| 390 | } |