package com.zhou.java1;
import org.junit.Test;
import java.io.*;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
/**
* 实现 TCP 的网络编程
* 例子 3: 客户端发送文件给服务端,服务端将文件保存在本地.并返回"发送成功"给客户端
* @author upzhou
* @create 2022-03-31 17:20
*/
public class TCPTest3 {
/*
这里涉及到的异常应该使用 try-catch-finally 处理
*/
@Test
public void client() throws IOException {
//1. 造 socket
Socket socket = new Socket(InetAddress.getByName("127.0.0.1"), 9090);
//2. 获取输出流
OutputStream os = socket.getOutputStream();
//3. 获取输入流,读取数据
FileInputStream fis = new FileInputStream(new File("小猫.jpg"));
//4. 读写
byte[] buffer = new byte[1024];
int len;
while ((len = fis.read(buffer)) != -1){
os.write(buffer,0,len);
}
//5. 接收来自于服务器端的数据,并显示到控制台上
InputStream is = socket.getInputStream();
ByteArrayOutputStream baos = new ByteArrayOutputStream();
byte[] buffer1 = new byte[20];
int len1;
while ((len1 = is.read(buffer1)) != -1){
baos.write(buffer,0,len1);
}
System.out.println(baos.toString());
//6. 关闭
fis.close();
os.close();
socket.close();
}
@Test
public void server() throws IOException {
//1. 造 serversocket
ServerSocket ss = new ServerSocket(9090);
//2. 获取客服端的 serversocket
Socket socket = ss.accept();
//3. 获取客户端的输入流
InputStream is = socket.getInputStream();
//4. 保存数据到本地
FileOutputStream fos = new FileOutputStream(new File("小猫3.jpg"));
//5.读写
byte[] buffer = new byte[1024];
int len;
while ((len = is.read(buffer)) != -1){
fos.write(buffer,0,len);
}
//6.服务器端给与客户端反馈
OutputStream os = socket.getOutputStream();
os.write("你好,照片我已收到,很可爱!".getBytes());
//7. 关闭
fos.close();
is.close();
socket.close();
ss.close();
}
}