服务端客户端文件下载

1.服务端提供可供下载的标号

package com.softeem.ftp;

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.io.ObjectOutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.TreeMap;

public class FtpServer extends Thread{
	private File dir;//需要被共享的目录
	private Socket s;//当前连接的客户端
	
	public FtpServer(File dir, Socket s) {
		this.dir = dir;
		this.s = s;
	}
	@Override
	public void run() {
		//将目录中的文件列表传输到客户端
		sendList();
		
	
	}
	
	private void sendList(){
		//列出目录中所有的文件名称
		String[] filesName = dir.list();
		//将文件名以键值对形式存储到map集合中
		TreeMap<Integer, String> files = new TreeMap<>();
		for (int i = 0; i < filesName.length; i++) {
			files.put(i+1, filesName[i]);
		}
		try {
			//获取基于Socket的对象输出流(通过Socket向客户端发送文件列表)
			ObjectOutputStream oos = new ObjectOutputStream(s.getOutputStream());
			oos.writeObject(files);
			oos.flush();
			//接受指令
			receiverCommond(files);
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	private void receiverCommond(TreeMap<Integer, String> files) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(s.getInputStream()));
		String s = br.readLine();
		int fno = Integer.parseInt(s);
		String fname = files.get(fno);
		//根据供需目录以及客户端需要下载文件名，创建File对象
		File source = new File(dir,fname);
		//发送文件
		transFile(source);
	}
	private void transFile(File source) throws IOException{
		//获取源文件输入流
		BufferedInputStream bis = new BufferedInputStream(new FileInputStream(source));
		//获取Socket输出流
		BufferedOutputStream bos = new BufferedOutputStream(s.getOutputStream());
		
		byte[] b = new byte[1024];
		int len = 0;
		while(-1 != (len = bis.read(b))){
			bos.write(b, 0, len);
		}
		bis.close();
		bos.close();
	}
	
	public static void main(String[] args) {
	
		try {
			File dir = new File("E:\\larva搞笑虫子");
			ServerSocket ss = new ServerSocket(5678);
			Socket s = null;
			while(true){
				//监听客户端连接
				s = ss.accept();
				System.out.println("客户端已连接"+s.getInetAddress());
				//针对每一个连接的Socket对象，单独开启一个线程提供文件共享服务
				new FtpServer(dir,s).start();
			}
		
		} catch (IOException e) {
			e.printStackTrace();
		}
	
	
	}
}

2.客户端根据标号下载

package com.softeem.ftp;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.ObjectInputStream;
import java.io.PrintStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Set;
import java.util.TreeMap;

public class FtpClient {

	private Socket s;
	private File dir;
	
	public FtpClient(Socket s,File dir) {
		super();
		this.s = s;
		this.dir = dir;
	}

	public void init(){
		try {
			
			/************显示可供下载的文件列表******************/
			//获取服务端传递过来的输入流并包装为对象输入流
			ObjectInputStream ois = new ObjectInputStream(s.getInputStream());
			TreeMap<Integer,String> files = (TreeMap<Integer,String>) ois.readObject();
		
			System.out.println("请输入需要下载文件的编号:");
			 Set<Integer> ids = files.keySet();
			 for (Integer id : ids) {
				String fname = files.get(id);
				System.out.println(id+"---->"+fname);
			}
			 /***********向服务端传输文件编号*************/
			 sendCommond(files);
			 
		}catch (IOException e) {
			e.printStackTrace();
		} catch (ClassNotFoundException e) {
			e.printStackTrace();
		}
	}
	
	private void sendCommond(TreeMap<Integer,String> map) throws IOException{
		BufferedReader br  = new BufferedReader(new InputStreamReader(System.in));
		String comm = br.readLine();
		//检查输入内容是否为数字
		if(comm.matches("^\\d*$")){
			//将输入的编号转换为数字
			int inputNo = Integer.parseInt(comm);
			boolean isExist = false;//将设需要被下载的文件序号是不存在的
			for (int i : map.keySet()) {
				if(inputNo == i){
					isExist = true;
					break;
				}
			}
			if(isExist){
				//发送指令
				realSendComm(inputNo);
				//开始下载
				String fname = map.get(inputNo);
				//客户端存储的文件
				File distFile = new File(dir,fname);
				download(distFile);
					
			}else{
				System.out.println("输入文件编号有误，请重新输入");
				sendCommond(map);
			}
				
		}else{
			System.out.println("输入的文件编号有误，请重新输入");
			sendCommond(map);
		}
		
	}
	//下载文件
	private void download(File distFile) throws IOException {
		System.out.println("开始下载。。。");
		//将Socket中的数据取出，并复制到目标文件中
		BufferedInputStream bis = new BufferedInputStream(s.getInputStream());
	
		BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(distFile));
		byte[] b = new byte[1024];
		int len = 0;
		while(-1 != (len = bis.read(b))){
			bos.write(b, 0, len);
		}
		bos.close();
		bis.close();
		System.out.println("下载完成。。。");
	}

	private void realSendComm(int inputNo) throws IOException {
		PrintStream ps = new PrintStream(s.getOutputStream());
		//将文件编号转换为字符串发送到服务端
		ps.println(String.valueOf(inputNo));
		
	}

	public static void main(String[] args) throws UnknownHostException, IOException {
		File dist = new File("E:/");
		Socket s = new Socket("192.168.46.27",5678);
		new FtpClient(s,dist).init();
	}
}