Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 package org.chromium.net.urlconnection; | |
| 6 | |
| 7 import org.chromium.net.ExtendedResponseInfo; | |
| 8 import org.chromium.net.ResponseInfo; | |
| 9 import org.chromium.net.UrlRequest; | |
| 10 import org.chromium.net.UrlRequestContext; | |
| 11 import org.chromium.net.UrlRequestException; | |
| 12 import org.chromium.net.UrlRequestListener; | |
| 13 | |
| 14 import java.io.IOException; | |
| 15 import java.io.InputStream; | |
| 16 import java.net.HttpURLConnection; | |
| 17 import java.net.URL; | |
| 18 import java.nio.ByteBuffer; | |
| 19 import java.util.concurrent.BlockingQueue; | |
| 20 import java.util.concurrent.Executor; | |
| 21 import java.util.concurrent.LinkedBlockingQueue; | |
| 22 | |
| 23 /** | |
| 24 * An implementation of HttpURLConnection that uses Cronet to send requests and | |
| 25 * receive response. This class inherits a {@code connected} field from the | |
| 26 * superclass. That field indicates whether a connection has ever been | |
| 27 * attempted. | |
| 28 */ | |
| 29 public class CronetHttpURLConnection extends HttpURLConnection { | |
| 30 | |
| 31 private final UrlRequestContext mUrlRequestContext; | |
| 32 | |
| 33 private UrlRequest mRequest; | |
| 34 | |
| 35 private ResponseInfo mResponseInfo; | |
| 36 | |
| 37 private InputStream mResponseBody; | |
| 38 | |
| 39 private final MessageLoop mMessageLoop; | |
| 40 | |
| 41 private final BlockingQueue<MessageLoop.Message> mMessageQueue; | |
|
mmenke
2014/11/18 15:53:52
nit: Don't think we need a blank line between eve
xunjieli
2014/11/18 18:13:35
Done.
| |
| 42 | |
| 43 | |
|
mmenke
2014/11/18 15:53:52
nit: Remove extra line break.
xunjieli
2014/11/18 18:13:36
Done.
| |
| 44 protected CronetHttpURLConnection(URL url, | |
| 45 UrlRequestContext urlRequestContext) { | |
| 46 super(url); | |
| 47 mUrlRequestContext = urlRequestContext; | |
| 48 mMessageQueue = new LinkedBlockingQueue<MessageLoop.Message>(); | |
| 49 mMessageLoop = new MessageLoop(mMessageQueue); | |
| 50 } | |
| 51 | |
| 52 | |
|
mmenke
2014/11/18 15:53:52
nit: Remove extra blank line.
xunjieli
2014/11/18 18:13:35
Done.
| |
| 53 /** | |
| 54 * Wrapper executor class which posts tasks to current looper. | |
| 55 */ | |
| 56 private class HandlerThreadExecutor implements Executor { | |
| 57 | |
|
mmenke
2014/11/18 15:53:52
nit: -blank line.
xunjieli
2014/11/18 18:13:35
Done.
| |
| 58 public HandlerThreadExecutor() { | |
| 59 } | |
| 60 | |
| 61 @Override | |
| 62 public void execute(Runnable command) { | |
| 63 try { | |
| 64 mMessageQueue.put(new MessageLoop.Message(command)); | |
| 65 } catch (InterruptedException e) { | |
| 66 e.printStackTrace(); | |
| 67 } | |
| 68 } | |
| 69 } | |
| 70 | |
| 71 /** | |
| 72 * Opens a connection to the resource. | |
| 73 */ | |
| 74 @Override | |
| 75 public void connect() throws IOException { | |
| 76 maybeStartRequest(); | |
| 77 } | |
| 78 | |
| 79 /** | |
| 80 * Releases this connection so that its resources may be either reused or | |
| 81 * closed. | |
| 82 */ | |
| 83 @Override | |
| 84 public void disconnect() { | |
| 85 if (mResponseBody != null) { | |
| 86 try { | |
| 87 mResponseBody.close(); | |
| 88 } catch (IOException e) { | |
| 89 e.printStackTrace(); | |
| 90 } | |
| 91 } | |
| 92 } | |
| 93 | |
| 94 /** | |
| 95 * Returns the response message returned by the remote HTTP server. | |
| 96 */ | |
| 97 @Override | |
| 98 public String getResponseMessage() { | |
| 99 maybeStartRequest(); | |
| 100 return mResponseInfo.getHttpStatusText(); | |
| 101 } | |
| 102 | |
| 103 /** | |
| 104 * Returns the response code returned by the remote HTTP server. | |
| 105 */ | |
| 106 @Override | |
| 107 public int getResponseCode() { | |
| 108 maybeStartRequest(); | |
| 109 return mResponseInfo.getHttpStatusCode(); | |
| 110 } | |
| 111 | |
| 112 /** | |
| 113 * Returns an InputStream for reading data from the resource pointed by this | |
| 114 * URLConnection. | |
| 115 */ | |
| 116 @Override | |
| 117 public InputStream getInputStream() { | |
| 118 maybeStartRequest(); | |
| 119 return mResponseBody; | |
| 120 } | |
| 121 | |
| 122 /** | |
| 123 * Adds the given property to the request header. | |
| 124 */ | |
| 125 @Override | |
| 126 public final void addRequestProperty(String key, String value) { | |
| 127 setRequestProperty(key, value); | |
| 128 } | |
| 129 | |
| 130 /** | |
| 131 * Sets the value of the specified request header field. | |
| 132 */ | |
| 133 @Override | |
| 134 public final void setRequestProperty(String key, String value) { | |
| 135 if (connected) { | |
| 136 throw new IllegalStateException( | |
| 137 "Cannot set request property after connection is made"); | |
| 138 } | |
| 139 maybeCreateRequest(); | |
| 140 mRequest.addHeader(key, value); | |
| 141 } | |
| 142 | |
| 143 /** | |
| 144 * Returns whether this connection uses a proxy server or not. | |
| 145 */ | |
| 146 @Override | |
| 147 public boolean usingProxy() { | |
| 148 // TODO(xunjieli): implement this. | |
| 149 return false; | |
| 150 } | |
| 151 | |
| 152 private class CronetUrlRequestListener implements UrlRequestListener { | |
| 153 | |
|
mmenke
2014/11/18 15:53:52
nit: -blank line
xunjieli
2014/11/18 18:13:36
Done.
| |
| 154 public CronetUrlRequestListener() { | |
| 155 } | |
| 156 | |
| 157 @Override | |
| 158 public void onResponseStarted(UrlRequest request, ResponseInfo info) { | |
| 159 mResponseInfo = info; | |
| 160 } | |
| 161 | |
| 162 @Override | |
| 163 public void onDataReceived(UrlRequest request, ResponseInfo info, | |
| 164 ByteBuffer byteBuffer) { | |
| 165 mResponseInfo = info; | |
| 166 // TODO(xunjieli): handle streaming. | |
|
mmenke
2014/11/18 15:53:52
nit: Capitalize
xunjieli
2014/11/18 18:13:35
Done.
| |
| 167 mResponseBody = new ByteBufferInputStream(byteBuffer); | |
| 168 } | |
| 169 | |
| 170 @Override | |
| 171 public void onRedirect(UrlRequest request, ResponseInfo info, | |
| 172 String newLocationUrl) { | |
| 173 // TODO(xunjieli): handle redirect. | |
|
mmenke
2014/11/18 15:53:52
nit: Capitalize
xunjieli
2014/11/18 18:13:35
Done.
| |
| 174 mResponseInfo = info; | |
| 175 } | |
| 176 | |
| 177 @Override | |
| 178 public void onSucceeded(UrlRequest request, ExtendedResponseInfo info) { | |
| 179 mMessageLoop.quit(); | |
| 180 } | |
| 181 | |
| 182 @Override | |
| 183 public void onFailed(UrlRequest request, ResponseInfo info, | |
| 184 UrlRequestException exception) { | |
| 185 // TODO(xunjieli): handle failure. | |
|
mmenke
2014/11/18 15:53:52
nit: Capitalize
xunjieli
2014/11/18 18:13:36
Done.
| |
| 186 mMessageLoop.quit(); | |
| 187 } | |
| 188 } | |
| 189 | |
| 190 /** | |
| 191 * Maybe starts {@code mRequest}, and waits for it to complete. | |
|
mmenke
2014/11/18 15:53:52
Should be clearer here. Maybe "On first call, cre
xunjieli
2014/11/18 18:13:36
Done.
| |
| 192 */ | |
| 193 private void maybeStartRequest() { | |
| 194 if (connected) { | |
| 195 return; | |
| 196 } | |
| 197 maybeCreateRequest(); | |
| 198 mRequest.start(); | |
| 199 connected = true; | |
| 200 mMessageLoop.loop(); | |
|
mmenke
2014/11/18 15:53:52
Reading over the API docs, should we really get th
xunjieli
2014/11/18 18:13:35
Done. Agreed, we should only block until we receiv
mmenke
2014/11/18 18:40:59
Which method(s) force us to to cache the response
| |
| 201 } | |
| 202 | |
| 203 /** | |
| 204 * Maybe creates {@code mRequest} if it is null. | |
| 205 */ | |
| 206 private void maybeCreateRequest() { | |
| 207 if (mRequest != null) { | |
|
mmenke
2014/11/18 15:53:52
Can mRequest be non-NULL here?
xunjieli
2014/11/18 18:13:35
Yes, if setRequestProperty is called before maybeS
| |
| 208 return; | |
| 209 } | |
| 210 mRequest = mUrlRequestContext.createRequest(url.toString(), | |
| 211 new CronetUrlRequestListener(), new HandlerThreadExecutor()); | |
| 212 } | |
| 213 } | |
| OLD | NEW |