Java集合框架：使用TreeMap和List分组数据的最佳实践

本文旨在解决java开发中常见的`unsupportedoperationexception`，该错误通常在使用`arrays.aslist()`创建的固定大小列表中尝试添加元素时发生。我们将探讨`arrays.aslist()`的工作原理，并提供两种正确且更优雅的解决方案：手动初始化可变列表，以及利用`map.computeifabsent()`方法实现高效的数据分组，特别是在将数据聚合到`treemap>`等结构时。

在Java开发中，我们经常需要将数据按照某个键进行分组，例如将用户列表按照城市分组，每个城市对应一个邮政编码列表。一个常见的实现方式是使用Map>，其中键是城市名称，值是该城市所有邮政编码的列表。然而，在实现过程中，一个常见的陷阱可能导致运行时错误：java.lang.UnsupportedOperationException。

理解Arrays.asList()的局限性

UnsupportedOperationException通常发生在尝试修改一个不支持结构性操作（如添加、删除元素）的List时。这在以下场景中尤为常见：

List fixedList = Arrays.asList("element1");
fixedList.add("element2"); // 这将抛出 UnsupportedOperationException

Arrays.asList()方法返回的是一个由指定数组支持的固定大小的List。这意味着它实际上是一个“视图”而不是一个独立的、可变大小的列表。它不允许添加或删除元素，因为底层数组的大小是固定的。尝试修改其大小会触发UnsupportedOperationException。

原始问题代码分析

考虑以下尝试将用户数据按城市和邮政编码分组的代码：

立即学习“Java免费学习笔记（深入）”；

public TreeMap> getCityWithPincode() {
    List listOfUser = findAllUser(); // 假设获取所有用户数据

    TreeMap> tm = new TreeMap<>();

    for(User user : listOfUser) {
        if(tm.isEmpty()) {
            // 首次添加时使用 Arrays.asList()
            tm.put(user.getCity(), Arrays.asList(user.getPincode()));
        } else {
            if(tm.containsKey(user.getCity())) {
                List list = tm.get(user.getCity());
                // 尝试向 Arrays.asList() 返回的固定大小列表添加元素
                list.add(user.getPincode()); // 这里会抛出 UnsupportedOperationException
                tm.put(user.getCity(), list);
            } else {
                // 新城市首次添加时使用 Arrays.asList()
                tm.put(user.getCity(), Arrays.asList(user.getPincode()));
            }
        }
    }
    return tm;
}

上述代码中，当一个城市首次被添加到TreeMap时，Arrays.asList(user.getPincode())被用来创建一个包含单个邮政编码的列表。问题在于，这个列表是一个固定大小的列表。当后续遇到同一个城市并尝试通过list.add(user.getPincode())向其添加更多邮政编码时，就会触发UnsupportedOperationException。

解决方案一：使用可变列表类型

解决这个问题的核心是确保我们存储在TreeMap中的List是可变的，例如ArrayList。这意味着在任何需要创建新列表的地方，我们都应该显式地使用new ArrayList()。

以下是修改后的代码，它使用ArrayList来确保列表的可变性：

司马诸葛

基于企业知识文档，就可训练专属AI数字员工

下载

import java.util.ArrayList;
import java.util.List;
import java.util.TreeMap;
import java.util.Arrays; // 如果需要从数组转换为可变列表

// 假设User类定义如下
class User {
    private String city;
    private String pincode;

    public User(String city, String pincode) {
        this.city = city;
        this.pincode = pincode;
    }

    public String getCity() { return city; }
    public String getPincode() { return pincode; }
}

public class UserService {

    // 假设findAllUser()方法存在
    private List findAllUser() {
        // 示例数据
        return Arrays.asList(
            new User("New York", "10001"),
            new User("Los Angeles", "90001"),
            new User("New York", "10002"),
            new User("Chicago", "60601"),
            new User("Los Angeles", "90002")
        );
    }

