github-repos/testable-mock
1
0
Fork 0
mirror of https://github.com/alibaba/testable-mock.git synced 2025-01-09 20:00:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix omni get path with index

Browse Source
This commit is contained in:
金戟 2021-03-20 18:02:01 +08:00
parent 7b9a0964bc
commit bbc41f2987
2 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
testable-core/src/main/java/com/alibaba/testable/core/tool/OmniAccessor.java
View File

@ -1,8 +1,8 @@
package com.alibaba.testable.core.tool;
import com.alibaba.testable.core.util.CollectionUtil;
import com.alibaba.testable.core.util.FixSizeMap;
import com.alibaba.testable.core.util.TypeUtil;
import com.sun.deploy.util.StringUtils;
import java.lang.reflect.Array;
import java.lang.reflect.Field;
@ -138,7 +138,7 @@ public class OmniAccessor {
for (int i = 0; i < querySegments.length; i++) {
patternSegments[i] = toSinglePattern(querySegments[i]);
}
return PATTERN_PREFIX + StringUtils.join(Arrays.asList(patternSegments), SLASH);
return PATTERN_PREFIX + CollectionUtil.join(Arrays.asList(patternSegments), SLASH);
}
private static String toSinglePattern(String querySegment) {
@ -198,22 +198,29 @@ public class OmniAccessor {
int nth = extraIndexFromQuery(querySegments[n]);
return Collections.singletonList((T)(nth > 0 ? Array.get(target, nth) : target));
}
String name = extraNameFromMemberRecord(memberSegments[n]);
int nth = extraIndexFromQuery(querySegments[n]);
String fieldName = extraNameFromMemberRecord(memberSegments[n]);
List<T> nexts = new ArrayList<T>();
if (target.getClass().isArray()) {
for (int i = 0; i < Array.getLength(target); i++) {
List<T> all = getBySegment(getField(Array.get(target, i), name, nth), memberSegments, querySegments, n + 1);
if (nth < 0) {
for (int i = 0; i < Array.getLength(target); i++) {
List<T> all = getBySegment(getFieldValue(Array.get(target, i), fieldName, i),
memberSegments, querySegments, n + 1);
nexts.addAll(all);
}
} else {
List <T> all = getBySegment(getFieldValue(Array.get(target, nth), fieldName, nth),
memberSegments, querySegments, n + 1);
nexts.addAll(all);
}
} else {
List <T> all = getBySegment(getField(target, name, nth), memberSegments, querySegments, n + 1);
List <T> all = getBySegment(getFieldValue(target, fieldName, nth), memberSegments, querySegments, n + 1);
nexts.addAll(all);
}
return nexts;
}
private static Object getField(Object obj, String name, int nth) throws IllegalAccessException {
private static Object getFieldValue(Object obj, String name, int nth) throws IllegalAccessException {
if (obj == null) {
return null;
}

13
testable-core/src/main/java/com/alibaba/testable/core/util/CollectionUtil.java
View File

@ -1,5 +1,8 @@
package com.alibaba.testable.core.util;
import java.util.Collection;
import java.util.Iterator;
public class CollectionUtil {
public static Object[] slice(Object[] args, int pos) {
@ -12,4 +15,14 @@ public class CollectionUtil {
return slicedArgs;
}
public static String join(Collection<?> collection, String joinSymbol) {
StringBuilder sb = new StringBuilder();
for(Iterator<?> i = collection.iterator(); i.hasNext(); sb.append((String)i.next())) {
if (sb.length() != 0) {
sb.append(joinSymbol);
}
}
return sb.toString();
}
}