| // Copyright 2004, 2005 The Apache Software Foundation |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package org.apache.tapestry.test.mock; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.io.ObjectInputStream; |
| import java.io.ObjectOutputStream; |
| import java.util.Collections; |
| import java.util.Enumeration; |
| import java.util.HashMap; |
| import java.util.Map; |
| import java.util.Set; |
| |
| import org.apache.hivemind.ApplicationRuntimeException; |
| |
| /** |
| * |
| * Base class for holders of named attributes such as |
| * {@link javax.servlet.http.HttpSession}, |
| * {@link javax.servlet.http.HttpServletRequest} |
| * and {@link javax.servlet.ServletContext}. |
| * |
| * |
| * @author Howard Lewis Ship |
| * @since 4.0 |
| */ |
| |
| public class AttributeHolder |
| { |
| private Map _attributes = new HashMap(); |
| |
| public Object getAttribute(String name) |
| { |
| return _attributes.get(name); |
| } |
| |
| public Enumeration getAttributeNames() |
| { |
| return getEnumeration(_attributes); |
| } |
| |
| protected Enumeration getEnumeration(Map map) |
| { |
| return Collections.enumeration(map.keySet()); |
| } |
| |
| public String[] getAttributeNamesArray() |
| { |
| Set keys = _attributes.keySet(); |
| int count = keys.size(); |
| |
| String[] array = new String[count]; |
| |
| return (String[]) keys.toArray(array); |
| } |
| |
| public void setAttribute(String name, Object value) |
| { |
| _attributes.put(name, value); |
| } |
| |
| public void removeAttribute(String name) |
| { |
| _attributes.remove(name); |
| } |
| |
| /** |
| * Serializes and then deserializes the {@link Map} |
| * containing all attributes. |
| * |
| **/ |
| |
| public void simulateFailover() |
| { |
| byte[] serialized = serializeAttributes(); |
| |
| _attributes = null; |
| |
| Map restoredAttributes = deserializeAttributes(serialized); |
| |
| _attributes = restoredAttributes; |
| } |
| |
| private byte[] serializeAttributes() |
| { |
| try |
| { |
| ByteArrayOutputStream bos = new ByteArrayOutputStream(); |
| ObjectOutputStream oos = new ObjectOutputStream(bos); |
| |
| oos.writeObject(_attributes); |
| |
| oos.close(); |
| |
| return bos.toByteArray(); |
| } |
| catch (IOException ex) |
| { |
| throw new ApplicationRuntimeException("Unable to serialize attributes.", ex); |
| } |
| } |
| |
| private Map deserializeAttributes(byte[] serialized) |
| { |
| try |
| { |
| ByteArrayInputStream bis = new ByteArrayInputStream(serialized); |
| ObjectInputStream ois = new ObjectInputStream(bis); |
| |
| Map result = (Map) ois.readObject(); |
| |
| return result; |
| } |
| catch (Exception ex) |
| { |
| throw new ApplicationRuntimeException("Unable to deserialize attributes.", ex); |
| } |
| } |
| } |