用socket自定义简单协议实现文件上传与接受

一个上传的数据包，主要包含文件头和文件内容俩部分，主要按下面的格式，发送：

"File-Name:xxxxxx.zip;File-Type:exe;File-Length:1029292\r\n"

------文件内容---------

1、服务端的文件接受服务 MySockerServer

package com.my.socket.server;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import com.my.socket.constant.MyServerConstant;
import com.my.socket.util.StreamUtil;

public final class MySockerServer {

    private ServerSocket myServerSocket = null;
    private File savedir = null;

    public static void main(String[] args) {
	new MySockerServer().start();
    }

    private void start() {
	this.savedir = new File(MyServerConstant.SAVE_DIRECTORY);
	if (!savedir.exists())
	    savedir.mkdir();
	try {
	    this.myServerSocket = new ServerSocket(MyServerConstant.PORT, 50);
	    while (true) {
		Socket mySocket = myServerSocket.accept();
		ExecutorService executorService = Executors
			.newFixedThreadPool(8);
		executorService.execute(new ServerHander(mySocket, savedir));
	    }
	} catch (IOException e) {
	    e.printStackTrace();
	    throw new RuntimeException("启动文件接受服务出错！", e);
	} finally {
	    if (this.myServerSocket != null) {
		try {
		    this.myServerSocket.close();
		} catch (IOException e) {
		    e.printStackTrace();
		    throw new RuntimeException("文件接受服务关闭出错！", e);
		}
	    }
	}
    }

    private final class ServerHander implements Runnable {
	private Socket mySocket = null;
	private File saveFileDirectory = null;

	public ServerHander(Socket mySocket, File saveFileDirectory) {
	    super();
	    this.mySocket = mySocket;
	    this.saveFileDirectory = saveFileDirectory;
	}

	@Override
	public void run() {
	    try {
		InputStream in = mySocket.getInputStream();
		byte[] headData = StreamUtil.readHead(in);
		Map<String, String> headParamter = this.pareHeader(new String(
			headData, MyServerConstant.UTF_8));
		File saveFile = new File(this.saveFileDirectory,
			headParamter.get(MyServerConstant.FILE_NAME));
		if (!saveFile.exists())
		    saveFile.createNewFile();
		OutputStream fos = new FileOutputStream(saveFile);
		byte[] buffer = new byte[1024];
		int len = 0;
		while ((len = in.read(buffer, 0, 1024)) != -1) {
		    fos.write(buffer, 0, len);
		}
		fos.flush();
		fos.close();
		if (mySocket != null)
		    mySocket.close();
	    } catch (IOException e) {
		e.printStackTrace();
	    }
	}

	private Map<String, String> pareHeader(String param) {
	    Map<String, String> headParamter = new HashMap<String, String>();
	    if (param == null || "".endsWith(param))
		throw new RuntimeException("上传的包格式错误的！");
	    String[] params = param.split(";");
	    for (String s : params) {
		String[] entry = s.split(":");
		headParamter.put(entry[0], entry[1]);
	    }
	    return headParamter;

	}
    }
}

2、服务端的辅助类StreamUtil，用于读取文件头信息，在出现"\r\n"后，认为是头信息结束位置。

package com.my.socket.util;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

public final class StreamUtil {
    private StreamUtil() {
    }

    public static byte[] readHead(InputStream in) {
	ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
	char c;
	StringBuffer sb = new StringBuffer();
	try {
	    while ((c = (char) in.read()) != -1) {
		if (c == '\r') {
		    sb.append(c);
		    continue;
		}
		if (c == '\n') {
		    if (sb.charAt(0) == '\r')
			break;
		    else {
			byteArrayOutputStream.write(sb.charAt(0));
			byteArrayOutputStream.write(c);
			sb.delete(0, sb.length());
		    }
		}
		byteArrayOutputStream.write(c);
	    }
	} catch (IOException e) {
	    e.printStackTrace();
	}
	return byteArrayOutputStream.toByteArray();
    }
}

3、服务端启动的参数设置类MyServerConstant：

package com.my.socket.constant;