    public TreeMap> getCityWithPincodeCorrected() {
        List listOfUser = findAllUser();
        TreeMap> tm = new TreeMap<>();

        for(User user : listOfUser) {
            // 检查Map中是否已存在该城市
            if(tm.containsKey(user.getCity())) {
                // 如果存在，获取现有列表并添加邮政编码
                List pincodeList = tm.get(user.getCity());
                pincodeList.add(user.getPincode());
                // 无需再次put，因为获取到的是引用
            } else {
                // 如果不存在，创建一个新的ArrayList并添加邮政编码
                List newPincodeList = new ArrayList<>();
                newPincodeList.add(user.getPincode());
                tm.put(user.getCity(), newPincodeList);
            }
        }
        return tm;
    }
}

在这个修正版本中，无论是在添加新城市还是处理现有城市时，我们都确保了TreeMap中存储的List是ArrayList的实例，它支持add()操作。

解决方案二：使用Map.computeIfAbsent()（推荐）

Java 8引入的Map.computeIfAbsent()方法提供了一种更简洁、更高效的方式来处理“如果键不存在则计算并插入值”的逻辑。这非常适合于将数据分组到集合中，因为它可以原子性地处理获取或创建列表的逻辑。

computeIfAbsent(K key, Function super K, ? extends V> mappingFunction)方法的工作原理是：

1. 如果key在Map中已经存在，则直接返回其关联的值。
2. 如果key不存在，则使用mappingFunction来计算一个新值，将key和新值放入Map中，然后返回这个新值。

使用Map.computeIfAbsent()，上述代码可以被大大简化和优化：

import java.util.ArrayList;
import java.util.List;
import java.util.TreeMap;
import java.util.Arrays;
import java.util.Map; // 导入Map接口

// 假设User类定义如上

public class UserService {

    // 假设findAllUser()方法存在
    private List findAllUser() {
        // 示例数据
        return Arrays.asList(
            new User("New York", "10001"),
            new User("Los Angeles", "90001"),
            new User("New York", "10002"),
            new User("Chicago", "60601"),
            new User("Los Angeles", "90002")
        );
    }

    public TreeMap> getCityWithPincodeOptimized() {
        List listOfUser = findAllUser();
        TreeMap> tm = new TreeMap<>();

        for(User user : listOfUser) {
            // 如果城市不存在，则创建一个新的ArrayList并将其关联到该城市
            // 然后获取该ArrayList的引用，并添加邮政编码
            tm.computeIfAbsent(user.getCity(), k -> new ArrayList<>()).add(user.getPincode());
        }
        return tm;
    }

    public static void main(String[] args) {
        UserService service = new UserService();
        TreeMap> result = service.getCityWithPincodeOptimized();
        result.forEach((city, pincodes) -> {
            System.out.println("City: " + city + ", Pincodes: " + pincodes);
        });
        // 预期输出：
        // City: Chicago, Pincodes: [60601]
        // City: Los Angeles, Pincodes: [90001, 90002]
        // City: New York, Pincodes: [10001, 10002]
    }
}

这个优化后的版本不仅代码量更少，而且逻辑更清晰。computeIfAbsent()方法确保了对于每个城市，我们都能得到一个可变的ArrayList实例，无论是新创建的还是已存在的，然后直接向其添加邮政编码。

注意事项与总结

1. Arrays.asList()的用途： Arrays.asList()主要用于将一个数组作为List的视图来使用，进行遍历或读取操作。如果需要一个可变的List，切勿直接使用其返回值进行添加或删除操作。
2. 创建可变列表： 当你需要一个可以动态增删元素的列表时，请显式地使用new ArrayList()、new LinkedList()等可变列表的实现。如果你有一个数组并想将其内容转换为一个可变的列表，可以使用new ArrayList(Arrays.asList(array))。
3. Map.computeIfAbsent()的优势： 对于需要将数据分组到Map中，且值是集合类型（如List、Set）的场景，computeIfAbsent()是极其强大和简洁的工具。它避免了手动进行containsKey()和get()的检查，使代码更易读、更健壮。
4. 线程安全性： TreeMap本身不是线程安全的。如果你的应用是多线程环境，并且需要对TreeMap进行并发修改，你可能需要考虑使用ConcurrentHashMap或通过Collections.synchronizedSortedMap()进行包装。

通过理解Arrays.asList()的特性并恰当地使用Map.computeIfAbsent()，我们可以避免常见的UnsupportedOperationException，并编写出更高效、更简洁的Java集合操作代码。