이전 포스팅('자바 소켓통신')에서는 소켓통신으로 메세지를 주고받는 Sample 코드를 올렸는데, 이번에는 파일전송이 가능한 코드를
올린다. 이번 샘플 코드는 파일전송과 메세지 통신 모두 가능하며 이전 코드에 비해 휠씬 간단한 코드로 개발되었으므로 소켓통신을
이해하는데 많은 도움이 될 것으로 기대한다.
본 예제는 jdk 1.4, 1.5에서 테스트되었다.
다음은 파일 및 메세지를 전송하는 Socket Client의 소스 코드이다.
/* Author : 황철연 */ import java.io.BufferedInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.Socket;
public class SendClient {
public static void main(String[] args) {
String serverIp = "127.0.0.1";
Socket socket = null;
try {
// 서버 연결
socket = new Socket(serverIp, 7777);
System.out.println("서버에 연결되었습니다.");
// 파일 전송용 클래스 String filePath = "C:/develop/testdata/sender";
//String fileNm = "DSC01129.jpg";
String fileNm = "한글파일테스트.hwp";
FileSender fs = new FileSender(socket, filePath, fileNm);
fs.start();
// 메세지 전송용 클래스 /*
String msg = "사랑해요!";
MsgSender ms = new MsgSender(socket, msg);
ms.start();
*/
} catch (IOException e) {
e.printStackTrace();
}
}
} //파일 전송용 클래스
class FileSender extends Thread {
String filePath;
String fileNm;
Socket socket;
DataOutputStream dos;
FileInputStream fis;
BufferedInputStream bis;
public FileSender(Socket socket, String filePath, String fileNm) {
this.socket = socket;
this.fileNm = fileNm;
this.filePath = filePath;
try {
// 데이터 전송용 스트림 생성
dos = new DataOutputStream(socket.getOutputStream());
} catch (IOException e) {
e.printStackTrace();
}
}
// @Override
public void run() {
try {
//파일전송을 서버에 알린다.('file' 구분자 전송)
dos.writeUTF("file");
dos.flush();
//전송할 파일을 읽어서 Socket Server에 전송
String result = fileRead(dos);
/*test*/System.out.println("result : " + result);
}catch (IOException e){
e.printStackTrace();
}finally{ //리소스 초기화
try { dos.close(); } catch (IOException e) { e.printStackTrace(); }
try { bis.close(); } catch (IOException e) { e.printStackTrace(); }
} }
private String fileRead(DataOutputStream dos){
String result;
try {
dos.writeUTF(fileNm);
/*test*/System.out.println("파일 이름(" + fileNm + ")을 전송하였습니다.");
// 파일을 읽어서 서버에 전송 File file = new File(filePath + "/" + fileNm);
fis = new FileInputStream(file);
bis = new BufferedInputStream(fis);
int len;
int size = 4096;
byte[] data = new byte[size];
while ((len = bis.read(data)) != -1) {
dos.write(data, 0, len);
}
//서버에 전송 dos.flush();
/* -- 먹통된다.
DataInputStream dis = new DataInputStream(socket.getInputStream());
result = dis.readUTF();
if( result.equals("SUCCESS") ){
/*test*/System.out.println("파일 전송 작업을 완료하였습니다.");
/*test*/System.out.println("보낸 파일의 사이즈 : " + file.length());
}else{
/*test*/System.out.println("파일 전송 실패!.");
}
*/
result = "SUCCESS";
} catch (IOException e) {
e.printStackTrace();
result = "ERROR";
}finally{
try { fis.close(); } catch (IOException e) { e.printStackTrace(); }
}
return result;
}
} //메세지 전송용 클래스 class MsgSender extends Thread {
Socket socket;
String msg;
DataOutputStream dos;
FileInputStream fis;
BufferedInputStream bis;
public MsgSender(Socket socket, String msg) {
this.socket = socket;
this.msg = msg;
try {
// 데이터 전송용 스트림 생성
dos = new DataOutputStream(socket.getOutputStream());
} catch (IOException e) {
e.printStackTrace();
}
}
//@Override
public void run() {
try { //파일전송 구분자 전송('msg' 전송) dos.writeUTF("msg");
dos.flush(); //서버에 전송
dos.writeUTF( msg );
dos.flush(); //서버에 전송
/*test*/System.out.println("[" + msg + "] 전송");
} catch (IOException e) {
e.printStackTrace();
}finally{ //리소스 해제
try { dos.close(); } catch (IOException e) { e.printStackTrace(); }
}
}
} |
다음은 Socket Server의 소스 코드이다.
import java.io.BufferedOutputStream;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
public class ReceiveServer {
public static void main(String[] args) {
ServerSocket serverSocket = null;
Socket socket = null;
try {
serverSocket = new ServerSocket(7777);
System.out.println("서버가 시작되었습니다."); //클라이언트와의 연결 대기 루프
while( true ){
System.out.println("새로운 Client의 연결요청을 기다립니다."); // 연결되면 통신용 소켓 생성
socket = serverSocket.accept();
System.out.println("클라이언트와 연결되었습니다.");
// 파일 수신용 클래스 생성 및 시작
Receiver receiver = new Receiver(socket);
receiver.start();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
class Receiver extends Thread {
Socket socket;
DataInputStream dis = null;
FileOutputStream fos = null;
BufferedOutputStream bos = null;
public Receiver(Socket socket) {
this.socket = socket;
}
//@Override
public void run() {
try {
dis = new DataInputStream(socket.getInputStream());
String type = dis.readUTF(); /*type값('file'또는 'msg')을 기준으로 파일이 전송됐는지 문자열이 전송됐는지 구분한다.*/ if(type.equals("file")){
//전송된 파일 쓰기!
String result = fileWrite(dis);
System.out.println("result : " + result);
}else if(type.equals("msg")){
//수신된 메세지 쓰기
String result = getMsg(dis);
System.out.println("result : " + result);
}
//클라이언트에 결과 전송 - 먹통이된다.
//DataOutputStream dos = new DataOutputStream(socket.getOutputStream());
//dos.writeUTF(result);
}catch (IOException e) {
System.out.println("run() Fail!");
e.printStackTrace();
}
}
private String fileWrite(DataInputStream dis){
String result;
String filePath = "C:/develop/testdata/receiver";
try {
System.out.println("파일 수신 작업을 시작합니다.");
// 파일명을 전송 받고 파일명 수정
String fileNm = dis.readUTF();
System.out.println("파일명 " + fileNm + "을 전송받았습니다.");
// 파일을 생성하고 파일에 대한 출력 스트림 생성
File file = new File(filePath + "/" + fileNm);
fos = new FileOutputStream(file);
bos = new BufferedOutputStream(fos);
System.out.println(fileNm + "파일을 생성하였습니다.");
// 바이트 데이터를 전송받으면서 기록
int len;
int size = 4096;
byte[] data = new byte[size];
while ((len = dis.read(data)) != -1) {
bos.write(data, 0, len);
}
//bos.flush();
result = "SUCCESS";
System.out.println("파일 수신 작업을 완료하였습니다.");
System.out.println("받은 파일의 사이즈 : " + file.length());
} catch (IOException e) {
e.printStackTrace();
result = "ERROR";
}finally{
try { bos.close(); } catch (IOException e) { e.printStackTrace(); }
try { fos.close(); } catch (IOException e) { e.printStackTrace(); }
try { dis.close(); } catch (IOException e) { e.printStackTrace(); }
}
return result;
}
private String getMsg(DataInputStream dis){
String result;
try {
System.out.println("파일 수신 작업을 시작합니다.");
// 파일명을 전송 받고 파일명 수정
String msg = dis.readUTF();
System.out.println("msg : " + msg);
result = "SUCCESS";
System.out.println("메세지 수신 작업을 완료하였습니다.");
}catch (IOException e) {
e.printStackTrace();
result = "ERROR";
}finally{
try { dis.close(); } catch (IOException e) { e.printStackTrace(); }
}
return result;
}
} |