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.repr;
    21
    22  import java.util.ArrayList;
    23  import java.util.HashMap;
    24  import java.util.List;
    25  import java.util.Map;
    26
    27  public final class ServerExtensionRepresentation extends MappingRepresentation
    28  {
    29      private final Map<String, EntityExtensionRepresentation> extended;
    30
    31      public ServerExtensionRepresentation( String name, List<ExtensionPointRepresentation> methods )
    32      {
    33          super( RepresentationType.SERVER_PLUGIN_DESCRIPTION );
    34          this.extended = new HashMap<String, EntityExtensionRepresentation>();
    35          for ( ExtensionPointRepresentation extension : methods )
    36          {
    37              EntityExtensionRepresentation entity = extended.get( extension.getExtendedEntity() );
    38              if ( entity == null )
    39              {
    40                  extended.put( extension.getExtendedEntity(),
    41                          entity = new EntityExtensionRepresentation() );
    42              }
    43              entity.add( extension );
    44          }
    45      }
    46
    47      @Override
    48      protected void serialize( MappingSerializer serializer )
    49      {
    50          for ( Map.Entry<String, EntityExtensionRepresentation> entity : extended.entrySet() )
    51          {
    52              serializer.putMapping( entity.getKey(), entity.getValue() );
    53          }
    54      }
    55
    56      private static class EntityExtensionRepresentation extends MappingRepresentation
    57      {
    58          private final List<ExtensionPointRepresentation> extensions;
    59
    60          EntityExtensionRepresentation()
    61          {
    62              super( "entity-extensions" );
    63              this.extensions = new ArrayList<ExtensionPointRepresentation>();
    64          }
    65
    66          void add( ExtensionPointRepresentation extension )
    67          {
    68              extensions.add( extension );
    69          }
    70
    71          @Override
    72          protected void serialize( MappingSerializer serializer )
    73          {
    74              for ( ExtensionPointRepresentation extension : extensions )
    75              {
    76                  serializer.putMapping( extension.getName(), extension );
    77              }
    78          }
    79      }
    80  }