KLALB2.3

src/org/kne/cloud/network/SocketChannelBridge.java

@@ -0,0 +1,83 @@
+package org.kne.cloud.network;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.*;
+import java.nio.channels.SocketChannel;
+
+import org.kne.cloud.network.klalb.KLALBVirtualSocket;
+import org.kne.cloud.network.klalb.KLALBVirtualSocketChannel;
+import org.kne.io.Task;
+public class SocketChannelBridge extends Task{
+	protected SocketChannel a;
+	protected SocketChannel b;
+	protected StreamChannelBridge bridgeAB;
+	protected StreamChannelBridge bridgeBA;
+	public StreamChannelBridge getBridgeAB() {
+		return bridgeAB;
+	}
+	public StreamChannelBridge getBridgeBA() {
+		return bridgeBA;
+	}
+	public SocketChannelBridge(SocketChannel a, SocketChannel b) throws IOException {
+		super();
+		this.a = a;
+		this.b = b;
+		createStreamChannelBridge();
+	}
+	protected void createStreamChannelBridge() throws IOException {
+		bridgeAB = new StreamChannelBridge(a, b);
+		bridgeBA = new StreamChannelBridge(b, a);
+	}
+	@Override
+	protected void runTask() {
+		try {
+		if(a instanceof KLALBVirtualSocketChannel) {
+			((KLALBVirtualSocketChannel) a).associateSocketChannel(b);
+		}else if(b instanceof KLALBVirtualSocketChannel){
+			((KLALBVirtualSocketChannel) b).associateSocketChannel(a);
+		}else {
+			bridgeAB.runAtNewThread("SocketBridge A->B thread");
+			bridgeBA.runAtNewThread("SocketBridge B->A thread");
+			bridgeAB.waitfortask();
+			bridgeBA.waitfortask();
+		
+		}
+		} catch (Exception e) {
+			e.printStackTrace();
+		}finally {
+			//new Exception().printStackTrace();
+			try {
+				a.close();
+			} catch (IOException e) {
+				// TODO 自动生成的 catch 块
+				e.printStackTrace();
+			}
+			try {
+				b.close();
+			} catch (IOException e) {
+				// TODO 自动生成的 catch 块
+				e.printStackTrace();
+			}
+		}
+	}
+	public SocketChannel getA() {
+		return a;
+	}
+	public SocketChannel getB() {
+		return b;
+	}
+	/*protected OutputStream getBOUT() throws IOException {
+		return b.getOutputStream();
+	}
+	protected InputStream getBIN() throws IOException {
+		return b.getInputStream();
+	}
+	protected OutputStream getAOUT() throws IOException {
+		return a.getOutputStream();
+	}
+	protected InputStream getAIN() throws IOException {
+		return a.getInputStream();
+	}*/
+	
+}