2 * Copyright (c) 2020 PANTHEON.tech, s.r.o. 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.mdsal.replicate.netty;
10 import static com.google.common.base.Preconditions.checkState;
11 import static com.google.common.base.Verify.verify;
12 import static java.util.Objects.requireNonNull;
14 import com.google.common.util.concurrent.FutureCallback;
15 import com.google.common.util.concurrent.MoreExecutors;
16 import io.netty.buffer.ByteBuf;
17 import io.netty.buffer.ByteBufInputStream;
18 import io.netty.buffer.Unpooled;
19 import io.netty.channel.Channel;
20 import io.netty.channel.ChannelHandlerContext;
21 import io.netty.channel.SimpleChannelInboundHandler;
22 import java.io.IOException;
23 import java.util.ArrayList;
24 import java.util.List;
25 import org.opendaylight.mdsal.common.api.CommitInfo;
26 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
27 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
28 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
29 import org.opendaylight.mdsal.replicate.common.DataTreeCandidateUtils;
30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
31 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
32 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.stream.ReusableStreamReceiver;
34 import org.opendaylight.yangtools.yang.data.codec.binfmt.DataTreeCandidateInputOutput;
35 import org.opendaylight.yangtools.yang.data.codec.binfmt.NormalizedNodeDataInput;
36 import org.opendaylight.yangtools.yang.data.impl.schema.ReusableImmutableNormalizedNodeStreamWriter;
37 import org.opendaylight.yangtools.yang.data.spi.node.ImmutableNodes;
38 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeCandidate;
39 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
40 import org.slf4j.Logger;
41 import org.slf4j.LoggerFactory;
43 final class SinkRequestHandler extends SimpleChannelInboundHandler<ByteBuf> {
44 private static final Logger LOG = LoggerFactory.getLogger(SinkRequestHandler.class);
45 private static final ContainerNode EMPTY_ROOT = ImmutableNodes.newContainerBuilder()
46 .withNodeIdentifier(NodeIdentifier.create(SchemaContext.NAME))
49 private final ReusableStreamReceiver receiver = ReusableImmutableNormalizedNodeStreamWriter.create();
50 private final List<ByteBuf> chunks = new ArrayList<>();
51 private final DOMDataTreeIdentifier tree;
52 private final DOMTransactionChain chain;
54 SinkRequestHandler(final DOMDataTreeIdentifier tree, final DOMTransactionChain chain) {
55 this.tree = requireNonNull(tree);
56 this.chain = requireNonNull(chain);
60 protected void channelRead0(final ChannelHandlerContext ctx, final ByteBuf msg) throws IOException {
61 verify(msg.isReadable(), "Empty message received");
63 final short msgType = msg.readUnsignedByte();
64 final Channel channel = ctx.channel();
65 LOG.trace("Channel {} received message type {}", channel, msgType);
67 case Constants.MSG_EMPTY_DATA -> handleEmptyData();
68 case Constants.MSG_DTC_CHUNK -> chunks.add(msg.retain());
69 case Constants.MSG_DTC_APPLY -> handleDtcApply();
70 case Constants.MSG_PING -> {
71 LOG.trace("Received PING from Source, sending PONG");
72 ctx.channel().writeAndFlush(Constants.PONG);
74 default -> throw new IllegalStateException("Unexpected message type " + msgType);
78 private void handleEmptyData() {
79 final var tx = chain.newWriteOnlyTransaction();
81 if (tree.path().isEmpty()) {
82 tx.put(tree.datastore(), YangInstanceIdentifier.of(), EMPTY_ROOT);
84 tx.delete(tree.datastore(), tree.path());
89 private void handleDtcApply() throws IOException {
90 checkState(!chunks.isEmpty(), "No chunks to apply");
92 final ByteBuf bufs = Unpooled.wrappedBuffer(chunks.toArray(new ByteBuf[0]));
95 final DataTreeCandidate candidate;
96 try (ByteBufInputStream stream = new ByteBufInputStream(bufs)) {
97 candidate = DataTreeCandidateInputOutput.readDataTreeCandidate(NormalizedNodeDataInput.newDataInput(stream),
101 final DOMDataTreeWriteTransaction tx = chain.newWriteOnlyTransaction();
102 DataTreeCandidateUtils.applyToTransaction(tx, tree.datastore(), candidate);
106 private static void commit(final DOMDataTreeWriteTransaction tx) {
107 tx.commit().addCallback(new FutureCallback<CommitInfo>() {
109 public void onSuccess(final CommitInfo result) {
110 LOG.trace("Transaction committed with {}", result);
114 public void onFailure(final Throwable cause) {
115 // Handled by transaction chain listener
117 }, MoreExecutors.directExecutor());