1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.neo4j.server.rest.web;
21
22 import org.neo4j.server.rest.repr.*;
23
24 import javax.ws.rs.core.Response;
25 import java.net.URI;
26 import java.util.List;
27 import java.util.Map;
28
29 public class EntityOutputFormat extends OutputFormat
30 {
31 private Representation representation;
32
33 public EntityOutputFormat( RepresentationFormat format, URI baseUri, ExtensionInjector extensions )
34 {
35 super( format, baseUri, extensions);
36 }
37
38 @Override
39 protected Response response( Response.ResponseBuilder response, Representation representation )
40 {
41 this.representation = representation;
42
43 return super.response(response, representation);
44 }
45
46 public Map<String,Object> getResultAsMap() throws BadInputException
47 {
48 return (Map<String, Object>)RepresentationTestBase.serialize( representation );
49 }
50
51 public Representation getRepresentation()
52 {
53 return representation;
54 }
55
56 public List<Object> getResultAsList() throws BadInputException
57 {
58 return (List<Object>)RepresentationTestBase.serialize( representation );
59 }
60 }