使用Socket实现简单的文件上传

客户端

在这里插入代码片 package demo02FileUp; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.net.Socket; /* * 文件上传案例的客户端：读取本地文件，上传到服务器，读取服务器写回的数据 * 实现步骤： * 1.创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源 * 2.创建一个客户端Socket对象，构造方法中绑定服务器的ip地址和端口号 * 3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象 * 4.使用本地字节输入流FileInputStream对象中的方法read读取 本地文件 * 5.使用网络字节输出流OutputStream对象中的方法Write，把读取到的文件上传到服务器 * 6.使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象 * 7.使用网络字节输入流InputStream对象中的方法read读取服务回写的数据 * 8.释放资源（FileInputStream,Socket） * */ public class TCPupload { public static void main(String[] args) throws IOException { //创建一个本地字节输入流FileInputStream对象，构造方法要绑定读取的数据 FileInputStream fis = new FileInputStream("D:1.txt"); //创建一个客户端Socket对象，构造方法要绑定服务器的ip和端口号 Socket sc = new Socket("127.0.0.1",9898); //使用Socket中的方法getOutputString,获取网络字节流的对象 OutputStream os = sc.getOutputStream(); //使用本地字节输入流FileInputStream对象中的方法read读取 本地文件 int len = 0; byte[] bytes = new byte[1024]; while ((len = fis.read(bytes))!=-1){ //使用网络字节输出流OutputStream对象中的方法write，把读取的文件上传到服务器 os.write(bytes,0,len); } /* 解决：上传文件后给服务器一个结束标记 void shutdownOutput() 禁用此套接字的输出流。 对于 TCP 套接字，任何以前写入的数据都将被发送，并且后跟 TCP 的正常连接终止序列。 * */ sc.shutdownOutput(); //使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象 //使用网络字节输入流InputStream对象中的方法read读取服务回写的数据 InputStream is = sc.getInputStream(); len = is.read(bytes); System.out.println(new String(bytes,0,len)); //释放资源 fis.close(); sc.close(); } }

服务器端代码

在这里插入代码片 package demo02FileUp; import java.io.*; import java.net.ServerSocket; import java.net.Socket; /* * 文件上传案例的服务器端：读取客户端的上传的文件，保存在服务器的硬盘,给用户写回上传成功 * 实现步骤： * 1.创建一个服务器ServerSocket对象,和系统要指定的端口号 * 2.使用ServerSocket对象中的方法accept，获取请求到的Socket对象 * 3.Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象 * 4.判断指定的文件夹是否存在，不存在创建文件夹 * 5 创建一个本地字节的输出流FileOutputStream对象，构造方法要绑定输出的目的地 * 6 使用网络字节流InputStream对象中的方法read,读取客户端上传的文件 * 7 使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上 * 8 使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象 * 9 使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功" * 10 关闭服务（FileOutputStream,Socket,ServerSocket） * */ public class TCPServer { public static void main(String[] args) throws IOException { //1.创建一个服务器ServerSocket对象,和系统要指定的端口号 ServerSocket ss = new ServerSocket(9898); //2 使用ServerSocket对象中的方法accept，获取请求到的Socket对象 Socket sc = ss.accept();在这里插入代码片 //3.Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象 InputStream is = sc.getInputStream(); //4.判断指定的文件夹是否存在，不存在创建文件夹 File file = new File("D:\\upload"); if(!file.exists()){ file.mkdirs(); } //5 创建一个本地字节的输出流FileOutputStream对象，构造方法要绑定输出的目的地 FileOutputStream fos = new FileOutputStream(file+"\\1.txt"); //6 使用网络字节流InputStream对象中的方法read,读取客户端上传的文件 byte[] bytes =new byte[1024]; int len = 0; while((len = is.read(bytes))!=-1){ //7使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上 fos.write(bytes,0,len); } //8 使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象 OutputStream os = sc.getOutputStream(); os.write("上传成功".getBytes()); fos.close(); sc.close(); ss.close(); } }

优化服务器端代码，使用多线程使得多个客户端可以同时上传

在这里插入代码片 package demoFIleup; import java.io.*; import java.net.ServerSocket; import java.net.Socket; import java.util.Random; /* * 文件上传案例的服务器端：读取客户端的上传的文件，保存在服务器的硬盘,给用户写回上传成功 * 实现步骤： * 1.创建一个服务器ServerSocket对象,和系统要指定的端口号 * 2.使用ServerSocket对象中的方法accept，获取请求到的Socket对象 * 3.Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象 * 4.判断指定的文件夹是否存在，不存在创建文件夹 * 5 创建一个本地字节的输出流FileOutputStream对象，构造方法要绑定输出的目的地 * 6 使用网络字节流InputStream对象中的方法read,读取客户端上传的文件 * 7 使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上 * 8 使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象 * 9 使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功" * 10 关闭服务（FileOutputStream,Socket,ServerSocket） * */ public class TCPServer1 { public static void main(String[] args) throws IOException { //1.创建一个服务器ServerSocket对象,和系统要指定的端口号 ServerSocket ss = new ServerSocket(9898); /* * 让服务器一直处于监听状态（死循环accpet方法） * 有一个客户端上传文件，就保存一个文件 * */ while(true){ //2 使用ServerSocket对象中的方法accept，获取请求到的Socket对象 Socket sc = ss.accept(); /* * 使用多线程技术，提高程序的运行效率 * 有一个客户端上传文件，就开启一个线程完成文件的上传 * * */ new Thread(new Runnable() { @Override public void run() { try { //3.Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象 InputStream is = sc.getInputStream(); //4.判断指定的文件夹是否存在，不存在创建文件夹 File file = new File("D:\\upload"); if(!file.exists()){ file.mkdirs(); } /* * 自定义一个文件的命名规则：防止同名文件覆盖 * 规则：域名+毫秒值+随机数 * */ String flieName = "itcast"+System.currentTimeMillis()+new Random().nextInt(99999)+"jpg"; //5 创建一个本地字节的输出流FileOutputStream对象，构造方法要绑定输出的目的地 // FileOutputStream fos = new FileOutputStream(file+"\\1.txt"); FileOutputStream fos = new FileOutputStream(file+"\\"+flieName); //6 使用网络字节流InputStream对象中的方法read,读取客户端上传的文件 byte[] bytes =new byte[1024]; int len = 0; System.out.println("33333333"); while((len = is.read(bytes))!=-1){ //7使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上 fos.write(bytes,0,len); } //8 使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象 OutputStream os = sc.getOutputStream(); os.write("上传成功".getBytes()); fos.close(); sc.close(); }catch (IOException e){ System.out.println(e); } } }).start(); } } }