View Javadoc

1   /**
2    * Copyright (c) 2002-2011 "Neo Technology,"
3    * Network Engine for Objects in Lund AB [http://neotechnology.com]
4    *
5    * This file is part of Neo4j.
6    *
7    * Neo4j is free software: you can redistribute it and/or modify
8    * it under the terms of the GNU General Public License as published by
9    * the Free Software Foundation, either version 3 of the License, or
10   * (at your option) any later version.
11   *
12   * This program is distributed in the hope that it will be useful,
13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15   * GNU General Public License for more details.
16   *
17   * You should have received a copy of the GNU General Public License
18   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
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  }