JAVA網路程式設計之模擬表單提交
阿新 • • 發佈:2019-01-09
這一篇部落格是對上一篇《JAVA網路程式設計之獲取網路資源》的擴充,這一篇將使用HttpURLConnection來模擬一個表單的提交。在B/S架構的系統中,請求時通過瀏覽器與服務端進行互動的,提交請求引數時使用form表單進行提交,但是有很多時候,我們需要在程式中直接呼叫HTTP介面,同時需要傳遞一些引數,那麼這個時候我們就可以通過使用HttpURLConnection來模擬表單的提交過程。
首先,我們可以對請求進行封裝
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
/**
* 請求
*
* @author jianggujin
*
*/
public class HttpRequest
{
/** 請求地址 **/
private String url = null;
/** 請求方法 **/
private RequestMethod method = RequestMethod.GET;
/** 編碼 **/
private String charset = "UTF-8";
/** 請求頭資訊 **/
private Map<String, String> headers = new HashMap<String, String>();
/** 請求引數 **/
private Map<String, List<String>> params = new HashMap<String, List<String>>();
/**
* 預設構造方法
*/
public HttpRequest()
{
}
/**
* 構造方法
*
* @param url
*/
public HttpRequest(String url)
{
this.url = url;
}
/**
* 構造方法
*
* @param url
* @param method
*/
public HttpRequest(String url, RequestMethod method)
{
this.url = url;
this.method = method;
}
/**
* 獲得請求地址
*
* @return
*/
public String getUrl()
{
return url;
}
/**
* 設定請求地址
*
* @param url
*/
public void setUrl(String url)
{
this.url = url;
}
/**
* 獲得請求方法
*
* @return
*/
public RequestMethod getMethod()
{
return method;
}
/**
* 設定請求方法
*
* @param method
*/
public void setMethod(RequestMethod method)
{
this.method = method;
}
/**
* 獲得編碼
*
* @return
*/
public String getCharset()
{
return charset;
}
/**
* 設定編碼
*
* @param charset
*/
public void setCharset(String charset)
{
this.charset = charset;
}
/**
* 新增頭資訊
*
* @param name
* @param value
*/
public void addHeader(String name, String value)
{
headers.put(name, value);
}
/**
* 新增請求引數
*
* @param name
* @param value
*/
public void addParam(String name, String value)
{
List<String> list = params.get(name);
if (list == null)
{
list = new ArrayList<String>();
params.put(name, list);
}
list.add(value);
}
/**
* 設定請求引數
*
* @param name
* @param value
*/
public void setParam(String name, String value)
{
List<String> list = params.get(name);
if (list == null)
{
list = new ArrayList<String>();
params.put(name, list);
}
else
{
// 清空已有引數
list.clear();
}
list.add(value);
}
/**
* 獲得請求頭迭代器
*
* @return
*/
public Iterator<Entry<String, String>> getHeaders()
{
return headers.entrySet().iterator();
}
/**
* 獲得請求引數迭代器
*
* @return
*/
public Iterator<Entry<String, List<String>>> getParams()
{
return params.entrySet().iterator();
}
}
/**
* 請求方法
*
* @author jianggujin
*
*/
enum RequestMethod
{
GET, POST
}
在HttpRequest類中,包含了一個請求的基本資訊,然後我們再封裝一個類用於HTTP的請求。
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.Iterator;
import java.util.List;
import java.util.Map.Entry;
/**
* Http請求客戶端
*
* @author jianggujin
*
*/
public abstract class HttpClient
{
/** 連線超時時間 **/
private int connectTimeOut = 10 * 1000;
/** 讀超時時間 **/
private int readTimeOut = 10 * 1000;
/**
* 獲得連線超時時間
*
* @return
*/
public int getConnectTimeOut()
{
return connectTimeOut;
}
/**
* 設定連線超時發時間
*
* @param connectTimeOut
*/
public void setConnectTimeOut(int connectTimeOut)
{
this.connectTimeOut = connectTimeOut;
}
/**
* 獲得讀超時時間
*
* @return
*/
public int getReadTimeOut()
{
return readTimeOut;
}
/**
* 設定讀超時時間
*
* @param readTimeOut
*/
public void setReadTimeOut(int readTimeOut)
{
this.readTimeOut = readTimeOut;
}
/**
* 執行
*/
public void execute(HttpRequest request) throws IOException
{
String url = request.getUrl();
// GET方式請求
if (request.getMethod() == RequestMethod.GET)
{
// 拼接引數
if (url.indexOf('?') != -1)
{
if (url.endsWith("?"))
{
url += dealParam(request);
}
else
{
if (url.endsWith("&"))
{
url += dealParam(request);
}
else
{
url += ("&" + dealParam(request));
}
}
}
else
{
url += ("?" + dealParam(request));
}
}
// 獲得Http連線
HttpURLConnection connection = (HttpURLConnection) new URL(url)
.openConnection();
// 設定超時時間
connection.setConnectTimeout(connectTimeOut);
connection.setReadTimeout(readTimeOut);
// 設定請求頭
Iterator<Entry<String, String>> headers = request.getHeaders();
while (headers.hasNext())
{
Entry<String, String> entry = headers.next();
connection.setRequestProperty(entry.getKey(), entry.getValue());
}
connection.setUseCaches(false);
// 如果是POST請求
if (request.getMethod() == RequestMethod.POST)
{
connection.setDoInput(true);
connection.setDoOutput(true);
connection.setRequestMethod("POST");
}
connection.connect();
// 如果是POST請求
if (request.getMethod() == RequestMethod.POST)
{
OutputStreamWriter writer = new OutputStreamWriter(
connection.getOutputStream(), request.getCharset());
writer.write(dealParam(request));
writer.flush();
writer.close();
}
if (connection.getResponseCode() == HttpURLConnection.HTTP_OK)
{
onSuccess(HttpURLConnection.HTTP_OK, connection.getInputStream());
}
else
{
onSuccess(connection.getResponseCode(), connection.getErrorStream());
}
connection.disconnect();
}
/**
* 處理請求引數
*
* @param request
* @return
* @throws IOException
*/
private String dealParam(HttpRequest request) throws IOException
{
StringBuilder builder = new StringBuilder();
Iterator<Entry<String, List<String>>> iterator = request.getParams();
int count = 0;
// 遍歷請求引數
while (iterator.hasNext())
{
Entry<String, List<String>> entry = iterator.next();
List<String> list = entry.getValue();
for (String string : list)
{
if (count > 0)
{
builder.append("&");
}
builder.append(entry.getKey()).append("=");
if (request.getMethod() == RequestMethod.GET)
{
builder.append(URLEncoder.encode(string, request.getCharset()));
}
else
{
builder.append(string);
}
count++;
}
}
return builder.toString();
}
/**
* 出錯時執行
*
* @param code
* @param stream
* @throws IOException
*/
public abstract void onError(int code, InputStream stream)
throws IOException;
/**
* 成功時執行
*
* @param code
* @param stream
* @throws IOException
*/
public abstract void onSuccess(int code, InputStream stream)
throws IOException;
}
為了方便測試,簡單的編寫一個Servlet,程式碼如下:
package com.jhq.wx.servlet;
import java.io.IOException;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
public class Test extends HttpServlet
{
protected void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException
{
System.out.println(request.getParameter("p1"));
System.out.println(request.getParameter("p2"));
}
protected void doPost(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException
{
request.setCharacterEncoding("UTF-8");
System.out.println(request.getParameter("p1"));
System.out.println(request.getParameter("p2"));
}
}
最後編寫一個測試類,測試模擬表單提交
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
public class Test
{
public static void main(String[] args)
{
HttpRequest request = new HttpRequest("http://127.0.0.1/wx/s");
request.addParam("p1", "123");
request.addParam("p2", "蔣固金");
request.setMethod(RequestMethod.POST);
HttpClient client = new HttpClient()
{
@Override
public void onSuccess(int code, InputStream stream) throws IOException
{
// 內容是文字,直接以緩衝字元流讀取
BufferedReader reader = new BufferedReader(new InputStreamReader(
stream, "UTF-8"));
String data = null;
while ((data = reader.readLine()) != null)
{
System.out.print(data);
}
reader.close();
}
@Override
public void onError(int code, InputStream stream)
{
System.out.println("請求出現異常");
}
};
try
{
client.execute(request);
}
catch (IOException e)
{
e.printStackTrace();
}
}
}
執行之後,我們可以看到服務端後臺輸出:
123
蔣固金
這樣就完成了Java模擬表單提交的操作。