[jboss-cvs] JBossAS SVN: r108408 - in projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache: simple and 1 other directory.

jboss-cvs-commits at lists.jboss.org jboss-cvs-commits at lists.jboss.org
Fri Oct 1 16:24:56 EDT 2010


Author: pferraro
Date: 2010-10-01 16:24:55 -0400 (Fri, 01 Oct 2010)
New Revision: 108408

Added:
   projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/
   projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManager.java
   projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManagerFactory.java
   projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionPersistenceManager.java
Log:
Add this back - it causes circular references otherwise.

Added: projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManager.java
===================================================================
--- projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManager.java	                        (rev 0)
+++ projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManager.java	2010-10-01 20:24:55 UTC (rev 108408)
@@ -0,0 +1,551 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2006, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This 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.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software 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 software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.ejb3.cache.simple;
+
+import java.io.BufferedInputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
+import java.security.AccessController;
+import java.security.PrivilegedAction;
+import java.security.PrivilegedActionException;
+import java.security.PrivilegedExceptionAction;
+
+import java.util.List;
+import java.util.LinkedList;
+
+import javax.ejb.EJBException;
+import org.jboss.ejb3.Container;
+import org.jboss.ejb3.stateful.StatefulBeanContext;
+import org.jboss.serial.io.JBossObjectInputStream;
+import org.jboss.serial.io.JBossObjectOutputStream;
+import org.jboss.system.server.ServerConfigLocator;
+import org.jboss.util.id.UID;
+
+/**
+ * A file-based stateful session bean persistence manager.
+ * <p/>
+ * <p/>
+ * Reads and writes session bean objects to files by using the
+ * standard Java serialization mechanism.
+ * <p/>
+ * <p/>
+ * Passivated state files are stored under:
+ * <tt><em>jboss-server-data-dir</em>/<em>storeDirectoryName</em>/<em>ejb-name</em>-<em>unique-id</em></tt>.
+ * <p/>
+ * <p/>
+ * Since ejb-name is not unique across deployments we generate a <em>unique-id</em> to make
+ * sure that beans with the same EJB name do not collide.
+ *
+ * @author <a href="mailto:rickard.oberg at telkel.com">Rickard ?berg</a>
+ * @author <a href="mailto:marc.fleury at telkel.com">Marc Fleury</a>
+ * @author <a href="mailto:sebastien.alborini at m4x.org">Sebastien Alborini</a>
+ * @author <a href="mailto:jason at planet57.com">Jason Dillon</a>
+ * @author <a href="mailto:bill at jboss.org">Bill Burke</a>
+ * @version <tt>$Revision: 80079 $</tt>
+ * @jmx:mbean extends="org.jboss.system.ServiceMBean"
+ */
+public class StatefulSessionFilePersistenceManager implements StatefulSessionPersistenceManager
+{
+
+   /**
+    * The sub-directory name under the server data directory where
+    * session data is stored.
+    *
+    * @see #DEFAULT_STORE_DIRECTORY_NAME
+    * @see #setStoreDirectoryName
+    */
+   private String storeDirName = DEFAULT_STORE_DIRECTORY_NAME;
+
+   /**
+    * The base directory where sessions state files are stored for our container.
+    */
+   private File storeDir;
+
+   private Container con;
+
+   /**
+    * Enable purging leftover state files at create and destroy
+    * time (default is true).
+    */
+   private boolean purgeEnabled = true;
+
+   /**
+    * Set the sub-directory name under the server data directory
+    * where session data will be stored.
+    * <p/>
+    * <p/>
+    * This value will be appened to the value of
+    * <tt><em>jboss-server-data-dir</em></tt>.
+    * <p/>
+    * <p/>
+    * This value is only used during creation and will not dynamically
+    * change the store directory when set after the create step has finished.
+    *
+    * @param dirName A sub-directory name.
+    */
+   public void setStoreDirectoryName(final String dirName)
+   {
+      this.storeDirName = dirName;
+   }
+
+   /**
+    * Get the sub-directory name under the server data directory
+    * where session data is stored.
+    *
+    * @return A sub-directory name.
+    * @see #setStoreDirectoryName
+    */
+   public String getStoreDirectoryName()
+   {
+      return storeDirName;
+   }
+
+   /**
+    * Set the stale session state purge enabled flag.
+    *
+    * @param flag The toggle flag to enable or disable purging.
+    */
+   public void setPurgeEnabled(final boolean flag)
+   {
+      this.purgeEnabled = flag;
+   }
+
+   /**
+    * Get the stale session state purge enabled flag.
+    *
+    * @return True if purge is enabled.
+    */
+   public boolean getPurgeEnabled()
+   {
+      return purgeEnabled;
+   }
+
+   /**
+    * Returns the directory used to store session passivation state files.
+    *
+    * @return The directory used to store session passivation state files.
+    */
+   public File getStoreDirectory()
+   {
+      return storeDir;
+   }
+
+   public void setContainer(Container con)
+   {
+      this.con = con;
+   }
+
+   /**
+    * Setup the session data storage directory.
+    * <p/>
+    * <p>Purges any existing session data found.
+    */
+   public void initialize(Container con) throws Exception
+   {
+      this.con = con;
+      boolean debug = log.isDebugEnabled();
+
+      // Initialize the dataStore
+
+      String ejbName = con.getEjbName();
+
+      // Get the system data directory
+      String sysPropJBossTempDir = "jboss.server.temp.dir";
+      String sysPropJavaTempDir = "java.io.tmpdir";
+      String tempDir = System.getProperty(sysPropJBossTempDir);
+      if(tempDir==null||tempDir.trim().length()==0)
+      {
+         tempDir = System.getProperty(sysPropJavaTempDir);
+      }
+      File dir = new File(tempDir);
+
+      // Setup the reference to the session data store directory
+      dir = new File(dir, storeDirName);
+      // ejbName is not unique across all deployments, so use a unique token
+      dir = new File(dir, ejbName + "-" + new UID().toString());
+      storeDir = dir;
+
+      if (debug)
+      {
+         log.debug("Storing sessions for '" + ejbName + "' in: " + storeDir);
+      }
+
+      // if the directory does not exist then try to create it
+      if (!storeDir.exists())
+      {
+         if (MkdirsFileAction.mkdirs(storeDir) == false)
+         {
+            throw new IOException("Failed to create directory: " + storeDir);
+         }
+      }
+
+      // make sure we have a directory
+      if (!storeDir.isDirectory())
+      {
+         throw new IOException("File exists where directory expected: " + storeDir);
+      }
+
+      // make sure we can read and write to it
+      if (!storeDir.canWrite() || !storeDir.canRead())
+      {
+         throw new IOException("Directory must be readable and writable: " + storeDir);
+      }
+
+      // Purge state session state files, should be none, due to unique directory
+      purgeAllSessionData();
+   }
+
+   /**
+    * Removes any state files left in the storgage directory.
+    */
+   public void purgeAllSessionData()
+   {
+      if (!purgeEnabled) return;
+
+      log.debug("Purging all session data in: " + storeDir);
+
+      File[] sessions = storeDir.listFiles();
+      for (int i = 0; i < sessions.length; i++)
+      {
+         if (!sessions[i].delete())
+         {
+            log.warn("Failed to delete session state file: " + sessions[i]);
+         }
+         else
+         {
+            log.debug("Removed stale session state: " + sessions[i]);
+         }
+      }
+   }
+
+   /**
+    * Purge any data in the store, and then the store directory too.
+    */
+   public void destroy() throws Exception
+   {
+      // Purge data and attempt to delete directory
+      purgeAllSessionData();
+
+      // Nuke the directory too if purge is enabled
+      if (purgeEnabled && !storeDir.delete())
+      {
+         log.warn("Failed to delete session state storage directory: " + storeDir);
+      }
+   }
+
+   /**
+    * Make a session state file for the given instance id.
+    */
+   private File getFile(final Object id)
+   {
+      //
+      // jason: may have to translate id into a os-safe string, though
+      //        the format of UID is safe on Unix and win32 already...
+      //
+
+      return new File(storeDir, String.valueOf(id) + ".ser");
+   }
+
+   /**
+    * Restores session state from the serialized file & invokes
+    * {@link javax.ejb.SessionBean#ejbActivate} on the target bean.
+    */
+   public StatefulBeanContext activateSession(Object id)
+   {
+      boolean debug = log.isDebugEnabled();
+      if (debug)
+      {
+         log.debug("Attempting to activate; id=" + id);
+      }
+
+      // Load state
+      File file = getFile(id);
+      if (!file.exists()) return null;
+      
+      if (debug)
+      {
+         log.debug("Reading session state from: " + file);
+      }
+
+      StatefulBeanContext bean = null;
+      try
+      {
+         FileInputStream fis = FISAction.open(file);
+         // todo need to rewrite SessionObjectInputStream to support EJB3 classes
+         ObjectInputStream in;
+
+         in = new JBossObjectInputStream(new BufferedInputStream(fis));
+         try
+         {
+            bean = (StatefulBeanContext) in.readObject();
+         }
+         finally
+         {
+            fis.close();
+            in.close();
+         }
+      }
+      catch (EJBException e)
+      {
+         throw e;
+      }
+      catch (Exception e)
+      {
+         throw new EJBException("Could not activate; failed to " +
+                                "restore state", e);
+      }
+
+      removePassivated(id);
+
+      bean.postActivate();
+      return bean;
+   }
+   
+   public List<StatefulBeanContext> getPassivatedBeans()
+   {
+      List beans = new LinkedList();
+      
+      File[] files = storeDir.listFiles();
+      for (File file : files)
+      {
+         try
+         {
+            ObjectInputStream in;
+            
+            FileInputStream fis = FISAction.open(file);
+   
+            in = new JBossObjectInputStream(new BufferedInputStream(fis));
+            try
+            {
+               StatefulBeanContext bean = (StatefulBeanContext) in.readObject();
+               beans.add(bean);
+            }
+            finally
+            {
+               fis.close();
+               in.close();
+            }
+         }
+         catch (Exception e)
+         {
+            log.warn("Could not read for timeout removal for file " + file.getName(), e);
+         }
+      }
+      
+      return beans;
+   }
+
+   /**
+    * Invokes {@link javax.ejb.SessionBean#ejbPassivate} on the target bean and saves the
+    * state of the session to a file.
+    */
+   public void passivateSession(StatefulBeanContext ctx)
+   {
+      boolean debug = log.isDebugEnabled();
+      if (debug)
+      {
+         log.debug("Attempting to passivate; id=" + ctx.getId());
+      }
+
+      ctx.prePassivate();
+      // Store state
+
+      File file = getFile(ctx.getId());
+      if (debug)
+      {
+         log.debug("Saving session state to: " + file);
+      }
+
+      try
+      {
+         FileOutputStream fos = FOSAction.open(file);
+         // todo need to rewrite SessionObjectOutputStream to support EJB3 classes
+         ObjectOutputStream out;
+
+         out = new JBossObjectOutputStream(fos, false);
+
+         try
+         {
+            out.writeObject(ctx);
+            out.flush();
+            fos.flush();
+            fos.close();
+         }
+         finally
+         {
+            out.close();
+         }
+      }
+      catch (EJBException e)
+      {
+         throw e;
+      }
+      catch (Exception e)
+      {
+         throw new EJBException("Could not passivate; failed to save state", e);
+      }
+
+      if (debug)
+      {
+         log.debug("Passivation complete; id=" + ctx.getId());
+      }
+   }
+
+   /**
+    * Removes the saved state file (if any) for the given session id.
+    */
+   public void removePassivated(Object id)
+   {
+      boolean debug = log.isDebugEnabled();
+
+      File file = getFile(id);
+
+      // only attempt to delete if the file exists
+      if (file.exists())
+      {
+         if (debug)
+         {
+            log.debug("Removing passivated state file: " + file);
+         }
+
+         if (DeleteFileAction.delete(file) == false)
+         {
+            log.warn("Failed to delete passivated state file: " + file);
+         }
+      }
+   }
+
+   static class DeleteFileAction implements PrivilegedAction
+   {
+      File file;
+
+      DeleteFileAction(File file)
+      {
+         this.file = file;
+      }
+
+      public Object run()
+      {
+         boolean deleted = file.delete();
+         return new Boolean(deleted);
+      }
+
+      static boolean delete(File file)
+      {
+         DeleteFileAction action = new DeleteFileAction(file);
+         Boolean deleted = (Boolean) AccessController.doPrivileged(action);
+         return deleted.booleanValue();
+      }
+   }
+
+   static class MkdirsFileAction implements PrivilegedAction
+   {
+      File file;
+
+      MkdirsFileAction(File file)
+      {
+         this.file = file;
+      }
+
+      public Object run()
+      {
+         boolean ok = file.mkdirs();
+         return new Boolean(ok);
+      }
+
+      static boolean mkdirs(File file)
+      {
+         MkdirsFileAction action = new MkdirsFileAction(file);
+         Boolean ok = (Boolean) AccessController.doPrivileged(action);
+         return ok.booleanValue();
+      }
+   }
+
+   static class FISAction implements PrivilegedExceptionAction
+   {
+      File file;
+
+      FISAction(File file)
+      {
+         this.file = file;
+      }
+
+      public Object run() throws Exception
+      {
+         FileInputStream fis = new FileInputStream(file);
+         return fis;
+      }
+
+      static FileInputStream open(File file) throws FileNotFoundException
+      {
+         FISAction action = new FISAction(file);
+         FileInputStream fis = null;
+         try
+         {
+            fis = (FileInputStream) AccessController.doPrivileged(action);
+         }
+         catch (PrivilegedActionException e)
+         {
+            throw (FileNotFoundException) e.getException();
+         }
+
+         return fis;
+      }
+   }
+
+   static class FOSAction implements PrivilegedExceptionAction
+   {
+      File file;
+
+      FOSAction(File file)
+      {
+         this.file = file;
+      }
+
+      public Object run() throws Exception
+      {
+         FileOutputStream fis = new FileOutputStream(file);
+         return fis;
+      }
+
+      static FileOutputStream open(File file) throws FileNotFoundException
+      {
+         FOSAction action = new FOSAction(file);
+         FileOutputStream fos = null;
+         try
+         {
+            fos = (FileOutputStream) AccessController.doPrivileged(action);
+         }
+         catch (PrivilegedActionException e)
+         {
+            throw (FileNotFoundException) e.getException();
+         }
+
+         return fos;
+      }
+   }
+}


