third-party/atomix/utils/src/main/java/io/atomix/utils/serializer/serializers/ImmutableMapSerializer.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.ImmutableMap;
  23 import com.google.common.collect.ImmutableMap.Builder;
  24
  25 import java.util.Map.Entry;
  26
  27 /**
  28  * Kryo Serializer for {@link ImmutableMap}.
  29  */
  30 public class ImmutableMapSerializer extends Serializer<ImmutableMap<?, ?>> {
  31
  32   /**
  33    * Creates {@link ImmutableMap} serializer instance.
  34    */
  35   public ImmutableMapSerializer() {
  36     // non-null, immutable
  37     super(false, true);
  38   }
  39
  40   @Override
  41   public void write(Kryo kryo, Output output, ImmutableMap<?, ?> object) {
  42     output.writeInt(object.size());
  43     for (Entry<?, ?> e : object.entrySet()) {
  44       kryo.writeClassAndObject(output, e.getKey());
  45       kryo.writeClassAndObject(output, e.getValue());
  46     }
  47   }
  48
  49   @Override
  50   public ImmutableMap<?, ?> read(Kryo kryo, Input input,
  51       Class<ImmutableMap<?, ?>> type) {
  52     final int size = input.readInt();
  53     switch (size) {
  54       case 0:
  55         return ImmutableMap.of();
  56       case 1:
  57         return ImmutableMap.of(kryo.readClassAndObject(input),
  58             kryo.readClassAndObject(input));
  59
  60       default:
  61         Builder<Object, Object> builder = ImmutableMap.builder();
  62         for (int i = 0; i < size; ++i) {
  63           builder.put(kryo.readClassAndObject(input),
  64               kryo.readClassAndObject(input));
  65         }
  66         return builder.build();
  67     }
  68   }
  69 }