2 * Copyright (c) 2015 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.java.api.generator;
10 import com.google.common.collect.Range;
11 import java.util.ArrayList;
12 import java.util.Collection;
14 import org.eclipse.jdt.annotation.NonNull;
15 import org.eclipse.jdt.annotation.Nullable;
16 import org.opendaylight.mdsal.binding.model.api.Type;
17 import org.opendaylight.mdsal.binding.model.ri.Types;
18 import org.opendaylight.yangtools.yang.model.api.type.LengthConstraint;
19 import org.slf4j.Logger;
20 import org.slf4j.LoggerFactory;
22 final class LengthGenerator {
23 private static final Logger LOG = LoggerFactory.getLogger(LengthGenerator.class);
25 private LengthGenerator() {
29 private static String lengthCheckerName(final String member) {
30 return "check" + member + "Length";
33 private static Collection<String> createExpressions(final LengthConstraint constraint) {
34 final Set<Range<Integer>> constraints = constraint.getAllowedRanges().asRanges();
35 final Collection<String> ret = new ArrayList<>(constraints.size());
37 for (Range<Integer> l : constraints) {
38 // We have to deal with restrictions being out of integer's range
39 final String expr = createExpression(l.lowerEndpoint(), l.upperEndpoint());
41 // This range is implicitly capped by String/byte[] length returns
42 LOG.debug("Constraint {} implied by int type value domain, skipping", l);
51 private static @Nullable String createExpression(final int min, final int max) {
53 return min < Integer.MAX_VALUE ? "length == " + min : null;
56 final StringBuilder sb = new StringBuilder("length >= ").append(min);
57 if (max < Integer.MAX_VALUE) {
58 sb.append(" && length <= ").append(max);
63 return max < Integer.MAX_VALUE ? "length <= " + max : null;
66 private static String createLengthString(final LengthConstraint constraint) {
67 return new ArrayList<>(constraint.getAllowedRanges().asRanges()).toString();
70 private static String generateArrayLengthChecker(final String member, final LengthConstraint constraint,
71 final JavaFileTemplate template) {
72 final StringBuilder sb = new StringBuilder();
73 final Collection<String> expressions = createExpressions(constraint);
75 sb.append("private static void ").append(lengthCheckerName(member)).append("(final byte[] value) {\n");
77 if (!expressions.isEmpty()) {
78 sb.append(" final int length = value.length;\n");
80 for (String exp : expressions) {
81 sb.append(" if (").append(exp).append(") {\n");
82 sb.append(" return;\n");
86 sb.append(" ").append(template.importedName(JavaFileTemplate.CODEHELPERS))
87 .append(".throwInvalidLength(\"").append(createLengthString(constraint)).append("\", value);\n");
90 return sb.append("}\n").toString();
93 private static String generateStringLengthChecker(final String member, final LengthConstraint constraint,
94 final JavaFileTemplate template) {
95 final StringBuilder sb = new StringBuilder();
96 final Collection<String> expressions = createExpressions(constraint);
98 sb.append("private static void ").append(lengthCheckerName(member))
99 .append("(final ").append(template.importedName(Types.STRING)).append(" value) {\n");
101 if (!expressions.isEmpty()) {
102 sb.append(" final int length = value.codePointCount(0, value.length());\n");
104 for (String exp : expressions) {
105 sb.append(" if (").append(exp).append(") {\n");
106 sb.append(" return;\n");
110 sb.append(" ").append(template.importedName(JavaFileTemplate.CODEHELPERS))
111 .append(".throwInvalidLength(\"").append(createLengthString(constraint)).append("\", value);\n");
114 return sb.append("}\n").toString();
117 static String generateLengthChecker(final String member, final @NonNull Type type,
118 final LengthConstraint constraint, final JavaFileTemplate template) {
119 return TypeUtils.getBaseYangType(type).getName().indexOf('[') != -1
120 ? generateArrayLengthChecker(member, constraint, template)
121 : generateStringLengthChecker(member, constraint, template);
124 static String generateLengthCheckerCall(final @Nullable String member, final @NonNull String valueReference) {
125 return lengthCheckerName(member) + '(' + valueReference + ");\n";