2017-03-07 73 views
-3

我试图使用套接字将图像从客户端发送到服务器。我可以将图像发送到服务器目录,但发送的图像不完整。 我应该改变什么?ImageIO只写入图片的一半(套接字)

客户端代码:

Part filePart = request.getPart("file"); 
InputStream fileContent = filePart.getInputStream();   

Socket clientSocket = new Socket("localhost",51519); 
OutputStream outputStream = clientSocket.getOutputStream(); 
try {      
    BufferedImage image = ImageIO.read(fileContent); 

    ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); 
    ImageIO.write(image, "jpg", byteArrayOutputStream); 

    byte[] size = ByteBuffer.allocate(4).putInt(byteArrayOutputStream.size()).array(); 

    outputStream.write(size); 
    outputStream.write(byteArrayOutputStream.toByteArray());     
} 
catch (SocketException ex){ 
    BLL_Common.write_log(ex.toString(), ""); 
} catch (IOException ex) { 
    BLL_Common.write_log(ex.toString(), ""); 
} finally { 
    outputStream.flush();  
} 

Server代码:

String dir = "web/Admin/img/";  
ServerSocket serverSocket = new ServerSocket(51519); 

while(true) { 
try { 
    Socket socket = serverSocket.accept(); 

    InputStream inputStream = socket.getInputStream(); 

    byte[] sizeAr = new byte[4]; 
    inputStream.read(sizeAr); 
    int size = ByteBuffer.wrap(sizeAr).asIntBuffer().get(); 
    byte[] imageAr = new byte[size]; 

    inputStream.read(imageAr); 
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(imageAr));     

    String fileName = String.valueOf(System.currentTimeMillis()); 

    ImageIO.write(image, "jpg", new File(dir + fileName + ".png")); 
} 
catch (SocketException ex) { 
    //... 
} catch (IOException ex) { 
    //... 
} 
} 

这是加载的图像: enter image description here

回答

1

你假设read()填充缓冲区。您需要DataInputStream.readFully()

看一下DataOutputStream.writeInt()就可以了。

当然,您可以直接在套接字输入流上直接使用ImageIO.read()并避免所有这些,并且类似地ImageIO.write()直接套接字输出流。但是,除非原始图像不是JPEG,否则应该只是复制字节,根本不使用ImageIOBufferedImage