Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
  /*
    *
    *  *  Copyright 2014 Orient Technologies LTD (info(at)orientechnologies.com)
    *  *
    *  *  Licensed under the Apache License, Version 2.0 (the "License");
    *  *  you may not use this file except in compliance with the License.
    *  *  You may obtain a copy of the License at
    *  *
    *  *       http://www.apache.org/licenses/LICENSE-2.0
   *  *
   *  *  Unless required by applicable law or agreed to in writing, software
   *  *  distributed under the License is distributed on an "AS IS" BASIS,
   *  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   *  *  See the License for the specific language governing permissions and
   *  *  limitations under the License.
   *  *
   *  * For more information: http://www.orientechnologies.com
   *
   */
 
 package com.tinkerpop.blueprints.impls.orient;
 
 
 import java.util.List;
 import java.util.Set;

Author(s):
Luca Garulli (http://www.orientechnologies.com)
 
 @SuppressWarnings("unchecked")
 public class OrientIndex<T extends OrientElementimplements Index<T> {
   public static final String         CONFIG_CLASSNAME       = "blueprintsIndexClass";
   public static final String         CONFIG_RECORD_MAP_NAME = "record_map_name";
   protected static final String      VERTEX                 = "Vertex";
   protected static final String      EDGE                   = "Edge";
   protected static final String      SEPARATOR              = "!=!";
 
   protected OrientBaseGraph          graph;
   protected OIndex<?>                underlying;
   protected OIndex<?>                recordKeyValueIndex;
 
   protected Class<? extends ElementindexClass;
 
   protected OrientIndex(final OrientBaseGraph graphfinal String indexNamefinal Class<? extends ElementindexClass,
       final OType iType) {
     this. = graph;
     this. = indexClass;
     create(indexNamethis.iType);
   }
 
   protected OrientIndex(final OrientBaseGraph orientGraphfinal OIndex<?> rawIndex) {
     this. = orientGraph;
     this. = rawIndex instanceof OIndexTxAwareMultiValue ? rawIndex : new OIndexTxAwareMultiValue(
         orientGraph.getRawGraph(), (OIndex<Set<OIdentifiable>>) rawIndex);
 
     final ODocument metadata = rawIndex.getMetadata();
     if (metadata == null) {
       load(rawIndex.getConfiguration());
     } else
       load(metadata);
   }
 
   public String getIndexName() {
     return .getName();
   }
 
   public Class<T> getIndexClass() {
     return (Class<T>) this.;
   }
 
   public void put(final String keyfinal Object valuefinal T element) {
     final String keyTemp = key +  + value;
 
     final ODocument doc = element.getRecord();
     if (!doc.getIdentity().isValid())
       doc.save();
 
     .put(keyTempdoc);
    .put(new OCompositeKey(doc.getIdentity(), keyTemp), doc.getIdentity());
  }
  @SuppressWarnings("rawtypes")
  public CloseableIterable<T> get(final String keyfinal Object iValue) {
    final String keyTemp = key +  + iValue;
    if (records == null || records.isEmpty())
      return new WrappingCloseableIterable(Collections.emptySet());
    return new OrientElementIterable<T>(records);
  }
  public CloseableIterable<T> query(final String keyfinal Object querythrows UnsupportedOperationException {
    throw new UnsupportedOperationException();
  }
  public long count(final String keyfinal Object value) {
    final String keyTemp = key +  + value;
    final Collection<OIdentifiablerecords = (Collection<OIdentifiable>) .get(keyTemp);
    if (records == null)
      return 0;
    return records.size();
  }
  public void remove(final String keyfinal Object valuefinal T element) {
    final String keyTemp = key +  + value;
    try {
      .remove(keyTempelement.getRecord());
      .remove(new OCompositeKey(element.getIdentity(), keyTemp), element.getIdentity());
    } catch (Exception e) {
      throw new RuntimeException(e.getMessage(), e);
    }
  }
  public String toString() {
    return StringFactory.indexString(this);
  }
  public OIndex<?> getUnderlying() {
    return ;
  }
  public void close() {
    if ( != null) {
      .flush();
       = null;
    }
     = null;
  }
  protected void removeElement(final T element) {
    final OSQLSynchQuery<ODocumentquery = new OSQLSynchQuery<ODocument>("select from index:" + .getName()
        + " where key between [" + element.getIdentity() + "] and [" + element.getIdentity() + "]");
    Collection<ODocumententries = (Collection<ODocument>) .getRawGraph().query(query);
    for (ODocument entry : entries) {
      OCompositeKey key = entry.field("key");
      List<Objectkeys = key.getKeys();
      .remove(keys.get(1).toString(), element.getIdentity());
      .remove(keyelement.getIdentity());
    }
  }
  private void create(final String indexNamefinal Class<? extends ElementindexClassOType iKeyType) {
    this. = indexClass;
    if (iKeyType == null)
      iKeyType = .;
        .getRawGraph()
        .getMetadata()
        .getIndexManager()
        .createIndex("__@recordmap@___" + indexName...toString(),
            new OSimpleKeyIndexDefinition(..), nullnullnull));
    final String className;
    if (Vertex.class.isAssignableFrom(indexClass))
      className = ;
    else if (Edge.class.isAssignableFrom(indexClass))
      className = ;
    else
      className = indexClass.getName();
    final ODocument metadata = new ODocument();
    metadata.field(className);
    // CREATE THE MAP
        .getRawGraph()
        .getMetadata()
        .getIndexManager()
        .createIndex(indexName...toString(), new OSimpleKeyIndexDefinition(iKeyType), nullnull,
            metadata));
  }
  private void load(final ODocument metadata) {
    // LOAD TREEMAP
    final String indexClassName = metadata.field();
    final String recordKeyValueMap = metadata.field();
    if (.equals(indexClassName))
      this. = Vertex.class;
    else if (.equals(indexClassName))
      this. = Edge.class;
    else
      try {
        this. = (Class<T>) Class.forName(indexClassName);
      } catch (ClassNotFoundException e) {
        throw new IllegalArgumentException("Index class '" + indexClassName
            + "' is not registered. Supported ones: Vertex, Edge and custom class that extends them");
      }
    if (recordKeyValueMap == null)
       = buildKeyValueIndex(metadata);
    else
          .getMetadata().getIndexManager().getIndex(recordKeyValueMap));
  }
  private OIndex<?> buildKeyValueIndex(ODocument metadata) {
    OIndex<?> recordKeyValueIndex = new OIndexTxAwareOneValue(.getRawGraph(), (OIndex<OIdentifiable>) 
        .getRawGraph()
        .getMetadata()
        .getIndexManager()
        .createIndex("__@recordmap@___" + .getName(), ...toString(),
            new OSimpleKeyIndexDefinition(..), nullnullnull));
    final List<ODocumententries = .getRawGraph().query(
        new OSQLSynchQuery<Object>("select  from index:" + .getName()));
    for (ODocument entry : entries) {
      final OIdentifiable rid = entry.field("rid");
      if (rid != null)
        recordKeyValueIndex.put(new OCompositeKey(ridentry.field("key")), rid);
    }
    metadata.field(recordKeyValueIndex.getName());
    return recordKeyValueIndex;
  }
New to GrepCode? Check out our FAQ X