public final class MyServerConstant {
    public final static String UTF_8 = "UTF-8";
    public final static String FILE_NAME = "File-Name";
    public final static int PORT = 10243;
    public final static String SAVE_DIRECTORY = "C:\\save";
}

4、客户端的实现主要是启动俩个线程来完成，当连接到服务器后，启动一线程用于组装数据包并负责发送，

另一线程负责监听发送任务是否完成，当发现发送任务完成立即退出并关闭连接。

5、客户端类 MySockerClient

package com.my.socket.client;

import java.io.File;
import java.net.InetAddress;
import java.net.Socket;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import com.my.socket.constant.MyClientConstant;

public final class MySockerClient {
    private static boolean finish = false;

    public static boolean isFinish() {
	return finish;
    }

    public static void setFinish(boolean finish) {
	MySockerClient.finish = finish;
    }

    public static void main(String[] args) {
	try {
	    Socket mySocket = new Socket(
		    InetAddress.getByName(MyClientConstant.HOST_URL),
		    MyClientConstant.HOST_PORT);
	    File sendFile = new File(MyClientConstant.SEND_FILE);
	    if (!sendFile.exists())
		throw new RuntimeException("发送的文件不存在，请检查输入的路径");
	    else {
		ExecutorService service = Executors.newFixedThreadPool(2);
		service.execute(new SendFileHander(mySocket, sendFile));
		service.execute(new SendFinishListenerHander(mySocket));
	    }

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

6、负责发送的线程 SendFileHander

package com.my.socket.client;

import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;

import com.my.socket.constant.MyClientConstant;

public class SendFileHander implements Runnable {

    private Socket mySocket = null;
    private File sendFile = null;

    public SendFileHander(Socket mySocket, File sendFile) {
	this.mySocket = mySocket;
	this.sendFile = sendFile;
    }

    @Override
    public void run() {
	FileInputStream fos = null;
	try {
	    fos = new FileInputStream(this.sendFile);
	    byte[] buffer = new byte[1024];
	    int len = 0;
	    OutputStream outputStream = mySocket.getOutputStream();
	    Map<String, String> headParamter = new HashMap<String, String>();
	    headParamter.put(MyClientConstant.FILE_NAME,
		    this.sendFile.getName());
	    outputStream.write(this.getHeadBytes(headParamter).getBytes(
		    MyClientConstant.UTF_8));
	    while ((len = fos.read(buffer, 0, 1024)) != -1) {
		outputStream.write(buffer, 0, len);
	    }
	    outputStream.flush();
	    MySockerClient.setFinish(true);
	} catch (Exception e) {
	    e.printStackTrace();
	}
    }

    private String getHeadBytes(Map<String, String> headParamter) {
	StringBuffer sb = new StringBuffer();
	for (Map.Entry<String, String> entry : headParamter.entrySet())
	    sb.append(entry.getKey() + ":" + entry.getValue() + ";");
	sb.append("\r\n");
	return sb.toString();
    }
}

7、负责监听的线程 SendFinishListenerHander

package com.my.socket.client;

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

public class SendFinishListenerHander implements Runnable {

    private Socket mySocket = null;

    public SendFinishListenerHander(Socket mySocket) {
	super();
	this.mySocket = mySocket;
    }

    @Override
    public void run() {
	while (true) {
	    if (MySockerClient.isFinish()) {
		if (mySocket != null)
		    try {
			mySocket.close();
		    } catch (IOException e) {
			e.printStackTrace();
		    }
		MySockerClient.setFinish(false);
		break;
	    }
	}
    }
}

8、启动客服端的配置信息类 MyClientConstant

 

package com.my.socket.constant;

public final class MyClientConstant {

    public final static String SEND_FILE = "C:\\jboss-4.2.3.GA-jdk6.zip";
    public final static String HOST_URL = "192.168.2.92";
    public final static int HOST_PORT = 10243;

    public static final String UTF_8 = "UTF-8";
    public static final String FILE_NAME = "File-Name";
    public static final String FILE_TYPE = "File-Type";
    public static final String FILE_LENGTH = "File-Length";
}