// This Source Code is in the Public Domain per: http://litesoft.org/License.txt
package org.litesoft.core;

import org.litesoft.commonfoundation.typeutils.Objects;

import java.util.*;

@SuppressWarnings({"UnusedDeclaration"})
public class ServerContext
{
    private static final ThreadLocal<ServerContext> DATA = new ThreadLocal<ServerContext>();

    /**
     * Get the current Thread's (ThreadLocal) Server Context.
     *
     * @return the current Server Context
     *
     * @throws IllegalStateException if 'set' has not been called on this Thread.
     */
    public static ServerContext get()
    {
        ServerContext zContext = DATA.get();
        if ( zContext == null )
        {
            throw new IllegalStateException( "No Server Context for current Thread" );
        }
        return zContext;
    }

    /**
     * Clear (or remove) the current Thread's (ThreadLocal) Server Context.
     */
    public static void clear()
    {
        DATA.remove();
    }

    /**
     * Set the current Thread's (ThreadLocal) Server Context to 'this'.
     */
    public void set()
    {
        DATA.set( this );
    }

    private ContextID mContextID;
    private ServerSession mServerSession;

    public ServerContext( ContextID pContextID, ServerSession pServerSession )
    {
        Objects.assertNotNull( "ContextID", mContextID = pContextID );
        Objects.assertNotNull( "ServerSession", mServerSession = pServerSession );
    }

    public ContextID getContextID()
    {
        return mContextID;
    }

    public ServerSession getServerSession()
    {
        return mServerSession;
    }

    private static final ServerStore TEMPLATE_ServerStore = new ServerStore();

    public static ServerStore getTemplateServerStore()
    {
        return TEMPLATE_ServerStore;
    }

    private static final ServerStore MASTER_ServerStore = new ServerStore();

    public static ServerStore getMasterServerStore()
    {
        return MASTER_ServerStore;
    }

    public ServerStore getServerStore()
    {
        synchronized ( SERVER_STORES )
        {
            ServerStore zServerStore = SERVER_STORES.get( mContextID );
            if ( zServerStore == null )
            {
                SERVER_STORES.put( mContextID, zServerStore = new ServerStore( TEMPLATE_ServerStore ) );
            }
            return zServerStore;
        }
    }

    public static void clearServerStore( ContextID pContextID )
    {
        if ( pContextID != null )
        {
            synchronized ( SERVER_STORES )
            {
                LLremoveServerStore( pContextID );
            }
        }
    }

    private static final Map<ContextID, ServerStore> SERVER_STORES = new HashMap<ContextID, ServerStore>();

    public static void clearAllServerStores()
    {
        synchronized ( SERVER_STORES )
        {
            for ( ContextID zContextID : SERVER_STORES.keySet() )
            {
                LLremoveServerStore( zContextID );
            }
        }
    }

    private static void LLremoveServerStore( ContextID pContextID )
    {
        ServerStore zServerStore = SERVER_STORES.remove( pContextID );
        if ( zServerStore != null )
        {
            zServerStore.dispose();
        }
    }
}