Skip to content

Commit bf8ca4b

Browse files
committed
[Fix][Common] Problems with comma-separated logic in variable values during variable substitution.(#9151)
1 parent 879b1e2 commit bf8ca4b

File tree

2 files changed

+51
-1
lines changed

2 files changed

+51
-1
lines changed

seatunnel-core/seatunnel-core-starter/src/main/java/org/apache/seatunnel/core/starter/command/ParameterSplitter.java

+13-1
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,21 @@ public List<String> split(String value) {
5050
}
5151

5252
if (currentToken.length() > 0) {
53-
result.add(currentToken.toString().trim());
53+
String token = processToken(currentToken.toString().trim());
54+
result.add(token);
5455
}
5556

5657
return result;
5758
}
59+
60+
private String processToken(String token) {
61+
int eqIdx = token.indexOf('=');
62+
if (eqIdx <= 0 || eqIdx == token.length() - 1) {
63+
return token;
64+
}
65+
String key = token.substring(0, eqIdx);
66+
String val = token.substring(eqIdx + 1).trim();
67+
val = val.replaceAll("^\"|\"$", "");
68+
return String.format("%s=%s", key, val);
69+
}
5870
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
/*
2+
* Licensed to the Apache Software Foundation (ASF) under one or more
3+
* contributor license agreements. See the NOTICE file distributed with
4+
* this work for additional information regarding copyright ownership.
5+
* The ASF licenses this file to You under the Apache License, Version 2.0
6+
* (the "License"); you may not use this file except in compliance with
7+
* the License. You may obtain a copy of the License at
8+
*
9+
* http://www.apache.org/licenses/LICENSE-2.0
10+
*
11+
* Unless required by applicable law or agreed to in writing, software
12+
* distributed under the License is distributed on an "AS IS" BASIS,
13+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14+
* See the License for the specific language governing permissions and
15+
* limitations under the License.
16+
*/
17+
18+
package org.apache.seatunnel.core.starter.utils;
19+
20+
import org.apache.seatunnel.core.starter.command.ParameterSplitter;
21+
22+
import org.junit.jupiter.api.Test;
23+
24+
import static org.junit.jupiter.api.Assertions.assertEquals;
25+
26+
public class ParameterSplitterTest {
27+
28+
@Test
29+
public void testSplit() {
30+
String value = "columns=\"c_id,c_area_code,c_date\"";
31+
32+
ParameterSplitter parameterSplitter = new ParameterSplitter();
33+
List<String> result = parameterSplitter.split(value);
34+
35+
assertEquals(1, result.size());
36+
assertEquals("columns=c_id,c_area_code,c_date", result.get(0));
37+
}
38+
}

0 commit comments

Comments
 (0)