moving metaDataCounter

Author: xinsong-cui

runtime/auth/aws-signing-common/api/aws-signing-common.api

@@ -18,6 +18,7 @@ public final class aws/smithy/kotlin/runtime/auth/awssigning/AwsChunkedByteReadC
 	public fun isClosedForRead ()Z
 	public fun isClosedForWrite ()Z
 	public fun read (Laws/smithy/kotlin/runtime/io/SdkBuffer;JLkotlin/coroutines/Continuation;)Ljava/lang/Object;
+	public final fun setTotalBytesTransferred (J)V
 }
 
 public final class aws/smithy/kotlin/runtime/auth/awssigning/AwsChunkedSource : aws/smithy/kotlin/runtime/io/SdkSource {
@@ -26,6 +27,7 @@ public final class aws/smithy/kotlin/runtime/auth/awssigning/AwsChunkedSource :
 	public fun close ()V
 	public final fun getTotalBytesTransferred ()J
 	public fun read (Laws/smithy/kotlin/runtime/io/SdkBuffer;J)J
+	public final fun setTotalBytesTransferred (J)V
 }
 
 public final class aws/smithy/kotlin/runtime/auth/awssigning/AwsSignatureType : java/lang/Enum {

runtime/auth/aws-signing-common/common/src/aws/smithy/kotlin/runtime/auth/awssigning/AwsChunkedByteReadChannel.kt

@@ -48,20 +48,12 @@ public class AwsChunkedByteReadChannel(
     /**
      * Tracks the total bytes transferred, including chunk metadata.
      */
-    private var totalBytesTransferred: Long = 0L
-
-    /**
-     * Gets the total bytes transferred for testing purposes only.
-     */
     @InternalApi
-    public fun getTotalBytesTransferred(): Long = totalBytesTransferred
+    public var totalBytesTransferred: Long = 0L
 
     override suspend fun read(sink: SdkBuffer, limit: Long): Long {
         require(limit >= 0L) { "Invalid limit ($limit) must be >= 0L" }
 
-        // Reset metadata bytes counter
-        chunkReader.chunkMetadataBytes = 0
-
         if (!chunkReader.ensureValidChunk()) {
             totalBytesTransferred = -1L
             return -1L

runtime/auth/aws-signing-common/common/src/aws/smithy/kotlin/runtime/auth/awssigning/AwsChunkedSource.kt

@@ -43,20 +43,12 @@ public class AwsChunkedSource(
     /**
      * Tracks the total bytes transferred, including chunk metadata.
      */
-    private var totalBytesTransferred: Long = 0L
-
-    /**
-     * Gets the total bytes transferred for testing purposes only.
-     */
     @InternalApi
-    public fun getTotalBytesTransferred(): Long = totalBytesTransferred
+    public var totalBytesTransferred: Long = 0L
 
     override fun read(sink: SdkBuffer, limit: Long): Long {
         require(limit >= 0L) { "Invalid limit ($limit) must be >= 0L" }
 
-        // Reset metadata bytes counter
-        chunkReader.chunkMetadataBytes = 0
-
         // COROUTINE SAFETY: runBlocking is allowed here because SdkSource is a synchronous blocking interface
         val isChunkValid = runBlocking {
             chunkReader.ensureValidChunk()

runtime/auth/aws-signing-common/common/src/aws/smithy/kotlin/runtime/auth/awssigning/internal/AwsChunkedReader.kt

@@ -71,6 +71,9 @@ internal class AwsChunkedReader(
      * @return true if the [chunk] is valid for reading, false if it's invalid (chunk data is exhausted)
      */
     internal suspend fun ensureValidChunk(): Boolean {
+        // Reset metadata bytes counter
+        chunkMetadataBytes = 0
+
         // check if the current chunk is still valid
         if (chunk.size > 0L) return true
 

runtime/auth/aws-signing-tests/common/src/aws/smithy/kotlin/runtime/auth/awssigning/tests/AwsChunkedTestBase.kt

@@ -35,7 +35,7 @@ fun interface AwsChunkedReaderFactory {
                 // Override read function to track totalBytesTransferred for testing
                 override suspend fun read(sink: SdkBuffer, limit: Long): Long {
                     chunked.read(sink, limit)
-                    return chunked.getTotalBytesTransferred()
+                    return chunked.totalBytesTransferred
                 }
             }
         }

runtime/auth/aws-signing-tests/jvm/src/aws/smithy/kotlin/runtime/auth/awssigning/tests/AwsChunkedSourceTestBase.kt

@@ -22,7 +22,7 @@ val AwsChunkedReaderFactory.Companion.Source: AwsChunkedReaderFactory
             // Override read function to track totalBytesTransferred for testing
             override suspend fun read(sink: SdkBuffer, limit: Long): Long {
                 chunked.read(sink, limit)
-                return chunked.getTotalBytesTransferred()
+                return chunked.totalBytesTransferred
             }
         }
     }