org.apache.directory.server.core.api.partition.Partition Java Examples

/**
 * {@inheritDoc}
 */
public boolean hasEntry( HasEntryOperationContext hasEntryContext ) throws LdapException
{
    Dn dn = hasEntryContext.getDn();

    if ( IS_DEBUG )
    {
        LOG.debug( "Check if Dn '" + dn + "' exists." );
    }

    if ( dn.isRootDse() )
    {
        return true;
    }

    Partition partition = getPartition( dn );

    return partition.hasEntry( hasEntryContext );
}
 

/**
 * {@inheritDoc}
 */
public void modify( ModifyOperationContext modifyContext ) throws LdapException
{
    // Special case : if we don't have any modification to apply, just return
    if ( modifyContext.getModItems().size() == 0 )
    {
        return;
    }

    Partition partition = getPartition( modifyContext.getDn() );

    partition.modify( modifyContext );

    if ( modifyContext.isPushToEvtInterceptor() )
    {
        directoryService.getInterceptor( InterceptorEnum.EVENT_INTERCEPTOR.getName() ).modify( modifyContext );
    }
}
 

/**
 * Add a new partition to the directory server.
 *
 * @param partitionName - The name of the partition.
 * @param indexes - The attributes to index.
 * @return This Builder for subsequent changes.
 */
public Builder addPartition(final String id, final String partitionName, final int indexSize, final String ... indexes) throws Exception {
    assertNotStarted();
    if (directoryService == null) {
        throw new IllegalStateException("The Directory service has not been created.");
    }

    SchemaManager schemaManager = directoryService.getSchemaManager();
    PartitionFactory partitionFactory = directoryServiceFactory.getPartitionFactory();
    Partition partition = partitionFactory.createPartition(schemaManager, directoryService.getDnFactory(), id, partitionName, 1000, workingDir);
    for (String current : indexes) {
        partitionFactory.addIndex(partition, current, indexSize);
    }
    partition.setCacheService(directoryService.getCacheService());
    partition.initialize();
    directoryService.addPartition(partition);

    return this;
}
 

private void initSystemPartition() throws Exception {
    JdbmPartitionFactory partitionFactory = new JdbmPartitionFactory();

    Partition systemPartition = partitionFactory.createPartition(
            service.getSchemaManager(),
            service.getDnFactory(),
            "system",
            ServerDNConstants.SYSTEM_DN,
            500,
            new File(service.getInstanceLayout().getPartitionsDirectory(), "system"));
    systemPartition.setSchemaManager(service.getSchemaManager());

    partitionFactory.addIndex(systemPartition, SchemaConstants.OBJECT_CLASS_AT, 100);

    service.setSystemPartition(systemPartition);
}
 

/**
 * {@inheritDoc}
 */
public Entry delete( DeleteOperationContext deleteContext ) throws LdapException
{
    Partition partition = getPartition( deleteContext.getDn() );
    Entry deletedEntry = partition.delete( deleteContext );

    Entry entry = deleteContext.getEntry();
    
    
    // MyVD doesn't care about csn
    // Attribute csn = entry.get( ENTRY_CSN_AT );
    // can be null while doing subentry deletion
    // //TODO verify if this gets in the way of replication
    // if ( csn != null )
    // {
    //     directoryService.setContextCsn( csn.getString() );
    // }

    return deletedEntry;
}
 

/**
 * {@inheritDoc}
 */
public boolean hasEntry( HasEntryOperationContext hasEntryContext ) throws LdapException
{
    Dn dn = hasEntryContext.getDn();

    if ( IS_DEBUG )
    {
        LOG.debug( "Check if Dn '" + dn + "' exists." );
    }

    if ( dn.isRootDse() )
    {
        return true;
    }

    Partition partition = getPartition( dn );

    return partition.hasEntry( hasEntryContext );
}
 

