replicate/mdsal-replicate-netty/src/main/java/org/opendaylight/mdsal/replicate/netty/AbstractSourceMessage.java

   1 /*
   2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. 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 package org.opendaylight.mdsal.replicate.netty;
   9
  10 import static java.util.Objects.requireNonNull;
  11
  12 import java.io.DataOutputStream;
  13 import java.io.IOException;
  14 import java.util.Collection;
  15 import java.util.List;
  16 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
  17 import org.opendaylight.yangtools.yang.data.codec.binfmt.DataTreeCandidateInputOutput;
  18 import org.opendaylight.yangtools.yang.data.codec.binfmt.NormalizedNodeDataOutput;
  19 import org.opendaylight.yangtools.yang.data.codec.binfmt.NormalizedNodeStreamVersion;
  20
  21 abstract class AbstractSourceMessage {
  22     private static final class Empty extends AbstractSourceMessage {
  23         @Override
  24         void encodeTo(final NormalizedNodeStreamVersion version, final List<Object> out) throws IOException {
  25             out.add(Constants.EMPTY_DATA);
  26         }
  27     }
  28
  29     private static final class Deltas extends AbstractSourceMessage {
  30         private final Collection<DataTreeCandidate> deltas;
  31
  32         Deltas(final Collection<DataTreeCandidate> deltas) {
  33             this.deltas = requireNonNull(deltas);
  34         }
  35
  36         @Override
  37         void encodeTo(final NormalizedNodeStreamVersion version, final List<Object> out) throws IOException {
  38             for (DataTreeCandidate candidate : deltas) {
  39                 try (DataOutputStream stream = new DataOutputStream(new SplittingOutputStream(out))) {
  40                     try (NormalizedNodeDataOutput output = version.newDataOutput(stream)) {
  41                         DataTreeCandidateInputOutput.writeDataTreeCandidate(output, candidate);
  42                     }
  43                 }
  44                 out.add(Constants.DTC_APPLY);
  45             }
  46         }
  47     }
  48
  49     private static final AbstractSourceMessage EMPTY = new Empty();
  50
  51     static AbstractSourceMessage empty() {
  52         return EMPTY;
  53     }
  54
  55     static AbstractSourceMessage of(final Collection<DataTreeCandidate> deltas) {
  56         return new Deltas(deltas);
  57     }
  58
  59     abstract void encodeTo(NormalizedNodeStreamVersion version, List<Object> out) throws IOException;
  60 }