|
|
java中computeIfAbsent 方法详解(可以按某一值分组)
# `computeIfAbsent` 方法详解
```java
aipGroups.computeIfAbsent(aip, k -> new ArrayDeque<>()).add(server);
```
这是一行非常简洁的 Java 8 代码,完成了多个操作。让我拆解说明:
## 功能分解
### 1. `computeIfAbsent(aip, k -> new ArrayDeque<>())`
| 情况 | 行为 |
|------|------|
| **键 `aip` 不存在** | 创建新的 `ArrayDeque<>()`,放入 Map,返回这个新队列 |
| **键 `aip` 已存在** | 直接返回已存在的队列,不创建新的 |
### 2. `.add(server)`
对返回的队列(无论是新建的还是已存在的)添加 `server`。
## 等价的传统写法
```java
// 传统写法(Java 7 及之前)
Deque<WDServers> queue = aipGroups.get(aip);
if (queue == null) {
queue = new ArrayDeque<>();
aipGroups.put(aip, queue);
}
queue.add(server);
```
## 示例演示
```java
Map<String, Deque<WDServers>> aipGroups = new LinkedHashMap<>();
// 假设有以下服务器数据:
// server1: aip = "A"
// server2: aip = "A"
// server3: aip = "B"
// 第一次:aip="A" 不存在,创建新队列,添加 server1
aipGroups.computeIfAbsent("A", k -> new ArrayDeque<>()).add(server1);
// 结果:{"A": [server1]}
// 第二次:aip="A" 已存在,直接获取队列,添加 server2
aipGroups.computeIfAbsent("A", k -> new ArrayDeque<>()).add(server2);
// 结果:{"A": [server1, server2]}
// 第三次:aip="B" 不存在,创建新队列,添加 server3
aipGroups.computeIfAbsent("B", k -> new ArrayDeque<>()).add(server3);
// 结果:{"A": [server1, server2], "B": [server3]}
```
## Lambda 表达式说明
```java
k -> new ArrayDeque<>()
```
| 部分 | 说明 |
|------|------|
| `k` | 键值(即 `aip`),这里虽然传入但未使用 |
| `->` | Lambda 箭头 |
| `new ArrayDeque<>()` | 创建新队列的表达式 |
## 优点
| 优点 | 说明 |
|------|------|
| **简洁** | 一行代码完成判断、创建、添加 |
| **原子性** | 避免并发问题(对于 ConcurrentHashMap) |
| **延迟创建** | 只在键不存在时才创建新对象 |
| **链式调用** | 返回值可直接操作 |
## 在轮询算法中的作用
```java
// 按 AIP 分组服务器
for (WDServers server : servers) {
String aip = server.getAip() != null ? server.getAip() : "";
// 自动按 aip 分组,相同 aip 的服务器放入同一个队列
aipGroups.computeIfAbsent(aip, k -> new ArrayDeque<>()).add(server);
}
// 最终结果示例:
// {
// "192.168.1.1": [server1, server4, server7],
// "192.168.1.2": [server2, server5],
// "192.168.1.3": [server3, server6, server8, server9]
// }
```
这样就可以按 AIP 分组后进行轮询分配了。
|
|