基于Java實現Socket編程入門
更新時間：2022年03月10日 11:43:37 作者：Lumin
Java最初是作為網絡編程語言出現的，使得客戶端和服務器的溝通變成了現實，而在網絡編程中，使用最多的就是Socket，本文就來介紹一下基于Java實現Socket編程入門，感興趣的可以來了解一下
認識Socket
socket，又稱套接字，是在不同的進程間進行網絡通訊的一種協(xié)議、約定或者說是規(guī)范。
對于socket編程，它更多的時候像是基于TCP/UDP等協(xié)議做的一層封裝或者說抽象，是一套系統(tǒng)所提供的用于進行網絡通信相關編程的接口。
建立socket的基本流程
我們以linux操作系統(tǒng)提供的基本api為例，了解建立一個socket通信的基本流程：
可以看到本質上，socket是對tcp連接（當然也有可能是udp等其他連接）協(xié)議，在編程層面上的簡化和抽象。
1.最基本的Socket示范
1.1 單向通信
首先，我們從只發(fā)送和接收一次消息的socket基礎代碼開始：
服務端:
package com.marklux.socket.base;
import java.io.IOException;
import java.io.InputStream;
import java.net.ServerSocket;
import java.net.Socket;
/**
* The very basic socket server that only listen one single message.
*/
public class BaseSocketServer {
private ServerSocket server;
private Socket socket;
private int port;
private InputStream inputStream;
private static final int MAX_BUFFER_SIZE = 1024;
public int getPort() {
return port;
}
public void setPort(int port) {
this.port = port;
}
public BaseSocketServer(int port) {
this.port = port;
}
public void runServerSingle() throws IOException {
this.server = new ServerSocket(this.port);
System.out.println("base socket server started.");
// the code will block here till the request come.
this.socket = server.accept();
this.inputStream = this.socket.getInputStream();
byte[] readBytes = new byte[MAX_BUFFER_SIZE];
int msgLen;
StringBuilder stringBuilder = new StringBuilder();
while ((msgLen = inputStream.read(readBytes)) != -1) {
stringBuilder.append(new String(readBytes,0,msgLen,"UTF-8"));
}
System.out.println("get message from client: " + stringBuilder);
inputStream.close();
socket.close();
server.close();
}
public static void main(String[] args) {
BaseSocketServer bs = new BaseSocketServer(9799);
try {
bs.runServerSingle();
}catch (IOException e) {
e.printStackTrace();
}
}
}客戶端:
package com.marklux.socket.base;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
/**
* The very basic socket client that only send one single message.
*/
public class BaseSocketClient {
private String serverHost;
private int serverPort;
private Socket socket;
private OutputStream outputStream;
public BaseSocketClient(String host, int port) {
this.serverHost = host;
this.serverPort = port;
}
public void connetServer() throws IOException {
this.socket = new Socket(this.serverHost, this.serverPort);
this.outputStream = socket.getOutputStream();
// why the output stream?
}
public void sendSingle(String message) throws IOException {
try {
this.outputStream.write(message.getBytes("UTF-8"));
} catch (UnsupportedEncodingException e) {
System.out.println(e.getMessage());
}
this.outputStream.close();
this.socket.close();
}
public static void main(String[] args) {
BaseSocketClient bc = new BaseSocketClient("127.0.0.1",9799);
try {
bc.connetServer();
bc.sendSingle("Hi from mark.");
}catch (IOException e) {
e.printStackTrace();
}
}
}先運行服務端，再運行客戶端，就可以看到效果。
- 注意這里的IO操作實現，我們使用了一個大小為
MAX_BUFFER_SIZE的byte數組作為緩沖區(qū)，然后從輸入流中取出字節(jié)放置到緩沖區(qū)，再從緩沖區(qū)中取出字節(jié)構建到字符串中去，這在輸入流文件很大時非常有用，事實上，后面要講到的NIO也是基于這種思路實現的。
1.2 雙向通信
上面的例子只實現了一次單向的通信，這顯然有點浪費通道。socket連接支持全雙工的雙向通信（底層是tcp），下面的例子中，服務端在收到客戶端的消息后，將返回給客戶端一個回執(zhí)。
并且我們使用了一些java.io包裝好的方法，來簡化整個通信的流程（因為消息長度不大，不再使用緩沖區(qū)）。
服務端:
public void runServer() throws IOException {
this.serverSocket = new ServerSocket(port);
this.socket = serverSocket.accept();
this.inputStream = socket.getInputStream();
String message = new String(inputStream.readAllBytes(), "UTF-8");
System.out.println("received message: " + message);
this.socket.shutdownInput(); // 告訴客戶端接收已經完畢，之后只能發(fā)送
// write the receipt.
this.outputStream = this.socket.getOutputStream();
String receipt = "We received your message: " + message;
outputStream.write(receipt.getBytes("UTF-8"));
this.outputStream.close();
this.socket.close();
}客戶端:
public void sendMessage(String message) throws IOException {
this.socket = new Socket(host,port);
this.outputStream = socket.getOutputStream();
this.outputStream.write(message.getBytes("UTF-8"));
this.socket.shutdownOutput(); // 告訴服務器，所有的發(fā)送動作已經結束，之后只能接收
this.inputStream = socket.getInputStream();
String receipt = new String(inputStream.readAllBytes(), "UTF-8");
System.out.println("got receipt: " + receipt);
this.inputStream.close();
this.socket.close();
}注意這里我們在服務端接受到消息以及客戶端發(fā)送消息后，分別調用了
shutdownInput()和shutdownOutput()而不是直接close對應的stream，這是因為在關閉任何一個stream，都會直接導致socket的關閉，也就無法進行后面回執(zhí)的發(fā)送了。但是注意，調用
shutdownInput()和shutdownOutput()之后，對應的流也會被關閉，不能再次向socket發(fā)送/寫入了。
2. 發(fā)送更多的消息：結束的界定
剛才的兩個例子中，每次打開流，都只能進行一次寫入/讀取操作，結束后對應流被關閉，就無法再次寫入/讀取了。
在這種情況下，如果要發(fā)送兩次消息，就不得不建立兩個socket，既耗資源又麻煩。其實我們完全可以不關閉對應的流，只要分次寫入消息就可以了。
但是這樣的話，我們就必須面對另一個問題：如何判斷一次消息發(fā)送的結束呢？
2.1 使用特殊符號
最簡單的辦法是使用一些特殊的符號來標記一次發(fā)送完成，服務端只要讀到對應的符號就可以完成一次讀取，然后進行相關的處理操作。
下面的例子中我們使用換行符\n來標記一次發(fā)送的結束，服務端每接收到一個消息，就打印一次，并且使用了Scanner來簡化操作:
服務端:
public void runServer() throws IOException {
this.server = new ServerSocket(this.port);
System.out.println("base socket server started.");
this.socket = server.accept();
// the code will block here till the request come.
this.inputStream = this.socket.getInputStream();
Scanner sc = new Scanner(this.inputStream);
while (sc.hasNextLine()) {
System.out.println("get info from client: " + sc.nextLine());
} // 循環(huán)接收并輸出消息內容
this.inputStream.close();
socket.close();
}客戶端:
public void connetServer() throws IOException {
this.socket = new Socket(this.serverHost, this.serverPort);
this.outputStream = socket.getOutputStream();
}
public void send(String message) throws IOException {
String sendMsg = message + "\n"; // we mark \n as a end of line.
try {
this.outputStream.write(sendMsg.getBytes("UTF-8"));
} catch (UnsupportedEncodingException e) {
System.out.println(e.getMessage());
}
// this.outputStream.close();
// this.socket.shutdownOutput();
}
public static void main(String[] args) {
CycleSocketClient cc = new CycleSocketClient("127.0.0.1", 9799);
try {
cc.connetServer();
Scanner sc = new Scanner(System.in);
while (sc.hasNext()) {
String line = sc.nextLine();
cc.send(line);
}
}catch (IOException e) {
e.printStackTrace();
}
}運行后效果是，客戶端每輸入一行文字按下回車后，服務端就會打印出對應的消息讀取記錄。
2.2 根據長度界定
回到原點，我們之所以不好定位消息什么時候結束，是因為我們不能夠確定每次消息的長度。
那么其實可以先將消息的長度發(fā)送出去，當服務端知道消息的長度后，就能夠完成一次消息的接收了。
總的來說，發(fā)送一次消息變成了兩個步驟
- 發(fā)送消息的長度
- 發(fā)送消息
最后的問題就是，“發(fā)送消息的長度”這一步驟所發(fā)送的字節(jié)量必須是固定的，否則我們仍然會陷入僵局。
一般來說，我們可以使用固定的字節(jié)數來保存消息的長度，比如規(guī)定前2個字節(jié)就是消息的長度，不過這樣我們能夠傳送的消息最大長度也就被固定死了，以2個字節(jié)為例，我們發(fā)送的消息最大長度不超過2^16個字節(jié)即64K。
如果你了解一些字符的編碼，就會知道，其實我們可以使用變長的空間來儲存消息的長度，比如：
第一個字節(jié)首位為0：即0XXXXXXX，表示長度就一個字節(jié)，最大128，表示128B
第一個字節(jié)首位為110，那么附帶后面一個字節(jié)表示長度：即110XXXXX 10XXXXXX，最大2048，表示2K
第一個字節(jié)首位為1110，那么附帶后面二個字節(jié)表示長度：即110XXXXX 10XXXXXX 10XXXXXX，最大131072，表示128K
依次類推
當然這樣實現起來會麻煩一些，因此下面的例子里我們仍然使用固定的兩個字節(jié)來記錄消息的長度。
服務端:
public void runServer() throws IOException {
this.serverSocket = new ServerSocket(this.port);
this.socket = serverSocket.accept();
this.inputStream = socket.getInputStream();
byte[] bytes;
while (true) {
// 先讀第一個字節(jié)
int first = inputStream.read();
if (first == -1) {
// 如果是-1，說明輸入流已經被關閉了，也就不需要繼續(xù)監(jiān)聽了
this.socket.close();
break;
}
// 讀取第二個字節(jié)
int second = inputStream.read();
int length = (first << 8) + second; // 用位運算將兩個字節(jié)拼起來成為真正的長度
bytes = new byte[length]; // 構建指定長度的字節(jié)大小來儲存消息即可
inputStream.read(bytes);
System.out.println("receive message: " + new String(bytes,"UTF-8"));
}
}客戶端:
public void connetServer() throws IOException {
this.socket = new Socket(host,port);
this.outputStream = socket.getOutputStream();
}
public void sendMessage(String message) throws IOException {
// 首先要把message轉換成bytes以便處理
byte[] bytes = message.getBytes("UTF-8");
// 接下來傳輸兩個字節(jié)的長度，依然使用移位實現
int length = bytes.length;
this.outputStream.write(length >> 8); // write默認一次只傳輸一個字節(jié)
this.outputStream.write(length);
// 傳輸完長度后，再正式傳送消息
this.outputStream.write(bytes);
}
public static void main(String[] args) {
LengthSocketClient lc = new LengthSocketClient("127.0.0.1",9799);
try {
lc.connetServer();
Scanner sc = new Scanner(System.in);
while (sc.hasNextLine()) {
lc.sendMessage(sc.nextLine());
}
} catch (IOException e) {
e.printStackTrace();
}
}3. 處理更多的連接：多線程
3.1 同時實現消息的發(fā)送與接收
在考慮服務端處理多連接之前，我們先考慮使用多線程改造一下原有的一對一對話實例。
在原有的例子中，消息的接收方并不能主動地向對方發(fā)送消息，換句話說我們并沒有實現真正的互相對話，這主要是因為消息的發(fā)送和接收這兩個動作并不能同時進行，因此我們需要使用兩個線程，其中一個用于監(jiān)聽鍵盤輸入并將其寫入socket，另一個則負責監(jiān)聽socket并將接受到的消息顯示。
出于簡單考慮，我們直接讓主線程負責鍵盤監(jiān)聽和消息發(fā)送，同時另外開啟一個線程用于拉取消息并顯示。
消息拉取線程 ListenThread.java
public class ListenThread implements Runnable {
private Socket socket;
private InputStream inputStream;
public ListenThread(Socket socket) {
this.socket = socket;
}
@Override
public void run() throws RuntimeException{
try {
this.inputStream = socket.getInputStream();
} catch (IOException e) {
e.printStackTrace();
throw new RuntimeException(e.getMessage());
}
while (true) {
try {
int first = this.inputStream.read();
if (first == -1) {
// 輸入流已經被關閉，無需繼續(xù)讀取
throw new RuntimeException("disconnected.");
}
int second = this.inputStream.read();
int msgLength = (first<<8) + second;
byte[] readBuffer = new byte[msgLength];
this.inputStream.read(readBuffer);
System.out.println("message from [" + socket.getInetAddress() + "]: " + new String(readBuffer,"UTF-8"));
} catch (IOException e) {
e.printStackTrace();
throw new RuntimeException(e.getMessage());
}
}
}
}主線程，啟動時由用戶選擇是作為server還是client：
public class ChatSocket {
private String host;
private int port;
private Socket socket;
private ServerSocket serverSocket;
private OutputStream outputStream;
// 以服務端形式啟動，創(chuàng)建會話
public void runAsServer(int port) throws IOException {
this.serverSocket = new ServerSocket(port);
System.out.println("[log] server started at port " + port);
// 等待客戶端的加入
this.socket = serverSocket.accept();
System.out.println("[log] successful connected with " + socket.getInetAddress());
// 啟動監(jiān)聽線程
Thread listenThread = new Thread(new ListenThread(this.socket));
listenThread.start();
waitAndSend();
}
// 以客戶端形式啟動，加入會話
public void runAsClient(String host, int port) throws IOException {
this.socket = new Socket(host, port);
System.out.println("[log] successful connected to server " + socket.getInetAddress());
Thread listenThread = new Thread(new ListenThread(this.socket));
listenThread.start();
waitAndSend();
}
public void waitAndSend() throws IOException {
this.outputStream = this.socket.getOutputStream();
Scanner sc = new Scanner(System.in);
while (sc.hasNextLine()) {
this.sendMessage(sc.nextLine());
}
}
public void sendMessage(String message) throws IOException {
byte[] msgBytes = message.getBytes("UTF-8");
int length = msgBytes.length;
outputStream.write(length>>8);
outputStream.write(length);
outputStream.write(msgBytes);
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
ChatSocket chatSocket = new ChatSocket();
System.out.println("select connect type: 1 for server and 2 for client");
int type = Integer.parseInt(scanner.nextLine().toString());
if (type == 1) {
System.out.print("input server port: ");
int port = scanner.nextInt();
try {
chatSocket.runAsServer(port);
} catch (IOException e) {
e.printStackTrace();
}
}else if (type == 2) {
System.out.print("input server host: ");
String host = scanner.nextLine();
System.out.print("input server port: ");
int port = scanner.nextInt();
try {
chatSocket.runAsClient(host, port);
} catch (IOException e) {
e.printStackTrace();
}
}
}
}3.2 使用線程池優(yōu)化服務端并發(fā)能力
作為服務端，如果一次只跟一個客戶端建立socket連接，未免顯得太過浪費資源，因此我們完全可以讓服務端和多個客戶端建立多個socket。
那么既然要處理多個連接，就不得不面對并發(fā)問題了（當然，你也可以寫循環(huán)輪流處理）。我們可以使用多線程來處理并發(fā)，不過線程的創(chuàng)建和銷毀都會消耗大量的資源和時間，所以最好一步到位，用一個線程池來實現。
下面給出一個示范性質的服務端代碼：
public class SocketServer {
public static void main(String args[]) throws Exception {
// 監(jiān)聽指定的端口
int port = 55533;
ServerSocket server = new ServerSocket(port);
// server將一直等待連接的到來
System.out.println("server將一直等待連接的到來");
//如果使用多線程，那就需要線程池，防止并發(fā)過高時創(chuàng)建過多線程耗盡資源
ExecutorService threadPool = Executors.newFixedThreadPool(100);
while (true) {
Socket socket = server.accept();
Runnable runnable=()->{
try {
// 建立好連接后，從socket中獲取輸入流，并建立緩沖區(qū)進行讀取
InputStream inputStream = socket.getInputStream();
byte[] bytes = new byte[1024];
int len;
StringBuilder sb = new StringBuilder();
while ((len = inputStream.read(bytes)) != -1) {
// 注意指定編碼格式，發(fā)送方和接收方一定要統(tǒng)一，建議使用UTF-8
sb.append(new String(bytes, 0, len, "UTF-8"));
}
System.out.println("get message from client: " + sb);
inputStream.close();
socket.close();
} catch (Exception e) {
e.printStackTrace();
}
};
threadPool.submit(runnable);
}
}
}4. 連接?；?/h2>
我想你不難發(fā)現一個問題，那就是當socket連接成功建立后，如果中途發(fā)生異常導致其中一方斷開連接，此時另一方是無法發(fā)現的，只有在再次嘗試發(fā)送/接收消息才會因為拋出異常而退出。
簡單的說，就是我們維持的socket連接，是一個長連接，但我們沒有保證它的時效性，上一秒它可能還是可以用的，但是下一秒就不一定了。
4.1 使用心跳包
保證連接隨時可用的最常見方法就是定時發(fā)送心跳包，來檢測連接是否正常。這對于實時性要求很高的服務而言，還是非常重要的（比如消息推送）。
大體的方案如下：
- 雙方約定好心跳包的格式，要能夠區(qū)別于普通的消息。
- 客戶端每隔一定時間，就向服務端發(fā)送一個心跳包
- 服務端每接收到心跳包時，將其拋棄
- 如果客戶端的某個心跳包發(fā)送失敗，就可以判斷連接已經斷開
- 如果對實時性要求很高，服務端也可以定時檢查客戶端發(fā)送心跳包的頻率，如果超過一定時間沒有發(fā)送可以認為連接已經斷開
4.2 斷開時重連
使用心跳包必然會增加帶寬和性能的負擔，對于普通的應用我們其實并沒有必要使用這種方案，如果消息發(fā)送時拋出了連接異常，直接嘗試重新連接就好了。
跟上面的方案對比，其實這個拋出異常的消息就充當了心跳包的角色。
總的來說，連接是否要?；?，如何?；?，需要根據具體的業(yè)務場景靈活地思考和定制。
到此這篇關于基于Java實現Socket編程入門的文章就介紹到這了,更多相關Java Socket編程入門內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了Spring如何解決單例bean線程不安全的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
使用spring?data的page和pageable如何實現分頁查詢
這篇文章主要介紹了使用spring?data的page和pageable如何實現分頁查詢，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-12-12
這篇文章主要介紹了Java中ArrayList實現原理及基本方法,ArrayList是開發(fā)中非常常用的數據存儲容器之一,其底層是數組實現的,我們可以在集合中存儲任意類型的數據,ArrayList是線程不安全的,擅長隨機訪問元素,插入和刪除較慢,需要的朋友可以參考下2023-08-08
SpringBoot2.0集成WebSocket實現后臺向前端推送信息
這篇文章主要介紹了SpringBoot2.0集成WebSocket實現后臺向前端推送信息,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01
這篇文章主要為大家介紹了java數據結構算法稀疏數組示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-06-06
Java的MyBatis框架中對數據庫進行動態(tài)SQL查詢的教程
這篇文章主要介紹了Java的MyBatis框架中對數據庫進行動態(tài)SQL查詢的教程,講解了MyBatis中一些控制查詢流程的常用語句,需要的朋友可以參考下2016-04-04
Mybatis中SqlSession下的四大對象之執(zhí)行器（executor）
mybatis中sqlsession下的四大對象是指：executor, statementHandler,parameterHandler，resultHandler對象。這篇文章主要介紹了Mybatis中SqlSession下的四大對象之執(zhí)行器（executor）,需要的朋友可以參考下2019-04-04
這篇文章主要介紹了Java線程安全中的有序性，在開發(fā)中，我們通常按照從上到下的順序編寫程序指令，并且希望cpu和編譯器按照我們預先編寫的順序去執(zhí)。但往往cpu和編譯器為了提高性能、優(yōu)化指令的執(zhí)行順序，會將我們編寫好的程序指令進行重排序2023-02-02
這篇文章主要給大家介紹了關于springboot實現登錄功能的完整步驟,在web應用程序中,用戶登錄權限驗證是非常重要的一個步驟,文中通過代碼以及圖文介紹的非常詳細,需要的朋友可以參考下2023-09-09
這篇文章主要給大家介紹了關于mybatis攔截器實現通用權限字段添加的相關資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用mybatis具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧2019-09-09