2 * Copyright (c) 2017 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.mdsal.binding.javav2.java.api.generator.range_generators;
10 import com.google.common.collect.Range;
11 import java.util.ArrayList;
12 import java.util.Collection;
14 import javax.annotation.Nonnull;
15 import javax.annotation.Nullable;
16 import org.opendaylight.mdsal.binding.javav2.model.api.Type;
17 import org.opendaylight.yangtools.yang.model.api.type.LengthConstraint;
19 public final class LengthGenerator {
20 private LengthGenerator() {
21 throw new UnsupportedOperationException();
24 private static String lengthCheckerName(final String member) {
25 return "check" + member + "Length";
28 private static Collection<String> createExpressions(final LengthConstraint constraint) {
29 final Set<Range<Integer>> constraints = constraint.getAllowedRanges().asRanges();
30 final Collection<String> ret = new ArrayList<>(constraints.size());
32 for (Range<Integer> l : constraints) {
33 final StringBuilder sb = new StringBuilder("length >");
35 // We have to deal with restrictions being out of integer's range
36 if (l.lowerEndpoint() <= Integer.MAX_VALUE) {
39 sb.append(' ').append(l.lowerEndpoint());
41 final int max = l.upperEndpoint();
42 if (max < Integer.MAX_VALUE) {
43 sb.append(" && length <= ").append(max);
46 ret.add(sb.toString());
52 private static String createLengthString(final LengthConstraint constraint) {
53 return new ArrayList<>(constraint.getAllowedRanges().asRanges()).toString();
56 private static String generateArrayLengthChecker(final String member, final LengthConstraint constraint) {
57 final StringBuilder sb = new StringBuilder();
58 final Collection<String> expressions = createExpressions(constraint);
60 sb.append("private static void ").append(lengthCheckerName(member)).append("(final byte[] value) {\n");
62 if (!expressions.isEmpty()) {
63 sb.append(" final int length = value.length;\n");
65 for (String exp : expressions) {
66 sb.append(" if (").append(exp).append(") {\n");
67 sb.append(" return;\n");
71 sb.append(" throw new IllegalArgumentException(String.format(\"Invalid length: %s, expected: ")
72 .append(createLengthString(constraint)).append(".\", java.util.Arrays.toString(value)));\n");
80 private static String generateStringLengthChecker(final String member, final LengthConstraint constraint) {
81 final StringBuilder sb = new StringBuilder();
82 final Collection<String> expressions = createExpressions(constraint);
84 sb.append("private static void ").append(lengthCheckerName(member)).append("(final String value) {\n");
86 if (!expressions.isEmpty()) {
87 sb.append(" final int length = value.length();\n");
89 for (String exp : expressions) {
90 sb.append(" if (").append(exp).append(") {\n");
91 sb.append(" return;\n");
95 sb.append(" throw new IllegalArgumentException(String.format(\"Invalid length: %s, expected: ")
96 .append(createLengthString(constraint)).append(".\", value));\n");
101 return sb.toString();
104 public static String generateLengthChecker(final String member, final Type type,
105 final LengthConstraint constraint) {
106 if (TypeUtils.getBaseYangType(type).getName().contains("[")) {
107 return generateArrayLengthChecker(member, constraint);
109 return generateStringLengthChecker(member, constraint);
113 public static String generateLengthCheckerCall(@Nullable final String member,
114 @Nonnull final String valueReference) {
115 return lengthCheckerName(member) + '(' + valueReference + ");\n";