Property changes on: projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManager.java
___________________________________________________________________
Name: svn:executable
   + *

Added: projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManagerFactory.java
===================================================================
--- projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManagerFactory.java	                        (rev 0)
+++ projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManagerFactory.java	2010-10-01 20:24:55 UTC (rev 108408)
@@ -0,0 +1,39 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2007, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This 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.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software 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 software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.ejb3.cache.simple;
+
+import org.jboss.ejb3.cache.persistence.PersistenceManagerFactory;
+
+/**
+ * Factory for obtaining StatefulSessionFilePersistenceManager instances
+ * 
+ * @author <a href="mailto:andrew.rubinger at redhat.com">ALR</a>
+ * @version $Revision: $
+ */
+public class StatefulSessionFilePersistenceManagerFactory implements PersistenceManagerFactory
+{
+   public StatefulSessionPersistenceManager createPersistenceManager()
+   {
+      return new StatefulSessionFilePersistenceManager();
+   }
+
+}


Property changes on: projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionFilePersistenceManagerFactory.java
___________________________________________________________________
Name: svn:executable
   + *

Added: projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionPersistenceManager.java
===================================================================
--- projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionPersistenceManager.java	                        (rev 0)
+++ projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionPersistenceManager.java	2010-10-01 20:24:55 UTC (rev 108408)
@@ -0,0 +1,66 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2006, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This 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.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software 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 software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.ejb3.cache.simple;
+
+import java.util.List;
+
+import org.jboss.ejb3.Container;
+import org.jboss.ejb3.stateful.StatefulBeanContext;
+import org.jboss.logging.Logger;
+
+/**
+ * Comment
+ *
+ * @author <a href="mailto:bill at jboss.org">Bill Burke</a>
+ * @version $Revision: 61333 $
+ */
+public interface StatefulSessionPersistenceManager
+{
+   Logger log = Logger.getLogger(StatefulSessionFilePersistenceManager.class);
+   /**
+    * The default store directory name ("<tt>sessions</tt>").
+    */
+   String DEFAULT_STORE_DIRECTORY_NAME = "sessions";
+
+   /**
+    * Restores session state from the serialized file & invokes
+    * {@link javax.ejb.SessionBean#ejbActivate} on the target bean.
+    */
+   StatefulBeanContext activateSession(Object id);
+
+   /**
+    * Invokes {@link javax.ejb.SessionBean#ejbPassivate} on the target bean and saves the
+    * state of the session to a file.
+    */
+   void passivateSession(StatefulBeanContext ctx);
+   
+   List<StatefulBeanContext> getPassivatedBeans();
+
+   /**
+    * Removes the saved state file (if any) for the given session id.
+    */
+   void removePassivated(Object id);
+
+   void destroy() throws Exception;
+
+   public void initialize(Container container) throws Exception;
+}


Property changes on: projects/ejb3/branches/infinispan-int/core/src/main/java/org/jboss/ejb3/cache/simple/StatefulSessionPersistenceManager.java
___________________________________________________________________
Name: svn:executable
   + *



More information about the jboss-cvs-commits mailing list