public void createStructure() throws Exception {
    String rootDN = AWSIAMAuthenticator.getConfig().rootDN;
    Dn dnIAM = service.getDnFactory().create(rootDN);
    if (!utils.exists(dnIAM)) {
        IAM_LOG.info("Creating partition " + rootDN);
        Partition iamPartition = utils.addPartition("iam", rootDN, service.getDnFactory());

        // Index some attributes on the apache partition
        utils.addIndex(iamPartition, "objectClass", "ou", "uid", "gidNumber", "uidNumber", "cn");

        if (!utils.exists(dnIAM)) {
            IAM_LOG.info("Creating root node " + rootDN);
            Rdn rdn = dnIAM.getRdn(0);
            Entry entryIAM = new DefaultEntry(service.getSchemaManager(), dnIAM, "objectClass: top", "objectClass: domain",
                    "entryCsn: " + service.getCSN(), SchemaConstants.ENTRY_UUID_AT + ": " + UUID.randomUUID().toString(),
                    rdn.getType() + ": " + rdn.getValue());
            service.getAdminSession().add(entryIAM);
            checkErrors();
        }
    }
    service.sync();
}
 

/**
 * {@inheritDoc}
 */
public void modify( ModifyOperationContext modifyContext ) throws LdapException
{
    // Special case : if we don't have any modification to apply, just return
    if ( modifyContext.getModItems().size() == 0 )
    {
        return;
    }

    Partition partition = getPartition( modifyContext.getDn() );

    partition.modify( modifyContext );

    if ( modifyContext.isPushToEvtInterceptor() )
    {
        directoryService.getInterceptor( InterceptorEnum.EVENT_INTERCEPTOR.getName() ).modify( modifyContext );
    }

    Entry alteredEntry = modifyContext.getAlteredEntry();

    if ( alteredEntry != null )
    {
        // MyVD doesn't care about csn
    	// directoryService.setContextCsn( alteredEntry.get( ENTRY_CSN_AT ).getString() );
    }
}
 

/**
 * {@inheritDoc}
 */
public Partition getPartition( Dn dn ) throws LdapException
{
    Partition parent = null;

    if ( !dn.isSchemaAware() )
    {
        dn.apply( schemaManager );
    }

    synchronized ( partitionLookupTree )
    {
        parent = partitionLookupTree.getElement( dn );
    }

    if ( parent == null )
    {
        throw new LdapNoSuchObjectException( I18n.err( I18n.ERR_268, dn ) );
    }
    else
    {
        return parent;
    }
}
 

/**
 * {@inheritDoc}
 */
public EntryFilteringCursor search( SearchOperationContext searchContext ) throws LdapException
{
    Dn base = searchContext.getDn();

    // TODO since we're handling the *, and + in the EntryFilteringCursor
    // we may not need this code: we need see if this is actually the
    // case and remove this code.
    if ( base.size() == 0 )
    {
        return searchFromRoot( searchContext );
    }

    // Not sure we need this code...
    base.apply( schemaManager );

    // Normal case : do a search on the specific partition
    Partition backend = getPartition( base );

    return backend.search( searchContext );
}
 

/**
 * {@inheritDoc}
 */
public Partition getPartition( Dn dn ) throws LdapException
{
    Partition parent = null;

    synchronized ( partitionLookupTree )
    {
        parent = partitionLookupTree.getElement( dn );
    }

    if ( parent == null )
    {
        throw new LdapNoSuchObjectException( I18n.err( I18n.ERR_268, dn ) );
    }
    else
    {
        return parent;
    }
}
 

/**
 * {@inheritDoc}
 */
public EntryFilteringCursor search( SearchOperationContext searchContext ) throws LdapException
{
    Dn base = searchContext.getDn();

    // TODO since we're handling the *, and + in the EntryFilteringCursor
    // we may not need this code: we need see if this is actually the
    // case and remove this code.
    if ( base.size() == 0 )
    {
        return searchFromRoot( searchContext );
    }

    // Not sure we need this code...
    base.apply( schemaManager );

    // Normal case : do a search on the specific partition
    Partition backend = getPartition( base );

    return backend.search( searchContext );
}
 

/**
 * {@inheritDoc}
 */
