Create StringBuilder with saner initial capacity when encoding a Cookie

Fixes #1951
This commit is contained in:
Trustin Lee 2013-11-02 19:16:23 +09:00
parent 31c1513229
commit 40c9de1b74
3 changed files with 93 additions and 4 deletions

View File

@ -42,7 +42,7 @@ public final class ClientCookieEncoder {
throw new NullPointerException("cookie"); throw new NullPointerException("cookie");
} }
StringBuilder buf = new StringBuilder(); StringBuilder buf = new StringBuilder(estimateClientLength(cookie));
encode(buf, cookie); encode(buf, cookie);
return stripTrailingSeparator(buf); return stripTrailingSeparator(buf);
} }
@ -52,7 +52,16 @@ public final class ClientCookieEncoder {
throw new NullPointerException("cookies"); throw new NullPointerException("cookies");
} }
StringBuilder buf = new StringBuilder(); int initialCapacity = 0;
for (Cookie c: cookies) {
if (c == null) {
break;
}
initialCapacity += estimateClientLength(c);
}
StringBuilder buf = new StringBuilder(initialCapacity);
for (Cookie c: cookies) { for (Cookie c: cookies) {
if (c == null) { if (c == null) {
break; break;
@ -68,7 +77,16 @@ public final class ClientCookieEncoder {
throw new NullPointerException("cookies"); throw new NullPointerException("cookies");
} }
StringBuilder buf = new StringBuilder(); int initialCapacity = 0;
for (Cookie c: cookies) {
if (c == null) {
break;
}
initialCapacity += estimateClientLength(c);
}
StringBuilder buf = new StringBuilder(initialCapacity);
for (Cookie c: cookies) { for (Cookie c: cookies) {
if (c == null) { if (c == null) {
break; break;

View File

@ -25,6 +25,77 @@ final class CookieEncoderUtil {
return buf.toString(); return buf.toString();
} }
static int estimateClientLength(Cookie c) {
int estimate = estimateLength(c.getName(), c.getValue());
if (c.getPath() != null) {
estimate += estimateLength(CookieHeaderNames.PATH, c.getPath());
}
if (c.getDomain() != null) {
estimate += estimateLength(CookieHeaderNames.DOMAIN, c.getDomain());
}
if (c.getVersion() > 0) {
estimate += 12; // '$Version=N; '
int numPorts = c.getPorts().size();
if (numPorts > 0) {
estimate += 10; // '$Port=""; '
estimate += numPorts * 4;
}
}
return estimate;
}
static int estimateServerLength(Cookie c) {
int estimate = estimateLength(c.getName(), c.getValue());
if (c.getMaxAge() != Long.MIN_VALUE) {
estimate += 41; // 'Expires=Sun, 06 Nov 1994 08:49:37 +0900; '
}
if (c.getPath() != null) {
estimate += estimateLength(CookieHeaderNames.PATH, c.getPath());
}
if (c.getDomain() != null) {
estimate += estimateLength(CookieHeaderNames.DOMAIN, c.getDomain());
}
if (c.isSecure()) {
estimate += 8; // 'Secure; '
}
if (c.isHttpOnly()) {
estimate += 10; // 'HTTPOnly; '
}
if (c.getVersion() > 0) {
estimate += 11; // 'Version=N; '
if (c.getComment() != null) {
estimate += estimateLength(CookieHeaderNames.COMMENT, c.getComment());
}
if (c.getCommentUrl() != null) {
estimate += estimateLength(CookieHeaderNames.COMMENTURL, c.getCommentUrl());
}
int numPorts = c.getPorts().size();
if (numPorts > 0) {
estimate += 9; // 'Port=""; '
estimate += numPorts * 4;
}
if (c.isDiscard()) {
estimate += 9; // 'Discard; '
}
}
return estimate;
}
private static int estimateLength(String name, String value) {
return name.length() + value.length() + 6;
}
static void add(StringBuilder sb, String name, String val) { static void add(StringBuilder sb, String name, String val) {
if (val == null) { if (val == null) {
addQuoted(sb, name, ""); addQuoted(sb, name, "");

View File

@ -47,7 +47,7 @@ public final class ServerCookieEncoder {
throw new NullPointerException("cookie"); throw new NullPointerException("cookie");
} }
StringBuilder buf = new StringBuilder(); StringBuilder buf = new StringBuilder(estimateServerLength(cookie));
add(buf, cookie.getName(), cookie.getValue()); add(buf, cookie.getName(), cookie.getValue());