buffer: optimize byteLength for common encodings

ronag · ronag · commit 42229995994c · 2024-08-12T22:32:38.000+02:00
diff --git a/lib/buffer.js b/lib/buffer.js
@@ -771,13 +771,22 @@ function byteLength(string, encoding) {
   if (len === 0)
     return 0;
 
-  if (encoding) {
-    const ops = getEncodingOps(encoding);
-    if (ops) {
-      return ops.byteLength(string);
-    }
+  if (!encoding || encoding === 'utf8') {
+    return byteLengthUtf8(string);
+  }
+
+  if (encoding === 'ascii') {
+    return len;
   }
-  return byteLengthUtf8(string);
+
+  const ops = getEncodingOps(encoding);
+  if (ops === undefined) {
+    // TODO (ronag): Makes more sense to throw here.
+    // throw new ERR_UNKNOWN_ENCODING(encoding);
+    return byteLengthUtf8(string);
+  }
+
+  return ops.byteLength(string);
 }
 
 Buffer.byteLength = byteLength;
