Java 上传文件 Socket类应用 多线程上传 有注释哈,上传完请看看文件大小是否正常(一)

2014-11-23 23:22:30 · 作者: · 浏览: 1

由于Java上传文件到服务器是流之间的相互流淌,所以文件名是个问题,我采用的首先把文件名传到服务器,使用用户上传的文件名进行保存,

同时进行了文件覆盖问题的操作,若是服务器存在同名文件,则重新命名后保存,这时有个问题!

就是在查找是否有重名文件时客户端已经再上传文件了,所以导致报错不能到达想要的效果,

我采用的是在服务器查找重名文件时,使客户端处于挂起状态,

即服务器重命名完成后才会给客户顿返回一个标志,收到标志客户端才会进行上传文件操作。

具体请看代码哈,欢迎交流!

文件上传使用的是:打印流,比较高效且自动刷新



 
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.PrintStream;
import java.net.Socket;
import java.net.SocketException;

/**
 * 服务器线程类
 * 
 * @author kaifang
 * 
 */
public class ServerUp implements Runnable {
	private Socket socket = null;

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

	@Override
	public void run() {
		BufferedOutputStream outputStream = null;
		try {
			System.out
					.println(socket.getInetAddress().getHostAddress() + " 连接");
			// 读取传过来的文件字节流对象,避免文件名重复覆盖
			BufferedInputStream inputStream = new BufferedInputStream(
					socket.getInputStream());
			// 存储上传文件的文件夹
			File dir = new File("F:\\temp");
			if (!dir.exists()) {
				dir.mkdir();
			}
			// 读取客户端发过来的文件名
			int len = 0;
			byte[] bys = new byte[1024 * 10];
			len = inputStream.read(bys);
			String name = new String(bys, 0, len);
			File file = new File(dir, name);
			int count = 0;
			// 如果服务器存在此文件则重名命名
			while (file.exists()) {
				count++;
				file = new File(dir, name.replaceAll("(.+)\\.(.+)", "$1("
						+ count + "\\).$2"));
			}
			// 告知客户端
			PrintStream writer = new PrintStream(socket.getOutputStream(), true);
			writer.write("ok".getBytes());
			System.out.println(file.getAbsolutePath());
			// 写入文件到服务器字节流
			outputStream = new BufferedOutputStream(new FileOutputStream(file));
			// 循环读写文件到服务器
			while ((len = inputStream.read(bys)) != -1) {
				outputStream.write(bys, 0, len);
				outputStream.flush();
			}
			// 告知客户端接收完毕
			writer.write("上传成功!".getBytes());
			socket.shutdownOutput();
		} catch (SocketException se) {
			System.out.println("客户端连接丢失!\n\t" + se.getMessage());
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			// 关闭资源
			if (outputStream != null) {
				try {
					outputStream.close();
					socket.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}
}

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * 服务器类
 * 
 * @author kaifang
 * 
 */
public class ServerSocketUp {

	public ServerSocketUp() {
		super();
		// TODO Auto-generated constructor stub
	}

	public static void main(String[] args) {
		System.out.println("服务器启动。。。");
		getFiles();
		System.out.println("服务器关闭。。。");
	}

	public static void getFiles() {
		// 创建ServerSocket对象
		ServerSocket server = null;
		try {
			server = new ServerSocket(10001);
		} catch (IOException e1) {
			e1.printStackTrace();
		}
		Socket socket = null;
		while (true) {
			try {
				// 获取服务器Socket对象
				socket = server.accept();
				new Thread(new ServerUp(socket)).start();
			} catch (IOException e) {
				e.printStackTrace();
			} finally {
//				try {
//					// 关闭服务器,一般不会关闭的
//					server.close();
//				} catch (IOException e) {
//					e.printStackTrace();
//				}
			}
		}
	}
}

import java.io.BufferedInputStream;
import java.io.F