Search code examples
javajerseyjax-rsjava-8jersey-2.0

Serialize Java 8 Stream with Jersey


How can I serialize a Java 8 java.util.Stream<T> with Jersey. I tried to write a MessageBodyWriter, but I need to know how to compose (decorate) existing MessageBodyWriters with a new MessageBodyWriter for my Stream.

Stream<String> get(){
  return some stream of strings  
}

public <T> class StreamMessageBodyWriter<Stream<T>> 
           implements MessageBodyWriter<Stream<T>> {

  public void writeTo(.......){
    //How can I get the handle to MessageBodyWriter that will write for type T, 
    //so that I can 'collect' the 'java.util.Stream<T>' and write it to 
    //OutputStream
  }
}

Solution

  • but I need to know how to compose (decorate) existing MessageBodyWriters with a new MessageBodyWriter for my Stream

    You can just inject Providers and use getMessagBodyWriter(...), passing in the required details to lookup the specific writer for that type. For example

    @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);
            
        }
    }
    

    If you look at the writeTo, first I call collect then get the returned type. Then lookup the writer for that type, then simply delegate to the writer.

    Here is a test

    @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
    Result:
    [{"name":"Stack"},{"name":"Sam"}]

    As an aside, if you plan on manipulating the Stream in the writer, maybe look into using an Interceptor. Won't make a difference really, but if you want to stick to the Single Responsibility Principle, this is what the Interceptor is for, manipulating the request body.


    Note: the above is standard JAX-RS

    Alternatively...

    Specifically with Jersey, you can also inject MessageBodyWorkers, for more specific lookup, and even calling its writeTo, which will delegate to the required writer, if one exsists.