yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/codec/BitsStringCodec.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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.yangtools.yang.data.impl.codec;
   9
  10 import com.google.common.base.Joiner;
  11 import com.google.common.base.Optional;
  12 import com.google.common.base.Preconditions;
  13 import com.google.common.base.Splitter;
  14 import com.google.common.collect.ImmutableSet;
  15 import com.google.common.collect.Sets;
  16 import java.util.List;
  17 import java.util.Set;
  18 import org.opendaylight.yangtools.yang.data.api.codec.BitsCodec;
  19 import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition;
  20 import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition.Bit;
  21
  22 final class BitsStringCodec extends TypeDefinitionAwareCodec<Set<String>, BitsTypeDefinition>
  23         implements BitsCodec<String> {
  24
  25     private static final Joiner JOINER = Joiner.on(" ").skipNulls();
  26     private static final Splitter SPLITTER = Splitter.on(' ').omitEmptyStrings().trimResults();
  27
  28     private final Set<String> bits;
  29
  30     @SuppressWarnings("unchecked")
  31     private BitsStringCodec(final Optional<BitsTypeDefinition> typeDef) {
  32         super(typeDef, (Class<Set<String>>) ((Class<?>) Set.class));
  33         if (typeDef.isPresent()) {
  34             final List<Bit> yangBits = typeDef.get().getBits();
  35             final Set<String> bitsBuilder = Sets.newHashSet();
  36             for (final Bit bit : yangBits) {
  37                 bitsBuilder.add(bit.getName());
  38             }
  39             bits = ImmutableSet.copyOf(bitsBuilder);
  40         } else {
  41             bits = null;
  42         }
  43     }
  44
  45     static TypeDefinitionAwareCodec<?, BitsTypeDefinition> from(final BitsTypeDefinition normalizedType) {
  46         return new BitsStringCodec(Optional.fromNullable(normalizedType));
  47     }
  48
  49     @Override
  50     public String serialize(final Set<String> data) {
  51         return data == null ? "" : JOINER.join(data);
  52     }
  53
  54     @Override
  55     public Set<String> deserialize(final String stringRepresentation) {
  56         if (stringRepresentation == null) {
  57             return ImmutableSet.of();
  58         }
  59
  60         final Iterable<String> strings = SPLITTER.split(stringRepresentation);
  61         validate(strings);
  62         return ImmutableSet.copyOf(strings);
  63     }
  64
  65     private void validate(final Iterable<String> strings) {
  66         if (bits != null) {
  67             for (final String bit : strings) {
  68                 Preconditions.checkArgument(bits.contains(bit),
  69                         "Invalid value '%s' for bits type. Allowed values are: %s", bit, bits);
  70             }
  71         }
  72     }
  73 }