Use UnsignedLongSet instead of RangeSet in metadata

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / utils / ImmutableUnsignedLongSet.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/utils/ImmutableUnsignedLongSet.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/utils/ImmutableUnsignedLongSet.java
new file mode 100644 (file)
index 0000000..e6c900b
--- /dev/null
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/utils/ImmutableUnsignedLongSet.java
@@ -0,0 +1,74 @@
+/*
+ * Copyright (c) 2021 PANTHEON.tech, s.r.o. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.cluster.datastore.utils;
+
+import com.google.common.annotations.Beta;
+import java.io.DataInput;
+import java.io.DataOutput;
+import java.io.IOException;
+import java.util.TreeSet;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.yangtools.concepts.Immutable;
+import org.opendaylight.yangtools.concepts.WritableObject;
+
+@Beta
+public final class ImmutableUnsignedLongSet extends UnsignedLongSet implements Immutable, WritableObject {
+    private static final @NonNull ImmutableUnsignedLongSet EMPTY = new ImmutableUnsignedLongSet(new TreeSet<>());
+
+    private ImmutableUnsignedLongSet(final TreeSet<Entry> ranges) {
+        super(ranges);
+    }
+
+    static @NonNull ImmutableUnsignedLongSet of(final TreeSet<Entry> ranges) {
+        return ranges.isEmpty() ? EMPTY : new ImmutableUnsignedLongSet(ranges);
+    }
+
+    public static @NonNull ImmutableUnsignedLongSet of() {
+        return EMPTY;
+    }
+
+    @Override
+    public ImmutableUnsignedLongSet immutableCopy() {
+        return this;
+    }
+
+    public static @NonNull ImmutableUnsignedLongSet readFrom(final DataInput in) throws IOException {
+        return readFrom(in, in.readInt());
+    }
+
+    public static @NonNull ImmutableUnsignedLongSet readFrom(final DataInput in, final int size) throws IOException {
+        if (size == 0) {
+            return EMPTY;
+        }
+
+        final var ranges = new TreeSet<Entry>();
+        for (int i = 0; i < size; ++i) {
+            ranges.add(Entry.readUnsigned(in));
+        }
+        return new ImmutableUnsignedLongSet(ranges);
+    }
+
+    @Override
+    public void writeTo(final DataOutput out) throws IOException {
+        out.writeInt(size());
+        writeRanges(out);
+    }
+
+    public void writeRangesTo(final @NonNull DataOutput out, final int size) throws IOException {
+        if (size != size()) {
+            throw new IOException("Mismatched size: expected " + size() + ", got " + size);
+        }
+        writeRanges(out);
+    }
+
+    private void writeRanges(final @NonNull DataOutput out) throws IOException {
+        for (var range : trustedRanges()) {
+            range.writeUnsigned(out);
+        }
+    }
+}