public void move( MoveOperationContext moveContext ) throws LdapException
{
    // Get the current partition
    Partition partition = getPartition( moveContext.getDn() );

    partition.move( moveContext );

    Entry entry = moveContext.getModifiedEntry();
    
    // MyVD doesn't care about csn
    // directoryService.setContextCsn( entry.get( ENTRY_CSN_AT ).getString() );
}
 

private static void createPartition(final DirectoryServiceFactory dsf, final SchemaManager schemaManager, final String id,
                                    final String suffix) throws Exception {
    PartitionFactory pf = dsf.getPartitionFactory();
    Partition p = pf.createPartition(schemaManager, id, suffix, 1000, workingDir.toFile());
    pf.addIndex(p, "krb5PrincipalName", 10);
    p.initialize();
    directoryService.addPartition(p);
}
 

private static void createPartition(final DirectoryServiceFactory dsf, final SchemaManager schemaManager, final String id,
                                    final String suffix) throws Exception {
    PartitionFactory pf = dsf.getPartitionFactory();
    Partition p = pf.createPartition(schemaManager, id, suffix, 1000, workingDir.toFile());
    pf.addIndex(p, "krb5PrincipalName", 10);
    p.initialize();
    directoryService.addPartition(p);
}
 

/**
 * {@inheritDoc}
 */
public void add( AddOperationContext addContext ) throws LdapException
{
    Partition partition = getPartition( addContext.getDn() );
    partition.add( addContext );

    
    // MyVD doesn't care about csn
    // Attribute at = addContext.getEntry().get( SchemaConstants.ENTRY_CSN_AT );
    // directoryService.setContextCsn( at.getString() );
}
 

/**
 * {@inheritDoc}
 */
public Entry lookup( LookupOperationContext lookupContext ) throws LdapException
{
    Dn dn = lookupContext.getDn();

    if ( dn.equals( subschemSubentryDn ) )
    {
        return new ClonedServerEntry( rootDse.clone() );
    }

    // This is for the case we do a lookup on the rootDSE
    if ( dn.isRootDse() )
    {
        Entry retval = new ClonedServerEntry( rootDse );

        return retval;
    }

    Partition partition = getPartition( dn );
    Entry entry = partition.lookup( lookupContext );

    if ( entry == null )
    {
        LdapNoSuchObjectException e = new LdapNoSuchObjectException( "Attempt to lookup non-existant entry: "
            + dn.getName() );

        throw e;
    }

    return entry;
}
 

/**
 * {@inheritDoc}
 */
public void rename( RenameOperationContext renameContext ) throws LdapException
{
    Partition partition = getPartition( renameContext.getDn() );
    partition.rename( renameContext );

    Entry entry = renameContext.getModifiedEntry();
    
    // MyVD doesn't care about csn
    // directoryService.setContextCsn( entry.get( ENTRY_CSN_AT ).getString() );
}
 

/**
 * {@inheritDoc}
 */
public void moveAndRename( MoveAndRenameOperationContext moveAndRenameContext ) throws LdapException
{
    Partition partition = getPartition( moveAndRenameContext.getDn() );
    partition.moveAndRename( moveAndRenameContext );

    Entry entry = moveAndRenameContext.getModifiedEntry();
    
    // MyVD doesn't care about csn
    // directoryService.setContextCsn( entry.get( ENTRY_CSN_AT ).getString() );
}
 

public void sync() throws Exception
{
    MultiException error = null;

    for ( Partition partition : this.partitions.values() )
    {
        try
        {
            partition.saveContextCsn();
            partition.sync();
        }
        catch ( Exception e )
        {
            LOG.warn( "Failed to flush partition data out.", e );
            if ( error == null )
            {
                //noinspection ThrowableInstanceNeverThrown
                error = new MultiException( I18n.err( I18n.ERR_265 ) );
            }

            // @todo really need to send this info to a monitor
            error.addThrowable( e );
        }
    }

    if ( error != null )
    {
        throw error;
    }
}
 

/**
 * Add a new set of index on the given attributes
 *
 * @param partition The partition on which we want to add index
 * @param attrs The list of attributes to index
 */
