collectionproperty.java

测试工具

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.jmeter.testelement.property;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.apache.jmeter.testelement.TestElement;

public class CollectionProperty extends MultiProperty {

	private static final long serialVersionUID = 221L; // Remember to change this when the class changes ...
	
	private Collection value;

	transient private Collection savedValue;

	public CollectionProperty(String name, Collection value) {
		super(name);
		this.value = normalizeList(value);
	}

	public CollectionProperty() {
		super();
		value = new ArrayList();
	}

	public boolean equals(Object o) {
		if (o instanceof CollectionProperty) {
			if (value != null) {
				return value.equals(((JMeterProperty) o).getObjectValue());
			}
		}
		return false;
	}

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

	public void remove(String prop) {
		PropertyIterator iter = iterator();
		while (iter.hasNext()) {
			if (iter.next().getName().equals(prop)) {
				iter.remove();
			}
		}
	}

	public void set(int index, String prop) {
		if (value instanceof List) {
			((List) value).set(index, new StringProperty(prop, prop));
		}
	}

	public void set(int index, JMeterProperty prop) {
		if (value instanceof List) {
			((List) value).set(index, prop);
		}
	}

	public JMeterProperty get(int row) {
		if (value instanceof List) {
			return (JMeterProperty) ((List) value).get(row);
		}
		return null;
	}

	public void remove(int index) {
		if (value instanceof List) {
			((List) value).remove(index);
		}
	}

	public void setObjectValue(Object v) {
		if (v instanceof Collection) {
			setCollection((Collection) v);
		}

	}

	public PropertyIterator iterator() {
		return getIterator(value);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see JMeterProperty#getStringValue()
	 */
	public String getStringValue() {
		return value.toString();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see JMeterProperty#getObjectValue()
	 */
	public Object getObjectValue() {
		return value;
	}

	public int size() {
		return value.size();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see Object#clone()
	 */
	public Object clone() {
		CollectionProperty prop = (CollectionProperty) super.clone();
		prop.value = cloneCollection();
		return prop;
	}

	private Collection cloneCollection() {
		try {
			Collection newCol = (Collection) value.getClass().newInstance();
			PropertyIterator iter = iterator();
			while (iter.hasNext()) {
				newCol.add(iter.next().clone());
			}
			return newCol;
		} catch (Exception e) {
			log.error("Couldn't clone collection", e);
			return value;
		}
	}

	public void setCollection(Collection coll) {
		value = normalizeList(coll);
	}

	public void clear() {
		value.clear();
	}

	/**
	 * Easy way to add properties to the list.
	 * 
	 * @param prop
	 */
	public void addProperty(JMeterProperty prop) {
		value.add(prop);
	}

	public void addItem(Object item) {
		addProperty(convertObject(item));
	}

	/**
	 * Figures out what kind of properties this collection is holding and
	 * returns the class type.
	 * 
	 * @see AbstractProperty#getPropertyType()
	 */
	protected Class getPropertyType() {
		if (value.size() > 0) {
			return value.iterator().next().getClass();
		}
		return NullProperty.class;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see JMeterProperty#recoverRunningVersion(TestElement)
	 */
	public void recoverRunningVersion(TestElement owner) {
		if (savedValue != null) {
			value = savedValue;
		}
		recoverRunningVersionOfSubElements(owner);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see JMeterProperty#setRunningVersion(boolean)
	 */
	public void setRunningVersion(boolean running) {
		super.setRunningVersion(running);
		if (running) {
			savedValue = value;
		} else {
			savedValue = null;
		}
	}
}