/*  PAT: Persistent Applications Toolkit (patsystem.sf.net)
 *  Copyright (C) 2004, 2005  Tomasz Nazar, nthx at irc dot pl
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *  Full version of the license is /docs/LICENSE.txt
 */
package org.nthx.pat;

import org.apache.log4j.Logger;
import org.jboss.aop.joinpoint.Invocation;
import org.jboss.aop.joinpoint.MethodInvocation;

/** Aspect takes care of taking snapshot of system, removing objects from 
 *  identity map, and similar..
 *  <p>Have a look at {@link Persisteble Persisteble} for more info.
 * 
 *  @version $Id: PersistebleAspect.java 3725 2005-06-09 23:57:03Z nthx $
 *  @author nthx@users.sourceforge.net
 */
public class PersistebleAspect
{
    private static Logger log = Logger.getLogger("pat");
    private int snapshotCounter = 0;

    public PersistebleAspect()
    {
    }

    public Object takeSnapshot(Invocation invocation) throws Throwable
    {
        snapshotCounter++;
        log.debug("Taking snapshot number: " + snapshotCounter);

        Pat.getInstance().takeSnapshot();
        return null;
    }

    public Object removeBO(Invocation invocation) throws Throwable
    {
        MethodInvocation mi = (MethodInvocation) invocation;
        Identifiable bo = (Identifiable) mi.getArguments()[0];
        
        Pat.getInstance().getRootAsIdentityMap().removeObject(bo);
        return null;
    }
    
    public Object injectRoot(Invocation invocation) throws Throwable
    {
        return Pat.getInstance().getRootAsIdentityMap();
    }
    
}