Socketサンプル

「Socketサンプル」の編集履歴（バックアップ）一覧はこちら

「Socketサンプル」(2009/08/10 (月) 16:50:35) の最新版変更点

追加された行は緑色になります。

削除された行は赤色になります。

 import java.io.BufferedInputStream;
 import java.io.BufferedOutputStream;
 import java.io.IOException;
 import java.net.InetSocketAddress;
 import java.net.ServerSocket;
 import java.net.Socket;
 import java.nio.ByteBuffer;
 
 public class ClientServerSocketSample {
 	private final static int port = 50000;
 
 	static class SimpleServerSession implements Runnable {
 
 		private Socket socket;
 
 		public SimpleServerSession(Socket socket) {
 			this.socket = socket;
 		}
 
 		@Override
 		public void run() {
 
 			try {
 				BufferedInputStream input = new BufferedInputStream(socket
 						.getInputStream());
 				BufferedOutputStream output = new BufferedOutputStream(socket
 						.getOutputStream());
 
 				while (true) {
 					// インプットストリームからバイトを取得
 					byte[] readByte = new byte[1024];
 					int streamSize = -1;
 					RecievedData data = new RecievedData();
 					while ((streamSize = input.read(readByte)) != -1) {
 						ByteBuffer re = ByteBuffer.wrap(new byte[streamSize]);
 						re.put(readByte, 0, streamSize);
 						data.putByte(re.array());
 						if (data.isCompleted()) {
 							break;
 						}
 					}
 
 					ByteBuffer bytebuf = ByteBuffer.wrap(new byte[data
 							.getSize()]);
 					bytebuf.putInt(data.getDataSize());
 					bytebuf.put(data.getData().array());
 					output.write(bytebuf.array());
 					output.flush();
 				}
 			} catch (Exception e) {
 				if (socket != null) {
 					try {
 						socket.close();
 					} catch (IOException e1) {
 						e1.printStackTrace();
 					}
 				}
 			}
 		}
 	}
 
 	public static void main(String[] args) {
 		new Thread(new Runnable() {
 			@Override
 			public void run() {
 				try {
 					ServerSocket server = new ServerSocket(port);
 					while (true) {
 						Socket soket = server.accept();
 						new Thread(new SimpleServerSession(soket)).start();
 						System.out.println("accept soket:" + soket);
 					}
 				} catch (IOException e) {
 					e.printStackTrace();
 				}
 				System.out.println("end");
 			}
 		}).start();
 
 		Socket socket = new Socket();
 		try {
 			socket.connect(new InetSocketAddress(port));
 
 			BufferedOutputStream out = new BufferedOutputStream(socket
 					.getOutputStream());
 			BufferedInputStream input = new BufferedInputStream(socket
 					.getInputStream());
 			for (int i = 0; i < 10000; i++) {
 
 				// 送信
 				String sendString = "sample　日本語";
 				byte[] senddata = sendString.getBytes();
 				ByteBuffer bufData = ByteBuffer.allocate(4 + senddata.length);
 				bufData.putInt(senddata.length);
 				bufData.put(senddata);
 				out.write(bufData.array());
 				out.flush();
 
 				// 受信
 				// インプットストリームからバイトを取得
 				byte[] readByte = new byte[1024];
 				int streamSize = -1;
 				RecievedData data = new RecievedData();
 				while ((streamSize = input.read(readByte)) != -1) {
 					ByteBuffer re = ByteBuffer.wrap(new byte[streamSize]);
 					re.put(readByte, 0, streamSize);
 					data.putByte(re.array());
 					if (data.isCompleted()) {
 						break;
 					}
 				}
 				String recieve = new String(data.getData().array());
 				if (recieve.equals(sendString)) {
 
 				} else {
 					System.err.println("errror");
 				}
 			}
 			System.out.println("終了");
 		} catch (IOException e) {
 			e.printStackTrace();
 		}
 
 	}
 }
 
 class RecievedData {
 	private RecievedState state;
 	private RecievedState sizeData;
 	private RecievedState messageData;
 	private boolean completed;
 
 	public RecievedData() {
 		sizeData = new RecievedState(this, 4);
 		state = sizeData;
 	}
 
 	public void putByte(byte[] data) {
 		state.putByte(data);
 	}
 
 	void full(RecievedState recievedState, ByteBuffer fullData,
 			ByteBuffer restData) {
 		if (recievedState == sizeData) {
 			fullData.position(0);
 			messageData = new RecievedState(this, fullData.getInt());
 			state = messageData;
 			if (restData != null) {
 				putByte(restData.array());
 			}
 		} else if (recievedState == messageData) {
 			if (restData != null) {
 				throw new RuntimeException("データサイズが不一致です");
 			}
 			completed = true;
 		}
 	}
 
 	public ByteBuffer getData() {
 		if (messageData != null) {
 			return messageData.getData();
 		}
 		return null;
 	}
 
 	public boolean isCompleted() {
 		return completed;
 	}
 
 	public int getSize() {
 		if (messageData != null) {
 			return sizeData.getSize() + messageData.getSize();
 		}
 		return sizeData.getSize();
 	}
 
 	public int getDataSize() {
 		if (messageData != null) {
 			return messageData.getSize();
 		}
 		return sizeData.getSize();
 	}
 
 	private class RecievedState {
 
 		private int size;
 
 		public int getSize() {
 			return size;
 		}
 
 		private RecievedData recievedData;
 
 		private ByteBuffer buf;
 
 		private boolean full;
 
 		public RecievedState(RecievedData recievedData, int size) {
 			this.size = size;
 			buf = ByteBuffer.wrap(new byte[size]);
 			this.recievedData = recievedData;
 		}
 
 		public void putByte(byte[] data) {
 			if (full) {
 				throw new RuntimeException("これ以上書きこむことはできません");
 			}
 			int rest = size - buf.position();
 			if (rest == data.length) {
 				buf.put(data, 0, data.length);
 				recievedData.full(this, buf, null);
 				full = true;
 			} else if (rest > data.length) {
 				buf.put(data, 0, data.length);
 			} else {
 				buf.put(data, 0, rest);
 				ByteBuffer restBuf = ByteBuffer.wrap(new byte[data.length
 						- rest]);
 				restBuf.put(data, rest, data.length - rest);
 				recievedData.full(this, buf, restBuf);
 
 				full = true;
 			}
 		}
 
 		public ByteBuffer getData() {
 			return buf;
 		}
 	}
 }