gov.nih.nci.logging.api.logger.hibernate.ObjectStateInterceptor.java Source code

Introduction

Here is the source code for gov.nih.nci.logging.api.logger.hibernate.ObjectStateInterceptor.java

Source

//======================================================================================
// Copyright 5AM Solutions Inc, Yale University
//
// Distributed under the OSI-approved BSD 3-Clause License.
// See http://ncip.github.com/caarray/LICENSE.txt for details.
//======================================================================================
package gov.nih.nci.logging.api.logger.hibernate;

/**
 * <!-- LICENSE_TEXT_START -->
 * 
 * <!-- LICENSE_TEXT_END -->
 */

import org.hibernate.*;
import org.hibernate.type.Type;
import gov.nih.nci.logging.api.logger.util.ThreadVariable;
import gov.nih.nci.logging.api.user.UserInfo;
import java.util.*;
import java.io.*;

/**
 * 
 * This class intercepts all the related events when the client application
 * performs the persistence such as save, udpate and delete. Also it generate
 * the audit information about the states of the entity object in different
 * stages.
 * 
 * @author Vijay Parmar (Ekagra Software Technologies Limited.)
 * 
 */
public class ObjectStateInterceptor extends EmptyInterceptor {
    private static ObjectStateLogger logger = ObjectStateLogger.getInstance();

    private String operation = null;

    /**
     * This method gets called before an object is saved.
     */

    public boolean onSave(Object entity, Serializable id, Object[] state, String[] propertyNames, Type[] types)
            throws CallbackException {
        operation = "insert";
        logger.logMessage(entity, id, state, null, propertyNames, types, operation);
        return false;
    }

    /**
     * 
     * This method gets Called before an object is saved
     */
    public void onDelete(Object entity, Serializable id, Object[] state, String[] propertyNames, Type[] types)
            throws CallbackException {
        operation = "delete";
        logger.logMessage(entity, id, state, null, propertyNames, types, operation);
    }

    /* (non-Javadoc)
     * @see org.hibernate.Interceptor#onFlushDirty(java.lang.Object, java.io.Serializable, java.lang.Object[], java.lang.Object[], java.lang.String[], org.hibernate.type.Type[])
     */
    public boolean onFlushDirty(Object entity, Serializable id, Object[] currentState, Object[] previousState,
            String[] propertyNames, Type[] types) throws CallbackException {
        operation = "update";
        logger.logMessage(entity, id, currentState, previousState, propertyNames, types, operation);
        return false;
    }

    /* (non-Javadoc)
     * @see org.hibernate.Interceptor#postFlush(java.util.Iterator)
     */
    public void postFlush(Iterator iterator) throws CallbackException {
        // no impl
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.hibernate.Interceptor#afterTransactionCompletion(org.hibernate.Transaction)
     */
    public void afterTransactionCompletion(Transaction arg0) {
        UserInfo user = (UserInfo) ThreadVariable.get();
        if (null == user)
            user = new UserInfo();
        if (user.getIsIntransaction()) {
            if (arg0.wasCommitted() && user.getTransactionLogs() != null) {
                Iterator it = user.getTransactionLogs().iterator();
                while (it.hasNext()) {
                    String str = (String) it.next();
                    logger.log(str);
                    it.remove();
                }
            } else if (user.getTransactionLogs() != null) {
                user.getTransactionLogs().clear();
            }
            user.setIsIntransaction(false);
        }
        // set back the local thread variable
        ThreadVariable.set(user);
    }

    /* (non-Javadoc)
     * @see org.hibernate.Interceptor#afterTransactionBegin(org.hibernate.Transaction)
     */
    public void afterTransactionBegin(Transaction tx) {
        UserInfo userInfo = (UserInfo) ThreadVariable.get();
        if (null == userInfo)
            userInfo = new UserInfo();
        userInfo.setIsIntransaction(true);
        ThreadVariable.set(userInfo);
    }
}