JDK源码-StringJoiner源码分析
背景
功能描述：将多个元素使用指定符号前后连接为字符串；eg：1 2 3 4 5 , => 1,2,3,4,5
要点：
- 多个元素
- 指定分隔符
- 分隔符只在元素之间，不能作为第一或最后一个
使用方法：
// 1 构造 设置分隔符/前缀/后缀
StringJoiner joiner = new StringJoiner(",", "start", "end");
// 2 添加元素
List<String> elements = Arrays.asList("1","2","3","4")
for (String s: elements) {
joiner.add(s);
}
// 3获取拼接后结果
joiner.toString(); // "start1,2,3,4end"
源代码
基础属性
public final class StringJoiner {
// 前缀
private final String prefix;
// 分隔符
private final String delimiter;
// 后缀
private final String suffix;
// 拼接后的value，包含前缀、分隔符、和拼接元素，不包括后缀，后缀在toString返回前才会填充
/*
* StringBuilder value -- at any time, the characters constructed from the
* prefix, the added element separated by the delimiter, but without the
* suffix, so that we can more easily add elements without having to jigger
* the suffix each time.
*/
private StringBuilder value;
// 空值，默认为prefix+suffix，支持自定义设置
private String emptyValue;
}
构造方法
public StringJoiner(CharSequence delimiter) {
this(delimiter, "", "");
}
public StringJoiner(CharSequence delimiter,
CharSequence prefix,
CharSequence suffix) {
Objects.requireNonNull(prefix, "The prefix must not be null");
Objects.requireNonNull(delimiter, "The delimiter must not be null");
Objects.requireNonNull(suffix, "The suffix must not be null");
// make defensive copies of arguments
this.prefix = prefix.toString();
this.delimiter = delimiter.toString();
this.suffix = suffix.toString();
this.emptyValue = this.prefix + this.suffix;
}
可以看到，构造方法只是简单的根据参数设置响应的属性，
值得注意的是：
this.emptyValue = this.prefix + this.suffix;
默认的 emptyValue 是由前缀和后缀简单拼接而成
添加元素
public StringJoiner add(CharSequence newElement) {
prepareBuilder().append(newElement);
return this;
}
private StringBuilder prepareBuilder() {
if (value != null) {
value.append(delimiter);
} else {
value = new StringBuilder().append(prefix);
}
return value;
}
add 方法首先调用了 prepareBuilder 方法用来执行添加元素的前置操作 ；
prepareBuilder 内部会先判断 value 是否为空，
若为空则构建StringBuilder对象且追加前缀，如果不为空则追加分隔符；
执行逻辑如下：
joiner.add("1");
---
prepareBuilder() // value = "start"
.append("1"); // value = "start1"
=========================================================
joiner.add("2");
---
prepareBuilder() // value = "start1,"
.append("2"); // value = "start1,2"
=========================================================
joiner.add("3");
---
prepareBuilder() // value = "start1,2,"
.append("3"); // value = "start1,2,3"
拼接结果
public String toString() {
// 没有调用add方法添加元素，直接返回空值
if (value == null) {
return emptyValue;
} else {
// 没有后缀，直接将value.toString()即可
if (suffix.equals("")) {
return value.toString();
} else {
// 追加后缀后返回
int initialLength = value.length();
String result = value.append(suffix).toString();
// reset value to pre-append initialLength
value.setLength(initialLength);
return result;
}
}
}
核心的逻辑的在：
String result = value.append(suffix).toString();
一句话描述该逻辑是：拼接后缀，然后把完整的结果返回；
疑问点
主要关注下追加后缀后得到结果的前后：
int initialLength = value.length(); // 获取value长度
String result = value.append(suffix).toString(); // value中追加后缀，得到最终结果
value.setLength(initialLength); // 长度重置回追加后缀前的长度，可以理解为将原长度之后字符清空
return result; // 返回拼接完成的结果
- 获取value的长度
- 添加后缀生成最终的结果result
- 将value长度重置回添加后缀之前
- 返回结果
只需要2就拿到结果了，为什么要1和3？如果把1和3去掉，会怎么样？
StringJoiner joiner = new StringJoiner(",", "start", "end");
List<String> elements = Arrays.asList("1","2","3","4")
for (String s: elements) {
joiner.add(s);
}
joiner.toString(); // "start1,2,3,4end"
joiner.add("5");
joiner.toString(); // "start1,2,3,4end,5end"
在StringJoiner#toString方法中用于拿到完整结果临时填充进去后缀没有被去掉，污染了后续使用joiner对象；
显而易见，1&3步骤是为了清理临时填充的后缀"end"。
清理怎么实现的? setLength设置长度就好了?
看下StringBuilder中的setLength的实现：
//java.lang.AbstractStringBuilder#setLength
public void setLength(int newLength) {
if (newLength < 0)
throw new StringIndexOutOfBoundsException(newLength);
ensureCapacityInternal(newLength);
if (count < newLength) {
Arrays.fill(value, count, newLength, '\0');
}
// 设置count为填充后缀之前的长度值
count = newLength;
}
// java.lang.StringBuilder#toString
public String toString() {
// 使用value[0,count]位置的字符生成结果
// Create a copy, don't share the array
return new String(value, 0, count);
}
可以看到，setLength方法是将原长度值赋值给count属性；count是StringBuilder中维护用来记录填充字符数量的属性；StringBuilder#append会从count的位置向后操作value数组；StringBuilder#toString方法调用时也会只会使用[0,count]位置的字符生成字符串结果；
至此核心源码分析就结束了；
新姿势
对于有状态的对象，幂等方法内如果要临时改动内部状态，逻辑完成后需要将变动的属性恢复原貌；
对于使用者而言，StringJoiner#toString方法应该是幂等的，不能对当前象内的数据状态做任何变动，这样才能保证得到的结果是一致的；
再回顾下StringJoiner#value上面的注释：
at any time, the characters constructed from the prefix, the added element separated by the delimiter, but without the suffix
处理数组&列表等数据时，可以标记删除，性能更好；
删除数据时不一定要把指定位置数据清理掉，使用used_index标记有效位置，删除时修改下used_index的方法会更高效；
