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  }