前言
在开发的过程中,会涉及到很多判断一个数组集合是否包含某个元素的需求。强大的list集合框架通过contains这样一个方法实现了我们的需求。list元素类型是基本类型则可以直接调用contains方法来实现。但是如果元素类型是一个类则需要对这个类做一些处理才能实现我们预想的需求。
分析
通过分析contains方法的源码可知,该方法是通过遍历集合中的每一个元素并用equals方法比较是否存在指定的元素。
contains方法源码:
/**
* {@inheritDoc}
*
* <p>This implementation iterates over the elements in the collection,
* checking each element in turn for equality with the specified element.
*
* @throws ClassCastException {@inheritDoc}
* @throws NullPointerException {@inheritDoc}
*/
public boolean contains(Object o) {
Iterator<E> it = iterator();
if (o==null) {
while (it.hasNext())
if (it.next()==null)
return true;
} else {
while (it.hasNext())
if (o.equals(it.next()))
return true;
}
return false;
}
再分析Object的equals源码,发现底层的比较是通过this == obj实现的,即是比较两个对象的内存地址。我们一般的需求都是比较两个对象的值是否相等,因为基本类型都重写了equals方法,所以基本类型可以直接使用equals来比较两个对象的值是否相等。这也是list元素如果是基本类型可以直接使用contains的原因。所以我们要想实现list元素是一个类,并可以使用contains方法来判断是否存在某个元素,需要重写该类的equals方法。
实现
新建一个person类,并重写equals方法。
person.java
public class Person {
private String name;
private String sex;
public Person() {
}
public Person(String name, String sex) {
this.name = name;
this.sex = sex;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getSex() {
return sex;
}
public void setSex(String sex) {
this.sex = sex;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Person person = (Person) o;
return Objects.equals(name, person.name) &&
Objects.equals(sex, person.sex);
}
}
测试
Main.java
public class Main {
public static void main(String[] args) {
Person person1 = new Person("Tom","boy");
Person person2 = new Person("Mike","boy");
Person person3 = new Person("Susan","girl");
Person person4 = new Person("Mary","girl");
Person person5 = new Person("Jams","boy");
List<Person> personList = new ArrayList<>();
personList.add(person1);
personList.add(person2);
personList.add(person3);
personList.add(person4);
personList.add(person5);
Person personTest1 = new Person("Tom","boy");
Person personTest2 = new Person("Tom","girl");
System.out.println("personTest1是否存在?" + personList.contains(personTest1) + "\n");
System.out.println("personTest2是否存在?" + personList.contains(personTest2) + "\n");
}
}
运行结果如下:
如有任何疑问或者建议欢迎在评论区留言