third-party/atomix/storage/src/main/java/io/atomix/utils/serializer/serializers/ImmutableSetSerializer.java

   1 /*
   2  * Copyright 2014-present Open Networking Foundation
   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 package io.atomix.utils.serializer.serializers;
  17
  18 import com.esotericsoftware.kryo.Kryo;
  19 import com.esotericsoftware.kryo.Serializer;
  20 import com.esotericsoftware.kryo.io.Input;
  21 import com.esotericsoftware.kryo.io.Output;
  22 import com.google.common.collect.ImmutableSet;
  23
  24 /**
  25  * Kryo Serializer for {@link ImmutableSet}.
  26  */
  27 public class ImmutableSetSerializer extends Serializer<ImmutableSet<?>> {
  28
  29   /**
  30    * Creates {@link ImmutableSet} serializer instance.
  31    */
  32   public ImmutableSetSerializer() {
  33     // non-null, immutable
  34     super(false, true);
  35   }
  36
  37   @Override
  38   public void write(Kryo kryo, Output output, ImmutableSet<?> object) {
  39     output.writeInt(object.size());
  40     for (Object e : object) {
  41       kryo.writeClassAndObject(output, e);
  42     }
  43   }
  44
  45   @Override
  46   public ImmutableSet<?> read(Kryo kryo, Input input,
  47       Class<ImmutableSet<?>> type) {
  48     final int size = input.readInt();
  49     switch (size) {
  50       case 0:
  51         return ImmutableSet.of();
  52       case 1:
  53         return ImmutableSet.of(kryo.readClassAndObject(input));
  54       default:
  55         Object[] elms = new Object[size];
  56         for (int i = 0; i < size; ++i) {
  57           elms[i] = kryo.readClassAndObject(input);
  58         }
  59         return ImmutableSet.copyOf(elms);
  60     }
  61   }
  62 }