MemoryContext.java

/*
 * Copyright (C) The Spice Group. All rights reserved.
 *
 * This software is published under the terms of the Spice
 * Software License version 1.1, a copy of which has been included
 * with this distribution in the LICENSE.txt file.
 */
package org.realityforge.spice.jndikit.memory;

import java.util.Hashtable;
import javax.naming.Binding;
import javax.naming.Context;
import javax.naming.Name;
import javax.naming.NameClassPair;
import javax.naming.NameNotFoundException;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;
import org.realityforge.spice.jndikit.AbstractLocalContext;
import org.realityforge.spice.jndikit.Namespace;

/**
 * An in memory context implementation.
 */
public class MemoryContext
  extends AbstractLocalContext
{
  private Hashtable<String, Object> m_bindings;

  protected MemoryContext( final Namespace namespace,
                           final Hashtable<String, Object> environment,
                           final Context parent,
                           final Hashtable<String, Object> bindings )
  {
    super( namespace, environment, parent );
    m_bindings = bindings;
  }

  public MemoryContext( final Namespace namespace,
                        final Hashtable<String, Object> environment,
                        final Context parent )
  {
    this( namespace, environment, parent, new Hashtable<String, Object>( 11 ) );
  }

  protected Context newContext()
    throws NamingException
  {
    return new MemoryContext( getNamespace(), getRawEnvironment(), getParent() );
  }

  protected Context cloneContext()
    throws NamingException
  {
    return new MemoryContext( getNamespace(), getRawEnvironment(), getParent(), m_bindings );
  }

  protected void doLocalBind( final Name name, final Object object )
    throws NamingException
  {
    m_bindings.put( name.get( 0 ), object );
  }

  @SuppressWarnings( "unchecked" )
  protected NamingEnumeration<NameClassPair> doLocalList()
    throws NamingException
  {
    return (NamingEnumeration<NameClassPair>) (NamingEnumeration) new MemoryNamingEnumeration( this,
                                                                                               getNamespace(),
                                                                                               m_bindings,
                                                                                               false );
  }

  @SuppressWarnings( "unchecked" )
  protected NamingEnumeration<Binding> doLocalListBindings()
    throws NamingException
  {
    return (NamingEnumeration<Binding>) (NamingEnumeration) new MemoryNamingEnumeration( this,
                                                                                         getNamespace(),
                                                                                         m_bindings,
                                                                                         true );
  }

  /**
   * Actually lookup raw entry in local context.
   * When overriding this it is not necessary to resolve references etc.
   *
   * @param name the name in local context (size() == 1)
   * @return the bound object
   * @throws javax.naming.NamingException if an error occurs
   */
  protected Object doLocalLookup( final Name name )
    throws NamingException
  {
    final Object object = m_bindings.get( name.get( 0 ) );
    if ( null == object )
    {
      throw new NameNotFoundException( name.get( 0 ) );
    }
    return object;
  }

  /**
   * Actually unbind raw entry in local context.
   *
   * @param name the name in local context (size() == 1)
   * @throws javax.naming.NamingException if an error occurs
   */
  protected void doLocalUnbind( final Name name )
    throws NamingException
  {
    m_bindings.remove( name.get( 0 ) );
  }
}