loading黑马程序员之网络编程：文件传输（3-2）

package com.heima.net;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;

public class TcpUploadImgClient {

public static void main(String[] args) {
try {
Socket socket = new Socket("127.0.0.1", 10003);
new Thread(new ImgRead(socket)).start();
new Thread(new ImgWrite(socket)).start();

} catch (Exception e) {
e.printStackTrace();
}
}

}
class ImgRead implements Runnable {

private Socket socket;

public ImgRead(Socket socket) {
this.socket = socket;
}

@Override
public void run() {
try {
BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
String line = null;
while ((line = br.readLine()) != null) {
System.out.println("服务器" + socket.getInetAddress().getHostName()
+ "端口:" + socket.getPort() + "发来了报文："+line);
}
} catch (IOException e) {
e.printStackTrace();
}
}

}

class ImgWrite implements Runnable {

private Socket socket;

public ImgWrite(Socket socket) {
this.socket = socket;
}

@Override
public void run() {
BufferedInputStream bis = null;
BufferedOutputStream bos = null;
try {
bis = new BufferedInputStream(new FileInputStream(new File("1.jpg")));
bos = new BufferedOutputStream(socket.getOutputStream());
byte[] buf = new byte[1024];
int len = 0;
while((len = bis.read(buf))!=-1){
bos.write(buf, 0, len);
bos.flush();
}
//定义标记，告诉服务器，文件已写完
socket.shutdownOutput();
//socket.close();
} catch (Exception e) {
e.printStackTrace();
}finally{
try {
bis.close();
} catch (Exception e) {
e.printStackTrace();
}
}

}

}


package com.heima.net;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;

public class TcpUploadImgServer {

public static void main(String[] args) {
try {
ServerSocket ss = new ServerSocket(10004);
while(true){
Socket socket = ss.accept();
new Thread(new ImgServerRead(socket)).start();
}

} catch (Exception e) {
e.printStackTrace();
}
}
}

class ImgServerRead implements Runnable {

private Socket socket;

public ImgServerRead(Socket socket) {
this.socket = socket;
}

@Override
public void run() {
BufferedInputStream bis = null;
BufferedOutputStream bos = null;
try {
bis = new BufferedInputStream(socket.getInputStream());
String ip = socket.getInetAddress().getHostName();
int count = 0;
File file = new File(ip+".jpg");
//如果文件存在
while(file.exists()){
file = new File(ip+"("+(++count)+")"+".jpg");
}
bos = new BufferedOutputStream(new FileOutputStream(file));

PrintWriter out = new PrintWriter(socket.getOutputStream(),true);
byte[] buf = new byte[1024];
int len = 0;
while ((len = bis.read(buf)) != -1) {
bos.write(buf, 0, len);
bos.flush();
}
out.println("文件复制完毕");
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
bos.close();
socket.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}

}