protected void addIndex(Partition partition, String... attrs) {
  // Index some attributes on the apache partition
  Set<Index<?, String>> indexedAttributes = new HashSet<>();

  for (String attribute : attrs) {
    indexedAttributes.add(new JdbmIndex<String>(attribute, false));
  }

  ((JdbmPartition) partition).setIndexedAttributes(indexedAttributes);
}
 

/**
 * {@inheritDoc}
 */
public Entry delete( DeleteOperationContext deleteContext ) throws LdapException
{
    Partition partition = getPartition( deleteContext.getDn() );
    Entry deletedEntry = partition.delete( deleteContext );

    return deletedEntry;
}
 

/**
 * {@inheritDoc}
 */
public void move( MoveOperationContext moveContext ) throws LdapException
{
    // Get the current partition
    Partition partition = getPartition( moveContext.getDn() );

    partition.move( moveContext );
}
 

/**
 * Add a new set of index on the given attributes
 *
 * @param partition The partition on which we want to add index
 * @param attrs The list of attributes to index
 */
public void addIndex(Partition partition, String... attrs)
{
    // Index some attributes on the apache partition
    Set<Index<?,String>> indexedAttributes = new HashSet<Index<?,String>>();

    for ( String attribute : attrs )
    {
        indexedAttributes.add( new JdbmIndex( attribute, false ) );
    }

    ( ( JdbmPartition ) partition ).setIndexedAttributes( indexedAttributes );
}
 

/**
 * Add a new partition to the server
 *
 * @param partitionId The partition Id
 * @param partitionDn The partition DN
 * @param dnFactory the DN factory
 * @return The newly added partition
 * @throws Exception If the partition can't be added
 */
public Partition addPartition(String partitionId, String partitionDn, DnFactory dnFactory) throws Exception
{
    // Create a new partition with the given partition id
    JdbmPartition partition = new JdbmPartition(service.getSchemaManager(), dnFactory);
    partition.setId(partitionId);
    partition.setPartitionPath(new File(service.getInstanceLayout().getPartitionsDirectory(), partitionId).toURI());
    partition.setSuffixDn(new Dn(service.getSchemaManager(), partitionDn));
    partition.initialize();
    service.addPartition( partition );

    return partition;
}
 

/**
 * {@inheritDoc}
 */
public void unbind( UnbindOperationContext unbindContext ) throws LdapException
{
    Dn unbindContextDn = unbindContext.getDn();

    if ( !Dn.isNullOrEmpty( unbindContextDn ) )
    {
        Partition partition = getPartition( unbindContext.getDn() );
        partition.unbind( unbindContext );
    }
}
 

/**
 * {@inheritDoc}
 */
public Dn getSuffixDn( Dn dn ) throws LdapException
{
    Partition partition = getPartition( dn );

    return partition.getSuffixDn();
}
 

public Partition getPartition(DirectoryService directory, String id) throws LdapException {
    Set<? extends Partition> partitions = directory.getPartitions();
    for (Partition part : partitions) {
        if (part.getId().equalsIgnoreCase(id)) return part;
    }
    throw new LdapException("No partition with the ID " + id);
}
 

private static void createPartition(final DirectoryServiceFactory dsf, final SchemaManager schemaManager, final String id,
        final String suffix) throws Exception {
    PartitionFactory pf = dsf.getPartitionFactory();
    Partition p = pf.createPartition(schemaManager, id, suffix, 1000, workingDir.toFile());
    pf.addIndex(p, "krb5PrincipalName", 10);
    p.initialize();
    directoryService.addPartition(p);
}
 

private static void createPartition(final DirectoryServiceFactory dsf, final SchemaManager schemaManager, final String id,
        final String suffix, final DirectoryService directoryService, final File workingDir) throws Exception {
    PartitionFactory pf = dsf.getPartitionFactory();
    Partition p = pf.createPartition(schemaManager, id, suffix, 1000, workingDir);
    pf.addIndex(p, "uid", 10);
    pf.addIndex(p, "departmentNumber", 10);
    pf.addIndex(p, "member", 10);
    pf.addIndex(p, "memberOf", 10);
    p.initialize();
    directoryService.addPartition(p);
}