sendfilethread.java

来自「一个仿qq的程序源码 一个用纯java开发的」· Java 代码 · 共 71 行

package qianqian.p2pchat.filetrans;

import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.FileInputStream;

import qianqian.p2pchat.constant.Const;
import qianqian.p2pchat.io.SafeBufferedReader;

public class SendfileThread extends Thread {
	private OutputStream out;
	private SafeBufferedReader in;
	private java.net.Socket socket;
	private static boolean OK = true;

	public SendfileThread(java.net.Socket socket) {
		this.socket = socket;
	}

	public static void forceOff() {
		OK = false;
	}

	private void closeConnection() {
		try {
			if(in != null)
				in.close();
			if(out != null)
				out.close();
			if(socket != null && !socket.isClosed())
				socket.close();
		} catch (Exception e) {
		}
	}

	public void run() {
		int len = 0, count = 10;
		try {
			in = new SafeBufferedReader(new InputStreamReader(socket
					.getInputStream()));
			out = socket.getOutputStream();

			String fileName = null;
			while (true) {
				--count;
				fileName = in.readLine();
				if (fileName != null) {
					break;
				} else if (count < 0) {
					closeConnection();
					return;
				} else {
					sleep(Const.WaitTimeOff);
				}
			}
			byte[] buffer = new byte[Const.FileBufSize];
			FileInputStream stream = new FileInputStream(fileName);
			while ((len = stream.read(buffer)) != -1 && OK) {
				out.write(buffer, 0, len);
				out.flush();
			}
			stream.close();
			if (OK) {
				sleep(Const.WaitTimeOff);
			}
			closeConnection();
		} catch (Exception e) {
		}
	}
}