TCP文件上传实现

发布时间 2023-03-22 21:15:32作者: 惊鸿宴远赴人间
package edu.wtbu;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
//服务器端
public class Demo05 {
public static void main(String[] args) throws IOException {
//1.创建服务
ServerSocket serverSocket = new ServerSocket(9000);
//2.监听客户端的连接
Socket socket = serverSocket.accept();//阻塞式监听,会一直等待客户端连接
//3.获取输入流
InputStream is = socket.getInputStream();

//4.文件输出
FileOutputStream fos = new FileOutputStream(new File("b.jpg"));
byte[] bytes = new byte[1024];
int len;
while ((len=is.read())!=-1){
fos.write(bytes,0,len);
}


//通知客户端已经接收完毕了
OutputStream os = socket.getOutputStream();
os.write("我接收完毕了".getBytes());
//5.关闭资源

fos.close();
is.close();
socket.close();
serverSocket.close();
}

}


package edu.wtbu;

import java.io.*;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
//客户端
public class Demo04 {
public static void main(String[] args) throws IOException {
//1.创建一个Socket连接
Socket socket = new Socket(InetAddress.getByName("127.0.0.1"), 9000);
//2.创建一个输出流
OutputStream os = socket.getOutputStream();

//3.读取文件
FileInputStream fis = new FileInputStream(new File("a.jpg"));
//4.写出文件
byte[] bytes = new byte[1024];
int len;
while ((len=fis.read(bytes))!=-1){
os.write(bytes,0,len);
}
//通知服务器,我已经结束了
socket.shutdownOutput();//我已经传输完了
//确定服务器接收完毕,才断开连接
InputStream is = socket.getInputStream();
ByteArrayOutputStream baos = new ByteArrayOutputStream();
byte[] bytes1 = new byte[1024];
int len2;
while ((len2=is.read(bytes1))!=-1){
baos.write(bytes1,0,len2);
}
System.out.println(baos.toString());

//5.关闭资源
baos.close();
is.close();
fis.close();
os.close();
socket.close();
}
}