文件上传案例阻塞问题和文件上传案例优化
文件上传案例阻塞问题
解决:
void shutdownOutput() 禁用此套接字的输出流
对于TCP套接字,任何以前写入的数据都将被发送, 并且后跟TCP的正常连接终止序列。
文件上传案例优化
/* 文件上传客户端 */ public class TCPFileClient { public static void main(String[] args) throws IOException { //创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源 FileInputStream fis = new FileInputStream("F:\\1.jpg"); //创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号 Socket socket = new Socket("127.0.0.1", 8888); //使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象 OutputStream os = socket.getOutputStream(); //使用本地字节输入流FileInputStream对象中的方法read,读取本地文件 int len = 0; byte[] bytes = new byte[1024]; while ((len=fis.read(bytes))!=-1){ os.write(bytes,0,len); } socket.shutdownOutput(); //使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器 InputStream is = socket.getInputStream(); //使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象 while ((len=is.read(bytes))!=-1){ System.out.println(new String(bytes,0,len)); } //使用网络字节输入流InputStream对象中的方法read读取服务回写的数据 fis.close(); socket.close(); } }
可以使用多线程和循环让服务器一直监听
fileName 自定义名称
/* 文件上传服务器端 */ public class TCPFileServer { public static void main(String[] args) throws IOException { //创建一个服务器ServerSocket对象,和系统要指定的端口号 ServerSocket server = new ServerSocket(8888); while (true){ //使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象 Socket socket = server.accept(); new Thread(new Runnable() { @Override public void run() { try { //使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象 InputStream is = socket.getInputStream(); //判断文件夹是否存在,不存在则创建 File file = new File("F:\\a\\upload"); if (!file.exists()){ file.mkdirs(); } /* 自定义一个文件的命名规则:防止同名的文件被覆盖 规则:域名+毫秒值+随机数 */ String fileName = "itcast"+System.currentTimeMillis()+new Random().nextInt(99999)+".jpg"; //创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地 //FileOutputStream fos = new FileOutputStream(file + "\\1.jpg"); FileOutputStream fos = new FileOutputStream(file + "\\"+fileName); //使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件 int len = 0; byte[] bytes = new byte[1024]; while ((len=is.read(bytes))!=-1){ //使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上 fos.write(bytes,0,len); } //使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象 //使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功" socket.getOutputStream().write("上传成功".getBytes()); fos.close(); socket.close(); }catch (IOException e){ System.out.println(e); } } }).start(); } // server.close(); } }