Skip to content
This repository has been archived by the owner on Aug 5, 2021. It is now read-only.

Commit

Permalink
use native Request type for proxy interface (instead of Jsoup.Connect…
Browse files Browse the repository at this point in the history
…ion)
  • Loading branch information
hatsuyuki15 committed Dec 6, 2017
1 parent 6b532b5 commit b6bd4a7
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 13 deletions.
10 changes: 4 additions & 6 deletions src/main/java/org/hatsuyuki/proxy/DebugProxy.java
Original file line number Diff line number Diff line change
Expand Up @@ -3,16 +3,14 @@
import org.jsoup.Connection;

public class DebugProxy extends Proxy {
private final String debugHost;
private final int debugPort;
private final RemoteRequester debugger;

public DebugProxy(String debugHost, int debugPort) {
this.debugHost = debugHost;
this.debugPort = debugPort;
this.debugger = new RemoteRequester(debugHost, debugPort);
}

@Override
public Response request(Connection jsoupConnection) throws Exception {
return new Response(jsoupConnection.proxy(debugHost, debugPort).execute());
public Response request(Request request) throws Exception {
return debugger.forward(request);
}
}
4 changes: 2 additions & 2 deletions src/main/java/org/hatsuyuki/proxy/LocalProxy.java
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ public LocalProxy(Pipeline pipeline) {
}

@Override
public Response request(Connection jsoupConnection) throws Exception {
return pipeline.forward(new Request(jsoupConnection));
public Response request(Request request) throws Exception {
return pipeline.forward(request);
}
}
6 changes: 5 additions & 1 deletion src/main/java/org/hatsuyuki/proxy/Proxy.java
Original file line number Diff line number Diff line change
Expand Up @@ -6,5 +6,9 @@
* Created by Hatsuyuki on 12/27/2016.
*/
public abstract class Proxy {
public abstract Response request(Connection jsoupConnection) throws Exception;
public Response request(Connection jsoupConnection) throws Exception {
return request(new Request(jsoupConnection));
}

public abstract Response request(Request request) throws Exception;
}
6 changes: 2 additions & 4 deletions src/main/java/org/hatsuyuki/proxy/RealProxy.java
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
package org.hatsuyuki.proxy;

import org.apache.commons.io.IOUtils;
import org.jsoup.Connection;

import java.io.InputStream;
import java.io.OutputStream;
Expand Down Expand Up @@ -38,15 +37,14 @@ private boolean checkConnection() {
}

@Override
public Response request(Connection jsoupConnection) throws Exception {
public Response request(Request request) throws Exception {
Socket socket = null;
try {
int timeout = jsoupConnection.request().timeout();
int timeout = request.timeout();
socket = new Socket();
socket.setSoTimeout(timeout);
socket.connect(new InetSocketAddress(host, port), timeout);

Request request = new Request(jsoupConnection);
request.source = this.clientID;

OutputStream out = socket.getOutputStream();
Expand Down

0 comments on commit b6bd4a7

Please sign in to comment.