binding2/mdsal-binding2-java-api-generator/src/main/java/org/opendaylight/mdsal/binding2/java/api/generator/rangeGenerators/LengthGenerator.java

   1 /*
   2  * Copyright (c) 2016 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.mdsal.binding2.java.api.generator.rangeGenerators;
   9
  10 import com.google.common.collect.Range;
  11 import java.math.BigInteger;
  12 import java.util.ArrayList;
  13 import java.util.Collection;
  14 import java.util.List;
  15 import javax.annotation.Nonnull;
  16 import javax.annotation.Nullable;
  17 import org.opendaylight.mdsal.binding2.model.api.Type;
  18 import org.opendaylight.yangtools.yang.model.api.type.LengthConstraint;
  19
  20 public final class LengthGenerator {
  21     private LengthGenerator() {
  22         throw new UnsupportedOperationException();
  23     }
  24
  25     private static String lengthCheckerName(final String member) {
  26         return "check" + member + "Length";
  27     }
  28
  29     private static Collection<String> createExpressions(final Collection<LengthConstraint> constraints) {
  30         final Collection<String> ret = new ArrayList<>(constraints.size());
  31
  32         for (LengthConstraint l : constraints) {
  33             final StringBuilder sb = new StringBuilder("length >");
  34
  35             // We have to deal with restrictions being out of integer's range
  36             if (l.getMin().longValue() <= Integer.MAX_VALUE) {
  37                 sb.append('=');
  38             }
  39             sb.append(' ').append(l.getMin().intValue());
  40
  41             final int max = l.getMax().intValue();
  42             if (max < Integer.MAX_VALUE) {
  43                 sb.append(" && length <= ").append(max);
  44             }
  45
  46             ret.add(sb.toString());
  47         }
  48
  49         return ret;
  50     }
  51
  52     private static String createLengthString(final Collection<LengthConstraint> constraints) {
  53         final List<Range<BigInteger>> ranges = new ArrayList<>(constraints.size());
  54
  55         for (LengthConstraint c : constraints) {
  56             ranges.add(Range.closed(new BigInteger(c.getMin().toString()), new BigInteger(c.getMax().toString())));
  57         }
  58
  59         return ranges.toString();
  60     }
  61
  62     private static String generateArrayLengthChecker(final String member, final Collection<LengthConstraint> constraints) {
  63         final StringBuilder sb = new StringBuilder();
  64         final Collection<String> expressions = createExpressions(constraints);
  65
  66         sb.append("private static void ").append(lengthCheckerName(member)).append("(final byte[] value) {\n");
  67
  68         if (!expressions.isEmpty()) {
  69             sb.append("    final int length = value.length;\n");
  70
  71             for (String exp : expressions) {
  72                 sb.append("    if (").append(exp).append(") {\n");
  73                 sb.append("        return;\n");
  74                 sb.append("    }\n");
  75             }
  76
  77             sb.append("    throw new IllegalArgumentException(String.format(\"Invalid length: %s, expected: ")
  78               .append(createLengthString(constraints)).append(".\", java.util.Arrays.toString(value)));\n");
  79         }
  80
  81         sb.append("}\n");
  82
  83         return sb.toString();
  84     }
  85
  86     private static String generateStringLengthChecker(final String member, final Collection<LengthConstraint> constraints) {
  87         final StringBuilder sb = new StringBuilder();
  88         final Collection<String> expressions = createExpressions(constraints);
  89
  90         sb.append("private static void ").append(lengthCheckerName(member)).append("(final String value) {\n");
  91
  92         if (!expressions.isEmpty()) {
  93             sb.append("    final int length = value.length();\n");
  94
  95             for (String exp : expressions) {
  96                 sb.append("    if (").append(exp).append(") {\n");
  97                 sb.append("        return;\n");
  98                 sb.append("    }\n");
  99             }
 100
 101             sb.append("    throw new IllegalArgumentException(String.format(\"Invalid length: %s, expected: ")
 102               .append(createLengthString(constraints)).append(".\", value));\n");
 103         }
 104
 105         sb.append("}\n");
 106
 107         return sb.toString();
 108     }
 109
 110     public static String generateLengthChecker(final String member, final Type type, final Collection<LengthConstraint>
 111             constraints) {
 112         if (TypeUtils.getBaseYangType(type).getName().contains("[")) {
 113             return generateArrayLengthChecker(member, constraints);
 114         } else {
 115             return generateStringLengthChecker(member, constraints);
 116         }
 117     }
 118
 119     public static String generateLengthCheckerCall(@Nullable final String member, @Nonnull final String valueReference) {
 120         return lengthCheckerName(member) + '(' + valueReference + ");\n";
 121     }
 122 }