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  }