allow SelectJson to specify multiple field names, in case the server renamed something

This commit is contained in:
Adrian Cole 2013-01-30 09:03:50 -08:00 committed by adriancole
parent ff53b7161c
commit 6edaae8dce
3 changed files with 32 additions and 18 deletions

View File

@ -52,12 +52,16 @@ public class ParseFirstJsonValueNamed<T> implements Function<HttpResponse, T> {
private final GsonWrapper json;
private final TypeLiteral<T> type;
private final String name;
private final ImmutableSet<String> nameChoices;
public ParseFirstJsonValueNamed(GsonWrapper json, TypeLiteral<T> type, String name) {
/**
* @param nameChoices
* tried in order, first match wins
*/
public ParseFirstJsonValueNamed(GsonWrapper json, TypeLiteral<T> type, String... nameChoices) {
this.json = checkNotNull(json, "json");
this.type = checkNotNull(type, "type");
this.name = checkNotNull(name, "name");
this.nameChoices = ImmutableSet.copyOf(checkNotNull(nameChoices, "nameChoices"));
}
@Override
@ -71,20 +75,19 @@ public class ParseFirstJsonValueNamed<T> implements Function<HttpResponse, T> {
reader.setLenient(true);
AtomicReference<String> name = new AtomicReference<String>();
JsonToken token = reader.peek();
for (; token != JsonToken.END_DOCUMENT && nnn(this.name, reader, token, name); token = skipAndPeek(token,
reader))
;
for (; token != JsonToken.END_DOCUMENT && nnn(reader, token, name); token = skipAndPeek(token, reader)) {
}
if (name.get() == null) {
logger.trace("did not object named %s in json from response %s", this.name, arg0);
logger.trace("did not object named %s in json from response %s", nameChoices, arg0);
return nothing();
} else if (name.get().equals(this.name)) {
} else if (nameChoices.contains(name.get())) {
return json.delegate().<T> fromJson(reader, type.getType());
} else {
return nothing();
}
} catch (IOException e) {
throw new RuntimeException(String.format(
"error reading from stream, parsing object named %s from http response %s", this.name, arg0), e);
"error reading from stream, parsing object named %s from http response %s", nameChoices, arg0), e);
} finally {
Closeables.closeQuietly(reader);
arg0.getPayload().release();
@ -92,7 +95,7 @@ public class ParseFirstJsonValueNamed<T> implements Function<HttpResponse, T> {
}
@SuppressWarnings("unchecked")
protected T nothing() {
private T nothing() {
if (type.getRawType().isAssignableFrom(Set.class))
return (T) ImmutableSet.of();
else if (type.getRawType().isAssignableFrom(List.class))
@ -102,11 +105,10 @@ public class ParseFirstJsonValueNamed<T> implements Function<HttpResponse, T> {
return null;
}
protected boolean nnn(String toFind, JsonReader reader, JsonToken token, AtomicReference<String> name)
throws IOException {
private boolean nnn(JsonReader reader, JsonToken token, AtomicReference<String> name) throws IOException {
if (token == JsonToken.NAME) {
String name2 = reader.nextName();
if (toFind.equals(name2)) {
if (nameChoices.contains(name2)) {
name.set(name2);
return false;
}
@ -115,7 +117,7 @@ public class ParseFirstJsonValueNamed<T> implements Function<HttpResponse, T> {
}
public JsonToken skipAndPeek(JsonToken token, JsonReader reader) throws IOException {
private JsonToken skipAndPeek(JsonToken token, JsonReader reader) throws IOException {
switch (token) {
case BEGIN_ARRAY:
reader.beginArray();
@ -149,4 +151,4 @@ public class ParseFirstJsonValueNamed<T> implements Function<HttpResponse, T> {
}
return reader.peek();
}
}
}

View File

@ -36,9 +36,8 @@ import java.lang.annotation.Target;
public @interface SelectJson {
/**
*
* @return
* Each of the keys are tried in order. This helps in the case the server renamed a field in json.
*/
String value();
String[] value();
}

View File

@ -70,6 +70,19 @@ public class ParseFirstJsonValueNamedTest {
}, "event").apply(response);
assertEquals(val.toString(), "[(name=GREETINGS, source=guest)]");
}
/**
* scenario: server renames field from event to _event
*/
public void testParseRenamedField() throws IOException {
String nested = "{ \"count\":1 ,\"_event\" : [ {name:'GREETINGS',source:'guest'} ] }";
HttpResponse response = HttpResponse.builder().statusCode(200).message("goodie")
.payload(Payloads.newPayload(nested)).build();
List<Event> val = new ParseFirstJsonValueNamed<List<Event>>(json, new TypeLiteral<List<Event>>() {
}, "event", "_event").apply(response);
assertEquals(val.toString(), "[(name=GREETINGS, source=guest)]");
}
public void testParseNestedElementsWhenNotFoundIsEmpty() throws IOException {
String nested = "{ \"count\":1 ,\"evant\" : [ {name:'GREETINGS',source:'guest'} ] }";