data/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 static com.google.common.base.Preconditions.checkArgument;
  11 import static java.util.Objects.requireNonNull;
  12
  13 import com.google.common.annotations.Beta;
  14 import com.google.common.base.Joiner;
  15 import com.google.common.base.Splitter;
  16 import com.google.common.collect.Collections2;
  17 import com.google.common.collect.ImmutableSet;
  18 import java.util.ArrayList;
  19 import java.util.List;
  20 import java.util.Set;
  21 import org.opendaylight.yangtools.yang.data.api.codec.BitsCodec;
  22 import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition;
  23 import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition.Bit;
  24
  25 /**
  26  * Do not use this class outside of yangtools, its presence does not fall into the API stability contract.
  27  */
  28 @Beta
  29 public final class BitsStringCodec extends TypeDefinitionAwareCodec<Set<String>, BitsTypeDefinition>
  30         implements BitsCodec<String> {
  31
  32     private static final Joiner JOINER = Joiner.on(" ").skipNulls();
  33     private static final Splitter SPLITTER = Splitter.on(' ').omitEmptyStrings().trimResults();
  34
  35     private final ImmutableSet<String> validBits;
  36
  37     @SuppressWarnings("unchecked")
  38     private BitsStringCodec(final BitsTypeDefinition typeDef) {
  39         super(requireNonNull(typeDef), (Class<Set<String>>) (Class<?>) Set.class);
  40         validBits = ImmutableSet.copyOf(Collections2.transform(typeDef.getBits(), Bit::getName));
  41     }
  42
  43     public static BitsStringCodec from(final BitsTypeDefinition type) {
  44         return new BitsStringCodec(type);
  45     }
  46
  47     @Override
  48     protected Set<String> deserializeImpl(final String product) {
  49         final Set<String> strings = ImmutableSet.copyOf(SPLITTER.split(product));
  50
  51         // Normalize strings to schema first, retaining definition order
  52         final List<String> sorted = new ArrayList<>(strings.size());
  53         for (final String bit : validBits) {
  54             if (strings.contains(bit)) {
  55                 sorted.add(bit);
  56             }
  57         }
  58
  59         // Check sizes, if the normalized set does not match non-normalized size, non-normalized strings contain
  60         // an invalid bit.
  61         if (sorted.size() != strings.size()) {
  62             for (final String bit : strings) {
  63                 checkArgument(validBits.contains(bit), "Invalid value '%s' for bits type. Allowed values are: %s", bit,
  64                     validBits);
  65             }
  66         }
  67
  68         // In case all valid bits have been specified, retain the set we have created for this codec
  69         return sorted.size() == validBits.size() ? validBits : ImmutableSet.copyOf(sorted);
  70     }
  71
  72     @Override
  73     protected String serializeImpl(final Set<String> input) {
  74         return JOINER.join(input);
  75     }
  76 }