[Cronet] Check whether request is done before spinning up message loop

components/cronet/android/java/src/org/chromium/net/urlconnection/CronetChunkedOutputStream.java

Index: components/cronet/android/java/src/org/chromium/net/urlconnection/CronetChunkedOutputStream.java
diff --git a/components/cronet/android/java/src/org/chromium/net/urlconnection/CronetChunkedOutputStream.java b/components/cronet/android/java/src/org/chromium/net/urlconnection/CronetChunkedOutputStream.java
index 1f2c1711f7f01a81735455e18e8fd694bf5c46b6..4c83bfc8fea193d986308ed20cb36fa18b963e73 100644
--- a/components/cronet/android/java/src/org/chromium/net/urlconnection/CronetChunkedOutputStream.java
+++ b/components/cronet/android/java/src/org/chromium/net/urlconnection/CronetChunkedOutputStream.java
@@ -25,7 +25,6 @@ final class CronetChunkedOutputStream extends CronetOutputStream {
     private final UploadDataProvider mUploadDataProvider = new UploadDataProviderImpl();
     private long mBytesWritten;
     private boolean mLastChunk = false;
-    private boolean mClosed = false;
 
     /**
      * Package protected constructor.
@@ -68,8 +67,10 @@ final class CronetChunkedOutputStream extends CronetOutputStream {
             return;
         }
         int toSend = count;
+        // TODO(xunjieli): refactor commond code with CronetFixedModeOutputStream.
         while (toSend > 0) {
             if (mBuffer.position() == mBuffer.limit()) {
+                checkNotClosed();
                 // Wait until buffer is consumed.
                 mMessageLoop.loop();
             }
@@ -82,15 +83,9 @@ final class CronetChunkedOutputStream extends CronetOutputStream {
 
     @Override
     public void close() throws IOException {
+        super.close();
         // Last chunk is written.
         mLastChunk = true;
-        mClosed = true;
-    }
-
-    private void checkNotClosed() throws IOException {
-        if (mClosed) {
-            throw new IOException("Stream has been closed.");
-        }
     }
 
     // Below are CronetOutputStream implementations: