Github user arunma commented on a diff in the pull request:
https://github.com/apache/nifi/pull/2172#discussion_r140890957
--- Diff:
nifi-nar-bundles/nifi-standard-services/nifi-record-serialization-services-bundle/nifi-record-serialization-services/src/main/java/org/apache/nifi/csv/CSVValidators.java
---
@@ -0,0 +1,109 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.nifi.csv;
+
+import org.apache.commons.lang3.StringEscapeUtils;
+import org.apache.nifi.components.ValidationContext;
+import org.apache.nifi.components.ValidationResult;
+import org.apache.nifi.components.Validator;
+
+import java.util.HashSet;
+import java.util.Set;
+
+public class CSVValidators {
+
+ public static class SingleCharacterValidator implements Validator {
+ private static final Set<String> illegalChars = new HashSet<>();
+
+ static {
+ illegalChars.add("\r");
+ illegalChars.add("\n");
+ }
+
+ @Override
+ public ValidationResult validate(final String subject, final
String input, final ValidationContext context) {
+
+ if (input == null) {
+ return new ValidationResult.Builder()
+ .input(input)
+ .subject(subject)
+ .valid(false)
+ .explanation("Input is null for this property")
+ .build();
+ }
+
+ final String unescaped = CSVUtils.unescape(input);
+ if (unescaped.length() != 1) {
+ return new ValidationResult.Builder()
+ .input(input)
+ .subject(subject)
+ .valid(false)
+ .explanation("Value must be exactly 1 character
but was " + input.length() + " in length")
+ .build();
+ }
+
+ if (illegalChars.contains(unescaped)) {
+ return new ValidationResult.Builder()
+ .input(input)
+ .subject(subject)
+ .valid(false)
+ .explanation(input + " is not a valid character
for this property")
+ .build();
+ }
+
+ return new ValidationResult.Builder()
+ .input(input)
+ .subject(subject)
+ .valid(true)
+ .build();
+ }
+
+ }
+
+ public static final Validator UNESCAPED_SINGLE_CHAR_VALIDATOR = new
Validator() {
+ @Override
+ public ValidationResult validate(final String subject, final
String input, final ValidationContext context) {
+
+ if (input == null) {
+ return new ValidationResult.Builder()
+ .input(input)
+ .subject(subject)
+ .valid(false)
+ .explanation("Input is null for this property")
+ .build();
+ }
+
+ String unescapeString = unescapeString(input);
+
+ return new ValidationResult.Builder()
+ .subject(subject)
+ .input(unescapeString)
+ .explanation("Only non-null single characters are
supported")
+ .valid((input.length() == 1 && input.charAt(0) != 0)
|| context.isExpressionLanguagePresent(input))
--- End diff --
@mattyb149 Terribly sorry about this. This is super embarrassing. Was
wondering how the testcases failed me. It should have been "\\u0001" instead of
"\u0001". Fixed the testcase and the class. Will run a manual test on the UI
as well before pushing the changes.
---
