Polishing #1367

Prefer exact estimations. Original pull request: #1400.
redis · Sep 4, 2020 · c2b2e1a · c2b2e1a
1 parent 57489fb
commit c2b2e1a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/main/java/io/lettuce/core/codec/StringCodec.java b/src/main/java/io/lettuce/core/codec/StringCodec.java
@@ -191,18 +191,18 @@ public void encode(String str, ByteBuf target) {
      */
     int sizeOf(String value, boolean estimate) {
 
-        if (estimate) {
-            return (int) averageBytesPerChar * value.length();
-        }
-
         if (utf8) {
             return ByteBufUtil.utf8MaxBytes(value);
         }
-        
+
         if (ascii) {
             return value.length();
         }
-
+
+        if (estimate) {
+            return (int) averageBytesPerChar * value.length();
+        }
+
         return (int) maxBytesPerChar * value.length();
     }
 }
diff --git a/src/test/java/io/lettuce/core/codec/StringCodecUnitTests.java b/src/test/java/io/lettuce/core/codec/StringCodecUnitTests.java
@@ -116,7 +116,7 @@ void encodeAndDecodeIso88591Buf() {
     void estimateSize() {
 
         assertThat(new StringCodec(StandardCharsets.UTF_8).estimateSize(teststring))
-                .isEqualTo((int) (teststring.length() * 1.1));
+                .isEqualTo(ByteBufUtil.utf8MaxBytes(teststring));
         assertThat(new StringCodec(StandardCharsets.US_ASCII).estimateSize(teststring)).isEqualTo(teststring.length());
         assertThat(new StringCodec(StandardCharsets.ISO_8859_1).estimateSize(teststring)).isEqualTo(teststring.length());
     }