|
/*
* Copyright (c) 2002-2003 by OpenSymphony
* All rights reserved.
*/
import java.lang.reflect.Array;
import java.util.Collection;
import java.util.Map;
/**
* <code>ContainUtil</code> will check if object 1 contains object 2.
* Object 1 may be an Object, array, Collection, or a Map
*
* @author Matt Baldree (matt@smallleap.com)
* @version $Revision: 2737 $
*/
public class ContainUtil {
/**
* Determine if <code>obj2</code> exists in <code>obj1</code>.
*
* <table borer="1">
* <tr>
* <td>Type Of obj1</td>
* <td>Comparison type</td>
* </tr>
* <tr>
* <td>null<td>
* <td>always return false</td>
* </tr>
* <tr>
* <td>Map</td>
* <td>Map containsKey(obj2)</td>
* </tr>
* <tr>
* <td>Collection</td>
* <td>Collection contains(obj2)</td>
* </tr>
* <tr>
* <td>Array</td>
* <td>there's an array element (e) where e.equals(obj2)</td>
* </tr>
* <tr>
* <td>Object</td>
* <td>obj1.equals(obj2)</td>
* </tr>
* </table>
*
*
* @param obj1
* @param obj2
* @return
*/
public static boolean contains(Object obj1, Object obj2) {
if ((obj1 == null) || (obj2 == null)) {
//log.debug("obj1 or obj2 are null.");
return false;
}
if (obj1 instanceof Map) {
if (((Map) obj1).containsKey(obj2)) {
//log.debug("obj1 is a map and contains obj2");
return true;
}
} else if (obj1 instanceof Collection) {
if (((Collection) obj1).contains(obj2)) {
//log.debug("obj1 is a collection and contains obj2");
return true;
}
} else if (obj1.getClass().isArray()) {
for (int i = 0; i < Array.getLength(obj1); i++) {
Object value = null;
value = Array.get(obj1, i);
if (value.equals(obj2)) {
//log.debug("obj1 is an array and contains obj2");
return true;
}
}
} else if (obj1.equals(obj2)) {
//log.debug("obj1 is an object and equals obj2");
return true;
}
//log.debug("obj1 does not contain obj2: " + obj1 + ", " + obj2);
return false;
}
}
////////////////
package com.opensymphony.webwork.util;
import java.util.ArrayList;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;
import junit.framework.TestCase;
import com.opensymphony.webwork.util.ContainUtil;
public class ContainUtilTest extends TestCase {
public void testNull() throws Exception {
assertFalse(ContainUtil.contains(null, null));
assertFalse(ContainUtil.contains(new Object(), null));
assertFalse(ContainUtil.contains(null, new Object()));
}
public void testSimpleList() throws Exception {
List l = new ArrayList();
l.add("one");
l.add("two");
assertFalse(ContainUtil.contains(l, "three"));
assertTrue(ContainUtil.contains(l, "one"));
assertTrue(ContainUtil.contains(l, "two"));
}
public void testSimpleSet() throws Exception {
Set s = new LinkedHashSet();
s.add("one");
s.add("two");
assertFalse(ContainUtil.contains(s, "thre"));
assertTrue(ContainUtil.contains(s, "one"));
assertTrue(ContainUtil.contains(s, "two"));
}
public void testComplexList() throws Exception {
List l = new ArrayList();
l.add(new MyObject("tm_jee", Integer.valueOf("20")));
l.add(new MyObject("jenny", Integer.valueOf("22")));
assertFalse(ContainUtil.contains(l, new MyObject("paul", Integer.valueOf("50"))));
assertFalse(ContainUtil.contains(l, new MyObject("tm_jee", Integer.valueOf("44"))));
assertTrue(ContainUtil.contains(l, new MyObject("tm_jee", Integer.valueOf("20"))));
assertTrue(ContainUtil.contains(l, new MyObject("jenny", Integer.valueOf("22"))));
}
public void testComplexMap() throws Exception {
Set s = new LinkedHashSet();
s.add(new MyObject("tm_jee", Integer.valueOf("20")));
s.add(new MyObject("jenny", Integer.valueOf("22")));
assertFalse(ContainUtil.contains(s, new MyObject("paul", Integer.valueOf("50"))));
assertFalse(ContainUtil.contains(s, new MyObject("tm_jee", Integer.valueOf("44"))));
assertTrue(ContainUtil.contains(s, new MyObject("tm_jee", Integer.valueOf("20"))));
assertTrue(ContainUtil.contains(s, new MyObject("jenny", Integer.valueOf("22"))));
}
public void testObject() throws Exception {
assertFalse(ContainUtil.contains("aaa", "bbb"));
assertFalse(ContainUtil.contains(new MyObject("tm_jee", Integer.valueOf("22")), new MyObject("tmjee", Integer.valueOf("22"))));
assertTrue(ContainUtil.contains("apple", "apple"));
assertTrue(ContainUtil.contains(new MyObject("tm_jee", Integer.valueOf("22")), new MyObject("tm_jee", Integer.valueOf("22"))));
}
public static class MyObject {
private String name;
private Integer age;
public MyObject(String name, Integer age) {
this.name = name;
this.age = age;
}
public int hashCode() {
return this.name.hashCode();
}
public boolean equals(Object obj) {
if (obj == null) { return false; }
if (! (obj instanceof MyObject)) { return false; }
MyObject tmp = (MyObject) obj;
if (
tmp.name.equals(this.name) &&
tmp.age.equals(this.age)
) {
return true;
}
return false;
}
}
}
|