但我需要知道如何为我的Stream 用新的MessageBodyWriter 组合(装饰)现有的MessageBodyWriters
您可以只注入Providers 并使用getMessagBodyWriter(...),传入所需的详细信息以查找该类型的特定编写器。例如
@Provider
public class StreamBodyWriter implements MessageBodyWriter<Stream> {
@Context
private Providers providers;
@Override
public boolean isWriteable(Class<?> type, Type genericType,
Annotation[] annotations, MediaType mediaType) {
return Stream.class.isAssignableFrom(type);
}
@Override
public long getSize(Stream stream, Class<?> type, Type genericType,
Annotation[] annotations, MediaType mediaType) { return -1; }
@Override
public void writeTo(Stream stream, Class<?> type, Type genericType,
Annotation[] annotations, MediaType mediaType,
MultivaluedMap<String, Object> httpHeaders, OutputStream entityStream)
throws IOException, WebApplicationException {
Object obj = stream.collect(Collectors.toList());
Class<?> objType = obj.getClass();
MessageBodyWriter writer = providers.getMessageBodyWriter(objType,
null, annotations, mediaType);
writer.writeTo(obj, objType, null, annotations,
mediaType, httpHeaders, entityStream);
}
}
如果您查看writeTo,首先我调用collect,然后获取返回的类型。然后查找该类型的编写器,然后简单地委托给编写器。
这是一个测试
@Path("stream")
public class StreamResource {
@GET
@Produces(MediaType.APPLICATION_JSON)
public Response getStream() {
List<Person> myList = Arrays.asList(
new Person("Stack"),
new Person("Overflow"),
new Person("Sam"));
Stream<Person> stream = myList.stream()
.filter(p -> p.name.startsWith("S"));
return Response.ok(stream).build();
}
public static class Person {
public String name;
public Person(String name) { this.name = name; }
public Person() {}
}
}
C:\>curl -v http://localhost:8080/api/stream
结果:
[{"name":"Stack"},{"name":"Sam"}]
顺便说一句,如果您打算在 writer 中操作 Stream,不妨考虑使用 Interceptor。不会有什么不同,但如果你想坚持单一责任原则,这就是Interceptor 的用途,操纵请求正文。
注意:以上是标准的 JAX-RS
或者...
特别是对于 Jersey,您还可以注入 MessageBodyWorkers,以进行更具体的查找,甚至调用它的 writeTo,如果存在,它将委托给所需的作者。