Skip to content

Commit ad89c68

Browse files
authored
Add $.fromXmlMakeArrays(xml) method to lodash module.
1 parent 167bd7d commit ad89c68

File tree

2 files changed

+111
-4
lines changed

2 files changed

+111
-4
lines changed

src/main/java/com/github/underscore/lodash/$.java

Lines changed: 26 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2519,15 +2519,16 @@ private static Object getValue(final Object value) {
25192519
}
25202520

25212521
@SuppressWarnings("unchecked")
2522-
private static Map<String, Object> createMap(final org.w3c.dom.Node node) {
2522+
private static Map<String, Object> createMap(final org.w3c.dom.Node node,
2523+
final Function<Object, Object> nodeMapper) {
25232524
final Map<String, Object> map = newLinkedHashMap();
25242525
final org.w3c.dom.NodeList nodeList = node.getChildNodes();
25252526
for (int index = 0; index < nodeList.getLength(); index++) {
25262527
final org.w3c.dom.Node currentNode = nodeList.item(index);
25272528
final String name = currentNode.getNodeName();
25282529
final Object value;
25292530
if (currentNode.getNodeType() == org.w3c.dom.Node.ELEMENT_NODE) {
2530-
value = createMap(currentNode);
2531+
value = createMap(currentNode, nodeMapper);
25312532
} else {
25322533
value = currentNode.getTextContent();
25332534
}
@@ -2545,7 +2546,7 @@ private static Map<String, Object> createMap(final org.w3c.dom.Node node) {
25452546
map.put(name, objects);
25462547
}
25472548
} else {
2548-
map.put(name, getValue(value));
2549+
map.put(name, nodeMapper.apply(getValue(value)));
25492550
}
25502551
}
25512552
return map;
@@ -2558,7 +2559,28 @@ public static Object fromXml(final String xml) {
25582559
javax.xml.parsers.DocumentBuilderFactory.newInstance();
25592560
factory.setNamespaceAware(true);
25602561
final org.w3c.dom.Document document = factory.newDocumentBuilder().parse(stream);
2561-
return createMap(document);
2562+
return createMap(document, new Function<Object, Object>() {
2563+
public Object apply(Object object) {
2564+
return object;
2565+
}
2566+
});
2567+
} catch (Exception ex) {
2568+
throw new IllegalArgumentException(ex);
2569+
}
2570+
}
2571+
2572+
public static Object fromXmlMakeArrays(final String xml) {
2573+
try {
2574+
final java.io.InputStream stream = new java.io.ByteArrayInputStream(xml.getBytes("UTF-8"));
2575+
final javax.xml.parsers.DocumentBuilderFactory factory =
2576+
javax.xml.parsers.DocumentBuilderFactory.newInstance();
2577+
factory.setNamespaceAware(true);
2578+
final org.w3c.dom.Document document = factory.newDocumentBuilder().parse(stream);
2579+
return createMap(document, new Function<Object, Object>() {
2580+
public Object apply(Object object) {
2581+
return object instanceof List ? object : newArrayList(Arrays.asList(object));
2582+
}
2583+
});
25622584
} catch (Exception ex) {
25632585
throw new IllegalArgumentException(ex);
25642586
}

src/test/java/com/github/underscore/lodash/StringTest.java

Lines changed: 85 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1365,6 +1365,86 @@ public void toXml() {
13651365
+ "\n</glossary>", $.toXml((Map<String, Object>) $.fromJson(string)));
13661366
}
13671367

1368+
@SuppressWarnings("unchecked")
1369+
@Test
1370+
public void fromXmlMakeArrays() {
1371+
String string =
1372+
"<?xml version=\"1.0\" encoding=\"utf-8\"?>"
1373+
+ "\n"
1374+
+ "\n<root>"
1375+
+ "\n <list>"
1376+
+ "\n <item>"
1377+
+ "\n <key1>value</key1>"
1378+
+ "\n <key2>value</key2>"
1379+
+ "\n <key3>value</key3>"
1380+
+ "\n </item>"
1381+
+ "\n <item>"
1382+
+ "\n <key1>value</key1>"
1383+
+ "\n <key2>value</key2>"
1384+
+ "\n <key3>value</key3>"
1385+
+ "\n </item>"
1386+
+ "\n </list>"
1387+
+ "\n <list>"
1388+
+ "\n <item>"
1389+
+ "\n <key1>value</key1>"
1390+
+ "\n <key2>value</key2>"
1391+
+ "\n <key3>value</key3>"
1392+
+ "\n </item>"
1393+
+ "\n </list>"
1394+
+ "\n</root>";
1395+
assertEquals(
1396+
"{\n"
1397+
+ " \"root\": [\n"
1398+
+ " {\n"
1399+
+ " \"list\": [\n"
1400+
+ " {\n"
1401+
+ " \"item\": [\n"
1402+
+ " {\n"
1403+
+ " \"key1\": [\n"
1404+
+ " \"value\"\n"
1405+
+ " ],\n"
1406+
+ " \"key2\": [\n"
1407+
+ " \"value\"\n"
1408+
+ " ],\n"
1409+
+ " \"key3\": [\n"
1410+
+ " \"value\"\n"
1411+
+ " ]\n"
1412+
+ " },\n"
1413+
+ " {\n"
1414+
+ " \"key1\": [\n"
1415+
+ " \"value\"\n"
1416+
+ " ],\n"
1417+
+ " \"key2\": [\n"
1418+
+ " \"value\"\n"
1419+
+ " ],\n"
1420+
+ " \"key3\": [\n"
1421+
+ " \"value\"\n"
1422+
+ " ]\n"
1423+
+ " }\n"
1424+
+ " ]\n"
1425+
+ " },\n"
1426+
+ " {\n"
1427+
+ " \"item\": [\n"
1428+
+ " {\n"
1429+
+ " \"key1\": [\n"
1430+
+ " \"value\"\n"
1431+
+ " ],\n"
1432+
+ " \"key2\": [\n"
1433+
+ " \"value\"\n"
1434+
+ " ],\n"
1435+
+ " \"key3\": [\n"
1436+
+ " \"value\"\n"
1437+
+ " ]\n"
1438+
+ " }\n"
1439+
+ " ]\n"
1440+
+ " }\n"
1441+
+ " ]\n"
1442+
+ " }\n"
1443+
+ " ]\n"
1444+
+ "}",
1445+
$.toJson((Map<String, Object>) $.fromXmlMakeArrays(string)));
1446+
}
1447+
13681448
@SuppressWarnings("unchecked")
13691449
@Test
13701450
public void fromXml() {
@@ -1488,6 +1568,11 @@ public void testDecodeParseXmlErr13() {
14881568
$.fromXml("[\"abc\u0010\"]");
14891569
}
14901570

1571+
@Test(expected = IllegalArgumentException.class)
1572+
public void testDecodeParseXmlErr14() {
1573+
$.fromXmlMakeArrays("[\"abc\u0010\"]");
1574+
}
1575+
14911576
@SuppressWarnings("unchecked")
14921577
@Test
14931578
public void main() throws Exception {

0 commit comments

Comments
 (0)