Bài này được nâng cấp và phát triển từ bài Ví dụ TCP/IP transfer file
Nội dung chính
Kiến thức cần nhớ
- TCP/IP có cơ chế gửi tin tin cậy, có nghĩa là các gói tin được gửi đi sẽ được check xem đã đến đích hay chưa.
- Trong java, đối tượng java.net.ServerSocket được sử dụng để khởi tạo một TCP server, mỗi khi có client kết nối đến thì đối tượng java.net.Socket được tạo ra để tương tác với client. Đối tượng java.net.Socket cũng được sử dụng từ phía client để kết nối và tương tác với server.
- Phương thức Socket.getOutputStream() trả về OutputStream được sử dụng để gửi dữ liệu.
- Phương thức Socket.getInputStream() trả về InputStream được sử dụng để nhận dữ liệu.
- Việc cài đặt số lần gửi phải tương ứng với số lần nhận.
TCP/IP transfer file example
Đây là một ví dụ về việc sử dụng giao thức TCP/IP để truyền file.
Tạo lớp FileInfo.java
Tạo lớp vn.viettuts.common.FileInfo giống nhau cho cả phía client và server để gửi và nhận thông tin của một file thông qua đối tượng này.
package vn.viettuts.common;
import java.io.Serializable;
public class FileInfo implements Serializable {
private static final long serialVersionUID = 1L;
private String destinationDirectory;
private String sourceDirectory;
private String filename;
private long fileSize;
private int piecesOfFile;
private int lastByteLength;
private byte[] dataBytes;
private String status;
// Constructor
// getter & setter
}
Cài đặt phía server
Cấu trúc TCPServer project:
File: TCPServer.java
package vn.viettuts.server;
import java.io.BufferedOutputStream;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import vn.viettuts.common.FileInfo;
public class TCPServer extends Thread {
// create serverSocket object
private ServerSocket serverSocket;
private int port = 9900;
/**
* run program
*
* @author viettuts.vn
* @param args
*/
public static void main(String[] args) {
TCPServer tcpServer = new TCPServer();
tcpServer.open();
tcpServer.start();
}
/**
* open server
*
* @author viettuts.vn
*/
public void open() {
try {
serverSocket = new ServerSocket(port);
System.out.println("server is open on port " + port);
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* handle event
*
* @author vietuts.vn
*/
public void run() {
while (true) {
Socket server = null;
DataInputStream inFromClient = null;
ObjectInputStream ois = null;
ObjectOutputStream oos = null;
try {
// accept connect from client and create Socket object
server = serverSocket.accept();
System.out.println("connected to "
+ server.getRemoteSocketAddress());
// get greeting from client
inFromClient = new DataInputStream(server.getInputStream());
System.out.println(inFromClient.readUTF());
// receive file info
ois = new ObjectInputStream(server.getInputStream());
FileInfo fileInfo = (FileInfo) ois.readObject();
if (fileInfo != null) {
createFile(fileInfo);
}
// confirm that file is received
oos = new ObjectOutputStream(server.getOutputStream());
fileInfo.setStatus("success");
fileInfo.setDataBytes(null);
oos.writeObject(fileInfo);
} catch (IOException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} finally {
// close all stream
closeStream(ois);
closeStream(oos);
closeStream(inFromClient);
// close session
closeSocket(server);
}
}
}
/**
* create file with fileInfo
*
* @author viettuts.vn
* @param fileInfo
* @return file is created or not
*/
private boolean createFile(FileInfo fileInfo) {
BufferedOutputStream bos = null;
try {
if (fileInfo != null) {
File fileReceive = new File(fileInfo.getDestinationDirectory()
+ fileInfo.getFilename());
bos = new BufferedOutputStream(
new FileOutputStream(fileReceive));
// write file content
bos.write(fileInfo.getDataBytes());
bos.flush();
}
} catch (IOException e) {
e.printStackTrace();
return false;
} finally {
closeStream(bos);
}
return true;
}
/**
* close socket
*
* @author viettuts.vn
*/
public void closeSocket(Socket socket) {
try {
if (socket != null) {
socket.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* close input stream
*
* @author viettuts.vn
*/
public void closeStream(InputStream inputStream) {
try {
if (inputStream != null) {
inputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
/**
* close output stream
*
* @author viettuts.vn
*/
public void closeStream(OutputStream outputStream) {
try {
if (outputStream != null) {
outputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
}
Cài đặt phía client
Cấu trúc TCPClient project:
Không giống như giao thức UDP. Giao thức TCP/IP không giới hạn dung lượng cho gói tin mỗi lần transfer (Nó phụ thuôc vào dung bộ nhớ heap). Nên bạn có thể đính kèm nội dung của file dưới dạng byte stream vào đối tượng FileInfo để transfer một thể.
File: TCPClient.java
package vn.viettuts.client;
import java.io.BufferedInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import javax.swing.JTextArea;
import vn.viettuts.common.FileInfo;
public class TCPClient {
// create Socket object
private Socket client;
private String host;
private int port;
private JTextArea textAreaLog;
public TCPClient(String host, int port, JTextArea textAreaLog) {
this.host = host;
this.port = port;
this.textAreaLog = textAreaLog;
}
/**
* connect to server
*
* @author viettuts.vn
*/
public void connectServer() {
try {
client = new Socket(host, port);
textAreaLog.append("connected to server.\n");
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* send file to server
*
* @param sourceFilePath
* @param destinationDir
*/
public void sendFile(String sourceFilePath, String destinationDir) {
DataOutputStream outToServer = null;
ObjectOutputStream oos = null;
ObjectInputStream ois = null;
try {
// make greeting
outToServer = new DataOutputStream(client.getOutputStream());
outToServer.writeUTF("Hello from " + client.getLocalSocketAddress());
// get file info
FileInfo fileInfo = getFileInfo(sourceFilePath, destinationDir);
// send file
oos = new ObjectOutputStream(client.getOutputStream());
oos.writeObject(fileInfo);
// get confirmation
ois = new ObjectInputStream(client.getInputStream());
fileInfo = (FileInfo) ois.readObject();
if (fileInfo != null) {
textAreaLog.append("send file to server "
+ fileInfo.getStatus() + "\n");
}
} catch (IOException ex) {
ex.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} finally {
// close all stream
closeStream(oos);
closeStream(ois);
closeStream(outToServer);
}
}
/**
* get source file info
*
* @author viettuts.vn
* @param sourceFilePath
* @param destinationDir
* @return FileInfo
*/
private FileInfo getFileInfo(String sourceFilePath, String destinationDir) {
FileInfo fileInfo = null;
BufferedInputStream bis = null;
try {
File sourceFile = new File(sourceFilePath);
bis = new BufferedInputStream(new FileInputStream(sourceFile));
fileInfo = new FileInfo();
byte[] fileBytes = new byte[(int) sourceFile.length()];
// get file info
bis.read(fileBytes, 0, fileBytes.length);
fileInfo.setFilename(sourceFile.getName());
fileInfo.setDataBytes(fileBytes);
fileInfo.setDestinationDirectory(destinationDir);
} catch (IOException ex) {
ex.printStackTrace();
} finally {
closeStream(bis);
}
return fileInfo;
}
/**
* close socket
*
* @author viettuts.vn
*/
public void closeSocket() {
try {
if (client != null) {
client.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* close input stream
*
* @author viettuts.vn
*/
public void closeStream(InputStream inputStream) {
try {
if (inputStream != null) {
inputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
/**
* close output stream
*
* @author viettuts.vn
*/
public void closeStream(OutputStream outputStream) {
try {
if (outputStream != null) {
outputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
}
Cài đặt giao diện:
File: ClientTransferView.java
package vn.viettuts.client;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JTextArea;
import javax.swing.JTextField;
public class ClientTransferView extends JFrame {
private static final long serialVersionUID = 1L;
private JLabel labelHost;
private JTextField textFieldHost;
private JLabel labelPort;
private JTextField textFieldPort;
private JButton btnBrowse;
private JTextField textFieldFilePath;
private JButton btnSendFile;
private JTextArea textAreaResult;
public ClientTransferView() {
setTitle("Client - truyền file bằng giao thức TCP/IP");
labelHost = new JLabel("Host:");
textFieldHost = new JTextField();
labelPort = new JLabel("Port:");
textFieldPort = new JTextField();
labelHost.setBounds(20, 20, 50, 25);
textFieldHost.setBounds(55, 20, 120, 25);
labelPort.setBounds(190, 20, 50, 25);
textFieldPort.setBounds(220, 20, 50, 25);
textFieldFilePath = new JTextField();
textFieldFilePath.setBounds(20, 50, 450, 25);
btnBrowse = new JButton("Browse");
btnBrowse.setBounds(470, 50, 80, 25);
btnSendFile = new JButton("Send File");
btnSendFile.setBounds(20, 80, 80, 25);
textAreaResult = new JTextArea();
textAreaResult.setBounds(20, 110, 490, 150);
add(labelHost);
add(textFieldHost);
add(labelPort);
add(textFieldPort);
add(textFieldFilePath);
add(btnBrowse);
add(btnSendFile);
add(textAreaResult);
setLayout(null);
setSize(600, 350);
setVisible(true);
// thoát chương trình khi tắt window
setDefaultCloseOperation(EXIT_ON_CLOSE);
}
public void chooseFile() {
final JFileChooser fc = new JFileChooser();
fc.showOpenDialog(this);
try {
if (fc.getSelectedFile() != null) {
textFieldFilePath.setText(fc.getSelectedFile().getPath());
}
} catch (Exception e) {
e.printStackTrace();
}
}
public JLabel getLabelHost() {
return labelHost;
}
public void setLabelHost(JLabel labelHost) {
this.labelHost = labelHost;
}
public JTextField getTextFieldHost() {
return textFieldHost;
}
public void setTextFieldHost(JTextField textFieldHost) {
this.textFieldHost = textFieldHost;
}
public JLabel getLabelPort() {
return labelPort;
}
public void setLabelPort(JLabel labelPort) {
this.labelPort = labelPort;
}
public JTextField getTextFieldPort() {
return textFieldPort;
}
public void setTextFieldPort(JTextField textFieldPort) {
this.textFieldPort = textFieldPort;
}
public JButton getBtnBrowse() {
return btnBrowse;
}
public void setBtnBrowse(JButton btnBrowse) {
this.btnBrowse = btnBrowse;
}
public JTextField getTextFieldFilePath() {
return textFieldFilePath;
}
public void setTextFieldFilePath(JTextField textFieldFilePath) {
this.textFieldFilePath = textFieldFilePath;
}
public JButton getBtnSendFile() {
return btnSendFile;
}
public void setBtnSendFile(JButton btnSendFile) {
this.btnSendFile = btnSendFile;
}
public JTextArea getTextAreaResult() {
return textAreaResult;
}
public void setTextAreaResult(JTextArea textAreaResult) {
this.textAreaResult = textAreaResult;
}
}
File: ClientTransferController.java
package vn.viettuts.client;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.JOptionPane;
public class ClientTransferController implements ActionListener {
private ClientTransferView view;
public ClientTransferController(ClientTransferView view) {
this.view = view;
view.getBtnBrowse().addActionListener(this);
view.getBtnSendFile().addActionListener(this);
}
@Override
public void actionPerformed(ActionEvent e) {
if (e.getActionCommand().equals(view.getBtnBrowse().getText())) {
view.chooseFile();
}
if (e.getActionCommand().equals(view.getBtnSendFile().getText())) {
String host = view.getTextFieldHost().getText().trim();
int port = Integer.parseInt(view.getTextFieldPort().getText().trim());
String sourceFilePath = view.getTextFieldFilePath().getText();
if (host != "" && sourceFilePath != "") {
// định nghĩa thư mục đích trên server
String destinationDir = "D:\\server\\";
TCPClient tcpClient = new TCPClient(host, port,
view.getTextAreaResult());
tcpClient.connectServer();
tcpClient.sendFile(sourceFilePath, destinationDir);
tcpClient.closeSocket();
} else {
JOptionPane.showMessageDialog(view, "Host, Port "
+ "và FilePath phải khác rỗng.");
}
}
}
}
File: ClientTransferMain.java
package vn.viettuts.client;
public class ClientTransferMain {
public static void main(String[] args) {
ClientTransferView view = new ClientTransferView();
new ClientTransferController(view);
}
}
Kết quả
Run server và nhập các thông tin trên client host, port, file path.
Click button "Send File":
