added a Utils.setObjectByPath() method

This commit is contained in:
Noble Paul 2017-06-23 14:17:14 +09:30
parent 78731ea8f4
commit 4676410689
2 changed files with 85 additions and 3 deletions

View File

@ -201,6 +201,29 @@ public class TestUtils extends SolrTestCaseJ4 {
assertEquals( "v1" ,Utils.getObjectByPath(m, true, "/a/d[0]/k1")); assertEquals( "v1" ,Utils.getObjectByPath(m, true, "/a/d[0]/k1"));
assertEquals( "v2" ,Utils.getObjectByPath(m, true, "/a/d[1]/k2")); assertEquals( "v2" ,Utils.getObjectByPath(m, true, "/a/d[1]/k2"));
} }
public void testSetObjectByPath(){
String json = "{\n" +
" 'authorization':{\n" +
" 'class':'solr.RuleBasedAuthorizationPlugin',\n" +
" 'user-role':{\n" +
" 'solr':'admin',\n" +
" 'harry':'admin'},\n" +
" 'permissions':[{\n" +
" 'name':'security-edit',\n" +
" 'role':['admin']},\n" +
" {\n" +
" 'name':'x-update',\n" +
" 'collection':'x',\n" +
" 'path':'/update/*',\n" +
" 'role':'dev'}],\n" +
" '':{'v':4}}}";
Map m = (Map) fromJSONString(json);
Utils.setObjectByPath(m,"authorization/permissions[1]/role","guest");
Utils.setObjectByPath(m,"authorization/permissions[0]/role[-1]","dev");
assertEquals("guest", Utils.getObjectByPath(m,true,"authorization/permissions[1]/role"));
assertEquals("dev", Utils.getObjectByPath(m,true,"authorization/permissions[0]/role[1]"));
}
public void testUtilsJSPath(){ public void testUtilsJSPath(){

View File

@ -213,13 +213,72 @@ public class Utils {
} }
public static Object getObjectByPath(Object root, boolean onlyPrimitive, String hierarchy) { public static Object getObjectByPath(Object root, boolean onlyPrimitive, String hierarchy) {
if(root == null) return null;
if(!isMapLike(root)) throw new RuntimeException("must be a Map or NamedList");
List<String> parts = StrUtils.splitSmart(hierarchy, '/'); List<String> parts = StrUtils.splitSmart(hierarchy, '/');
if (parts.get(0).isEmpty()) parts.remove(0); if (parts.get(0).isEmpty()) parts.remove(0);
return getObjectByPath(root, onlyPrimitive, parts); return getObjectByPath(root, onlyPrimitive, parts);
} }
public static boolean setObjectByPath(Object root, String hierarchy, Object value) {
List<String> parts = StrUtils.splitSmart(hierarchy, '/');
if (parts.get(0).isEmpty()) parts.remove(0);
return setObjectByPath(root, parts, value);
}
public static boolean setObjectByPath(Object root, List<String> hierarchy, Object value) {
if (root == null) return false;
if (!isMapLike(root)) throw new RuntimeException("must be a Map or NamedList");
Object obj = root;
for (int i = 0; i < hierarchy.size(); i++) {
int idx = -2; //-1 means append to list, -2 means not found
String s = hierarchy.get(i);
if (s.endsWith("]")) {
Matcher matcher = ARRAY_ELEMENT_INDEX.matcher(s);
if (matcher.find()) {
s = matcher.group(1);
idx = Integer.parseInt(matcher.group(2));
}
}
if (i < hierarchy.size() - 1) {
Object o = getVal(obj, s);
if (o == null) return false;
if (idx > -1) {
List l = (List) o;
o = idx < l.size() ? l.get(idx) : null;
}
if (!isMapLike(o)) return false;
obj = o;
} else {
if (idx == -2) {
if (obj instanceof NamedList) {
NamedList namedList = (NamedList) obj;
int location = namedList.indexOf(s, 0);
if (location == -1) namedList.add(s, value);
else namedList.setVal(location, value);
} else if (obj instanceof Map) {
((Map) obj).put(s, value);
}
return true;
} else {
Object v = getVal(obj, s);
if (v instanceof List) {
List list = (List) v;
if (idx == -1) {
list.add(value);
} else {
if (idx < list.size()) list.set(idx, value);
else return false;
}
return true;
} else {
return false;
}
}
}
}
return false;
}
public static Object getObjectByPath(Object root, boolean onlyPrimitive, List<String> hierarchy) { public static Object getObjectByPath(Object root, boolean onlyPrimitive, List<String> hierarchy) {
@ -326,7 +385,7 @@ public class Utils {
} }
public static final Pattern ARRAY_ELEMENT_INDEX = Pattern public static final Pattern ARRAY_ELEMENT_INDEX = Pattern
.compile("(\\S*?)\\[(\\d+)\\]"); .compile("(\\S*?)\\[([-]?\\d+)\\]");
public static SpecProvider getSpec(final String name) { public static SpecProvider getSpec(final String name) {
return () -> { return () -> {