[英]How to implement toString() for collections?
我经常需要打印列表或地图,但我不喜欢其默认toString()
方法的输出。 例如HashMap#toString()
输出如下内容:
{key-1=value-1, key-2=value-2, key-3=value-3}
但我希望输出是这样的:
key-1 = value-1, key-2 = value-2, key-3 = value-3
我创建了一个扩展HashMap
的自定义类,并编写了自己的toString()
方法,这导致了我的真正问题:
public final class CustomMap<K, V> extends HashMap<K, V> {
private static final long serialVersionUID = 1493227382148892732L;
@Override
public String toString() {
final StringBuilder builder = new StringBuilder();
final Iterator<Map.Entry<K, V>> entryIterator = entrySet().iterator();
if (entryIterator.hasNext()) {
final Map.Entry<K, V> entry = entryIterator.next();
builder.append(entry.getKey());
builder.append(" = ");
builder.append(entry.getValue());
}
while (entryIterator.hasNext()) {
final Map.Entry<K, V> entry = entryIterator.next();
builder.append(", ");
builder.append(entry.getKey());
builder.append(" = ");
builder.append(entry.getValue());
}
return builder.toString();
}
}
正如你所看到的,我已经重复的代码,这是因为,
部分需要进行一次省略。 那么,如何在没有重复代码的情况下将其重写为一种快速有效的方法呢?
使用布尔值可以轻松解决:
final Iterator<Map.Entry<K, V>> entryIterator = entrySet().iterator();
boolean first = true;
while (entryIterator.hasNext()) {
final Map.Entry<K, V> entry = entryIterator.next();
if (!first)
builder.append(", ");
else
first=false;
builder.append(entry.getKey());
builder.append(" = ");
builder.append(entry.getValue());
}
另一个选择是使用Java 8 StringJoiner:
StringJoiner sj = new StringJoiner(",");
while (entryIterator.hasNext()) {
final Map.Entry<K, V> entry = entryIterator.next();
sj.add(entry.getKey() + " = " + entry.getValue());
}
return sj.toString();
不要扩展HashMap
而是在HashMap.toString()
上使用String.subString()
方法
String s=yourMap.toString();//eg-"{key-1=value-1, key-2=value-2, key-3=value-3}"
System.out.println(s.substring(1,s.length()-1));
输出: -key-1 = value-1,key-2 = value-2,key-3 = value-3
要么
创建删除最后一个逗号的方法
public StringBuilder removeComma(StringBuilder sbf) {
if (sbf.charAt(sbf.length() - 1) == ',') {
sbf.deleteCharAt(sbf.length() - 1);
}
return sbf;
}
public String toString() {
final StringBuilder builder = new StringBuilder();
final Iterator<Map.Entry<K, V>> entryIterator = entrySet().iterator();
while (entryIterator.hasNext()) {
final Map.Entry<K, V> entry = entryIterator.next();
builder.append(entry.getKey()).append(" = ").append(entry.getValue()).append(",");
}
return removeComma(builder).toString();
}
另一种选择是使用变量-
public String toString() {
String comma="";//initaly empty
final StringBuilder builder = new StringBuilder();
final Iterator<Map.Entry<K, V>> entryIterator = entrySet().iterator();
while (entryIterator.hasNext()) {
final Map.Entry<K, V> entry = entryIterator.next();
builder.append(comma).append(entry.getKey()).append(" = ").append(entry.getValue());
comma=",";//set value of comma
}
return removeComma(builder).toString();
}
我只是检查了AbstractCollection
的来源,就这样完成了:
public final class CustomMap<K, V> extends HashMap<K, V> {
/**
* Serial version UID.
*/
private static final long serialVersionUID = 1493227382148892732L;
@Override
public String toString() {
final Iterator<Map.Entry<K, V>> entryIterator = entrySet().iterator();
if (!entryIterator.hasNext()) {
return "";
}
final StringBuilder builder = new StringBuilder();
for (;;) {
final Map.Entry<K, V> entry = entryIterator.next();
builder.append(entry.getKey());
builder.append(" = ");
builder.append(entry.getValue());
if (!entryIterator.hasNext()) {
return builder.toString();
}
builder.append(", ");
}
}
}
这看起来很聪明,循环中只有一个检查。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.