sessionmap.java

webwork source

/*
 * WebWork, Web Application Framework
 *
 * Distributable under Apache license.
 * See terms of license at opensource.org
 */
package webwork.action.factory;

import javax.servlet.http.HttpSession;
import java.util.*;

/**
 *
 * @author Rickard 謆erg ([e-mail])
 * @version $Revision: 1.9 $
 */
public class SessionMap
   extends AbstractMap
{
   HttpSession session;
   Set entries;
   public SessionMap(HttpSession s)
   {
      this.session = s;
   }

   public Set entrySet()
   {
       synchronized (session) {
           if (entries == null)
           {
              entries = new HashSet();

              Enumeration enum = session.getAttributeNames();
              while (enum.hasMoreElements())
              {
                 final String key = enum.nextElement().toString();
                 final Object value = session.getAttribute(key);
                 entries.add(new Map.Entry()
                 {
                    public boolean equals(Object obj)
                    {
                       Map.Entry entry = (Map.Entry)obj;
                       return (key==null ?
                          entry.getKey()==null : key.equals(entry.getKey()))  &&
                          (value==null ?
                             entry.getValue()==null : value.equals(entry.getValue()));
                    }

                    public int hashCode()
                    {
                       return (key==null   ? 0 : key.hashCode()) ^
                               (value==null ? 0 : value.hashCode());
                    }

                    public Object getKey()
                    {
                       return key;
                    }

                    public Object getValue()
                    {
                       return value;
                    }

                    public Object setValue(Object obj)
                    {
                       session.setAttribute(key.toString(), obj);
                       return value;
                    }
                 });
              }
           }
       }

       return entries;
   }

   public Object put(Object key, Object value)
   {
       synchronized (session) {
           entries = null;
           session.setAttribute(key.toString(), value);
           return get(key);
       }
   }

   public Object get(Object key)
   {
       synchronized (session) {
           return session.getAttribute(key.toString());
       }
   }

   public Object remove(Object key) {
        synchronized (session) {
            entries = null;

            Object value = get(key);
            session.removeAttribute(key.toString());

            return value;
        }
    }

   public void clear()
   {
       synchronized (session) {
           entries = null;
           session.invalidate();
       }
   }
}