Use new V8 v2 write APIs

Author: danlapid

src/workerd/api/streams/encoding.c++

@@ -33,12 +33,10 @@ jsg::Ref<TextEncoderStream> TextEncoderStream::constructor(jsg::Lock& js) {
     auto buffer = maybeBuffer.ToLocalChecked();
 
     auto bytes = jsg::asBytes(buffer).releaseAsChars();
-    [[maybe_unused]] int read = 0;
-    [[maybe_unused]] auto written = str->WriteUtf8(js.v8Isolate, bytes.begin(), bytes.size(), &read,
-        v8::String::NO_NULL_TERMINATION | v8::String::REPLACE_INVALID_UTF8);
+    [[maybe_unused]] auto written = str->WriteUtf8V2(
+        js.v8Isolate, bytes.begin(), bytes.size(), v8::String::WriteFlags::kReplaceInvalidUtf8);
 
     KJ_DASSERT(written == buffer->ByteLength());
-    KJ_DASSERT(read == str->Length());
     controller->enqueue(js, v8::Uint8Array::New(buffer, 0, buffer->ByteLength()));
     return js.resolvedPromise();
   })},

src/workerd/jsg/jsvalue.c++

@@ -258,8 +258,8 @@ int JsString::length(jsg::Lock& js) const {
   return inner->Length();
 }
 
-int JsString::utf8Length(jsg::Lock& js) const {
-  return inner->Utf8Length(js.v8Isolate);
+size_t JsString::utf8Length(jsg::Lock& js) const {
+  return inner->Utf8LengthV2(js.v8Isolate);
 }
 
 kj::String JsString::toString(jsg::Lock& js) const {
@@ -284,29 +284,30 @@ JsString JsString::internalize(Lock& js) const {
 
 JsString::WriteIntoStatus JsString::writeInto(
     Lock& js, kj::ArrayPtr<char> buffer, WriteOptions options) const {
-  WriteIntoStatus result = {0, 0};
+  WriteIntoStatus result = {0};
   if (buffer.size() > 0) {
-    result.written =
-        inner->WriteUtf8(js.v8Isolate, buffer.begin(), buffer.size(), &result.read, options);
+    result.written = inner->WriteUtf8V2(js.v8Isolate, buffer.begin(), buffer.size(), options);
   }
   return result;
 }
 
 JsString::WriteIntoStatus JsString::writeInto(
     Lock& js, kj::ArrayPtr<uint16_t> buffer, WriteOptions options) const {
-  WriteIntoStatus result = {0, 0};
+  WriteIntoStatus result = {0};
   if (buffer.size() > 0) {
-    result.written = inner->Write(js.v8Isolate, buffer.begin(), 0, buffer.size(), options);
+    inner->WriteV2(js.v8Isolate, 0, buffer.size(), buffer.begin(), options);
   }
+  result.written = length(js);
   return result;
 }
 
 JsString::WriteIntoStatus JsString::writeInto(
     Lock& js, kj::ArrayPtr<kj::byte> buffer, WriteOptions options) const {
-  WriteIntoStatus result = {0, 0};
+  WriteIntoStatus result = {0};
   if (buffer.size() > 0) {
-    result.written = inner->WriteOneByte(js.v8Isolate, buffer.begin(), 0, buffer.size(), options);
+    inner->WriteOneByteV2(js.v8Isolate, 0, buffer.size(), buffer.begin(), options);
   }
+  result.written = 1;
   return result;
 }
 

src/workerd/jsg/jsvalue.h

@@ -232,7 +232,7 @@ class JsArray final: public JsBase<v8::Array, JsArray> {
 class JsString final: public JsBase<v8::String, JsString> {
  public:
   int length(Lock& js) const KJ_WARN_UNUSED_RESULT;
-  int utf8Length(Lock& js) const KJ_WARN_UNUSED_RESULT;
+  size_t utf8Length(Lock& js) const KJ_WARN_UNUSED_RESULT;
   kj::String toString(Lock& js) const KJ_WARN_UNUSED_RESULT;
   int hashCode() const;
 
@@ -249,11 +249,9 @@ class JsString final: public JsBase<v8::String, JsString> {
   static JsString concat(Lock& js, const JsString& one, const JsString& two) KJ_WARN_UNUSED_RESULT;
 
   enum WriteOptions {
-    NONE = v8::String::NO_OPTIONS,
-    MANY_WRITES_EXPECTED = v8::String::HINT_MANY_WRITES_EXPECTED,
-    NO_NULL_TERMINATION = v8::String::NO_NULL_TERMINATION,
-    PRESERVE_ONE_BYTE_NULL = v8::String::PRESERVE_ONE_BYTE_NULL,
-    REPLACE_INVALID_UTF8 = v8::String::REPLACE_INVALID_UTF8,
+    NONE = v8::String::WriteFlags::kNone,
+    NO_NULL_TERMINATION = v8::String::WriteFlags::kNullTerminate,
+    REPLACE_INVALID_UTF8 = v8::String::WriteFlags::kReplaceInvalidUtf8,
   };
 
   template <typename T>
@@ -262,9 +260,9 @@ class JsString final: public JsBase<v8::String, JsString> {
 
   struct WriteIntoStatus {
     // The number of elements (e.g. char, byte, uint16_t) read from this string.
-    int read;
+    // size_t read;
     // The number of elements (e.g. char, byte, uint16_t) written to the buffer.
-    int written;
+    size_t written;
   };
   WriteIntoStatus writeInto(
       Lock& js, kj::ArrayPtr<char> buffer, WriteOptions options = WriteOptions::NONE) const;
@@ -442,11 +440,11 @@ inline kj::Array<T> JsString::toArray(Lock& js, WriteOptions options) const {
   if constexpr (kj::isSameType<T, kj::byte>()) {
     KJ_ASSERT(inner->ContainsOnlyOneByte());
     auto buf = kj::heapArray<kj::byte>(inner->Length());
-    inner->WriteOneByte(js.v8Isolate, buf.begin(), 0, buf.size(), options);
+    inner->WriteOneByteV2(js.v8Isolate, 0, buf.size(), buf.begin(), options);
     return kj::mv(buf);
   } else {
     auto buf = kj::heapArray<uint16_t>(inner->Length());
-    inner->Write(js.v8Isolate, buf.begin(), 0, buf.size(), options);
+    inner->WriteV2(js.v8Isolate, 0, buf.size(), buf.begin(), options);
     return kj::mv(buf);
   }
 }

src/workerd/jsg/value.h

@@ -462,7 +462,7 @@ class StringWrapper {
 #else
     auto buf = kj::heapArray<char>(str->Utf8Length(isolate) + 1);
 #endif
-    str->WriteUtf8(isolate, buf.begin(), buf.size());
+    str->WriteUtf8V2(isolate, buf.begin(), buf.size());
     buf[buf.size() - 1] = 0;
     return kj::String(kj::mv(buf));
   }
@@ -1018,7 +1018,7 @@ class DictWrapper {
 #else
       auto buf = kj::heapArray<char>(v8String->Utf8Length(isolate) + 1);
 #endif
-      v8String->WriteUtf8(isolate, buf.begin(), buf.size());
+      v8String->WriteUtf8V2(isolate, buf.begin(), buf.size());
       buf[buf.size() - 1] = 0;
       return kj::String(kj::mv(buf));
     };