smparkes-eventmachine 0.12.10

data/java/src/com/rubyeventmachine/application/Application.java

@@ -0,0 +1,194 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+/**
+ * 
+ */
+package com.rubyeventmachine.application;
+
+import java.nio.ByteBuffer;
+import java.nio.channels.*;
+import java.util.*;
+import java.io.*;
+import java.net.*;
+import java.net.SocketAddress;
+
+import com.rubyeventmachine.*;
+
+/**
+ * @author francis
+ *
+ */
+public class Application {
+	
+	
+	public class Reactor extends EmReactor {
+
+		private Application application;
+		private TreeMap<Long, Timer> timers;
+		private TreeMap<Long, Connection> connections;
+		private TreeMap<Long, ConnectionFactory> acceptors;
+		/**
+		 * 
+		 */
+		public Reactor (Application app) {
+			application = app;
+			timers = new TreeMap<Long, Timer>();
+			connections = new TreeMap<Long, Connection>();
+			acceptors = new TreeMap<Long, ConnectionFactory>();
+		}
+
+
+		public void eventCallback (long sig, int eventType, ByteBuffer data, long data2) {
+			if (eventType == EM_TIMER_FIRED) {
+				Timer r = timers.remove(data2);
+				if (r != null)
+					r._fire();
+				else
+					throw new RuntimeException ("unable to run unknown timer");
+			}
+			else if (eventType == EM_CONNECTION_COMPLETED) {
+				Connection c = connections.get(sig);
+				if (c != null) {
+					c.connectionCompleted();
+				}
+				else
+					throw new RuntimeException ("connection completed to unknown object");
+
+			}
+			else if (eventType == EM_CONNECTION_UNBOUND) {
+				Connection c = connections.get(sig);
+				if (c != null) {
+					c.unbind();
+				}
+				else
+					throw new RuntimeException ("unbind received on unknown object");
+			}
+			else if (eventType == EM_CONNECTION_ACCEPTED) {
+				ConnectionFactory f = acceptors.get(sig);
+				if (f != null) {
+					Connection c = f.connection();
+					c.signature = data2;
+					c.application = application;
+					connections.put(c.signature, c);
+					c.postInit();
+					//System.out.println (sig+"..."+new String(data.array()));
+				}
+				else
+					throw new RuntimeException ("received connection on unknown acceptor");
+			}
+			else if (eventType == EM_CONNECTION_READ) {
+				Connection c = connections.get(sig);
+				if (c != null) {
+					c.receiveData(data);
+				}
+				else throw new RuntimeException ("received data on unknown object");
+			}
+			else {
+				System.out.println ("unknown event type: " + eventType);
+			}
+		}
+	}
+
+
+	Reactor reactor;
+	
+	public Application() {
+		reactor = new Reactor (this);
+	}
+	public void addTimer (double seconds, Timer t) {
+		t.application = this;
+		t.interval = seconds;
+		long s = reactor.installOneshotTimer ((int)(seconds * 1000));
+		reactor.timers.put(s, t);
+		
+	}
+
+	public void bindConnect (String bindAddr, int bindPort, String host, int port, Connection c) {
+		long s = reactor.connectTcpServer(bindAddr, bindPort, host, port);
+		c.application = this;
+		c.signature = s;
+		reactor.connections.put(s, c);
+		c.postInit();
+	}
+
+	public void connect (String host, int port, Connection c) {
+		bindConnect(null, 0, host, port, c);
+	}
+	
+	public void startServer (SocketAddress sa, ConnectionFactory f) throws EmReactorException {
+		long s = reactor.startTcpServer(sa);
+		reactor.acceptors.put(s, f);
+	}
+	
+	public void stop() {
+		reactor.stop();
+	}
+	public void run() {
+		reactor.run();
+	}
+	public void run (final Runnable r) {
+		addTimer(0, new Timer() {
+			public void fire() {
+				r.run();
+			}
+		});
+		run();
+	}
+	
+	public void sendData (long sig, ByteBuffer bb) {
+		try {
+			reactor.sendData(sig, bb);
+		} catch (IOException e) {}
+	}
+	
+	public void sendDatagram (long sig, ByteBuffer bb, InetSocketAddress target) {
+		reactor.sendDatagram(sig, bb, target.getHostName(), target.getPort());
+	}
+	
+	public void closeConnection (long sig, boolean afterWriting) {
+		reactor.closeConnection(sig, afterWriting);
+	}
+	
+	public void openDatagramSocket (Connection c) {
+		openDatagramSocket (new InetSocketAddress ("0.0.0.0", 0), c);
+	}
+	public void openDatagramSocket (InetSocketAddress addr, Connection c) {
+		try {
+			long s = reactor.openUdpSocket(addr);
+			c.application = this;
+			c.signature = s;
+			reactor.connections.put(s, c);
+			c.postInit();
+		} catch (ClosedChannelException e) {
+		} catch (IOException e) {
+			System.out.println ("Bad Datagram socket "+e+" "+addr);
+			/* TODO, can't catch this here, because it can happen on a bad address */
+		}
+	}
+}

data/java/src/com/rubyeventmachine/application/Connection.java

@@ -0,0 +1,74 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+
+package com.rubyeventmachine.application;
+
+//import java.io.*;
+import java.nio.*;
+import java.net.*;
+//import java.nio.channels.*;
+
+public class Connection {
+	
+	public Application application;
+	public long signature;
+	
+	public void postInit() {}
+	public void connectionCompleted() {}
+	public void unbind() {}
+	public void receiveData (ByteBuffer bytebuffer) {}
+	
+	
+	/**
+	 * Called by user code.
+	 * @param bytebuffer
+	 */
+	public void sendData (ByteBuffer b) {
+		application.sendData(signature, b);
+	}
+	
+	/**
+	 * This is called by user code.
+	 * TODO: don't expose the exception here.
+	 */
+	public void close() {
+		application.closeConnection(signature, false);
+	}
+	/**
+	 * This is called by user code/
+	 */
+	public void closeAfterWriting() {
+		application.closeConnection(signature, true);
+	}
+	
+	public void sendDatagram (ByteBuffer bb, InetSocketAddress addr) {
+		application.sendDatagram (signature, bb, addr);
+	}
+}

data/java/src/com/rubyeventmachine/application/ConnectionFactory.java

@@ -0,0 +1,37 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+
+package com.rubyeventmachine.application;
+
+//import com.rubyeventmachine.*;
+
+public interface ConnectionFactory {
+	public Connection connection();
+}

data/java/src/com/rubyeventmachine/application/DefaultConnectionFactory.java

@@ -0,0 +1,46 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+
+package com.rubyeventmachine.application;
+
+import com.rubyeventmachine.application.ConnectionFactory;
+
+public class DefaultConnectionFactory implements ConnectionFactory {
+
+	/**
+	 * Convenience class. Its connection() method returns an instance of class
+	 * Connection, which is usually overridden. This class is probably most
+	 * useful for unit testing.
+	 */
+	public Connection connection() {
+		return new Connection();
+	}
+
+}

data/java/src/com/rubyeventmachine/application/PeriodicTimer.java

@@ -0,0 +1,38 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+package com.rubyeventmachine.application;
+
+public class PeriodicTimer extends Timer {
+
+	public void _fire() {
+		fire();
+		application.addTimer(interval, this);
+	}
+}

data/java/src/com/rubyeventmachine/application/Timer.java

@@ -0,0 +1,54 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+package com.rubyeventmachine.application;
+
+public class Timer {
+	/**
+	 * User code is expected to call a method on a controlling Application,
+	 * which will fill in this field so subsequent user code can access it.
+	 */
+	public Application application;
+	public double interval;
+	
+	/**
+	 * The reactor calls here, and it may be overridden in subclasses.
+	 * User code should never call this method.
+	 */
+	public void _fire() {
+		fire();
+	}
+
+	/**
+	 * User code is expected to override this method.
+	 */
+	public void fire() {
+	}
+
+}

data/java/src/com/rubyeventmachine/tests/ApplicationTest.java

@@ -0,0 +1,109 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+
+package com.rubyeventmachine.tests;
+
+
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.Assert;
+import java.net.*;
+import java.io.*;
+import java.nio.*;
+
+import com.rubyeventmachine.*;
+import com.rubyeventmachine.application.*;
+
+public class ApplicationTest {
+
+	@BeforeClass
+	public static void setUpBeforeClass() throws Exception {
+	}
+
+	@AfterClass
+	public static void tearDownAfterClass() throws Exception {
+	}
+
+	@Before
+	public void setUp() throws Exception {
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+
+	@Test
+	public void testRunnableArgument() {
+		final Application a = new Application();
+		a.run (new Runnable() {
+			public void run() {
+				a.stop();
+			}
+		});
+	}
+	
+
+	
+	class F implements ConnectionFactory {
+		public Connection connection() {
+			return new Connection() {
+				public void receiveData (ByteBuffer bb) {
+					application.stop();
+				}
+			};
+		}
+		
+	};
+	
+	@Test
+	public void testTcpServer() throws EmReactorException {
+		final Application a = new Application();
+		final SocketAddress saddr = new InetSocketAddress ("127.0.0.1", 9008);
+		a.run (new Runnable() {
+			public void run() {
+				try {
+					a.startServer (saddr, new F());
+				} catch (EmReactorException e) { Assert.fail(); }
+				new Thread() {
+					public void run() {
+						try {
+							Socket s = new Socket ("127.0.0.1", 9008);
+							s.getOutputStream().write(new String ("boo").getBytes());
+						} catch (UnknownHostException e) {
+						} catch (IOException e) {}
+					}
+				}.start();
+			}
+		});
+	}
+}

data/java/src/com/rubyeventmachine/tests/ConnectTest.java

@@ -0,0 +1,148 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * See EventMachine and EventMachine::Connection for documentation and
+ * usage examples.
+ * 
+ *
+ *----------------------------------------------------------------------------
+ *
+ * Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+ * Gmail: blackhedd
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of either: 1) the GNU General Public License
+ * as published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version; or 2) Ruby's License.
+ * 
+ * See the file COPYING for complete licensing information.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+package com.rubyeventmachine.tests;
+
+
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import java.net.*;
+import java.io.*;
+import java.nio.*;
+import java.nio.channels.*;
+
+import com.rubyeventmachine.*;
+import com.rubyeventmachine.application.*;
+
+public class ConnectTest {
+
+	@BeforeClass
+	public static void setUpBeforeClass() throws Exception {
+	}
+
+	@AfterClass
+	public static void tearDownAfterClass() throws Exception {
+	}
+
+	@Before
+	public void setUp() throws Exception {
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+
+	@Test
+	public final void test1() throws IOException, ClosedChannelException {
+		Application a = new Application();
+		a.addTimer(0, new Timer() {
+			public void fire() {
+				application.connect("www.bayshorenetworks.com", 80, new Connection() {
+					public void connectionCompleted() {
+						close();
+					}
+					public void unbind() {
+						application.stop();
+					}
+				});
+			}
+		});
+		a.run();
+	}
+	
+	@Test
+	public final void test2() throws IOException {
+		class Bays extends Connection {
+			public void connectionCompleted() {
+				sendData (ByteBuffer.wrap( new String ("GET / HTTP/1.1\r\nHost: _\r\n\r\n").getBytes()));
+			}
+			public void receiveData (ByteBuffer b) {
+				System.out.println (new String(b.array()));
+				application.stop();
+			}
+		};
+		
+		Application a = new Application();
+		a.addTimer(0, new Timer() {
+			public void fire() {
+				application.connect("www.bayshorenetworks.com", 80, new Bays());
+			}
+		});
+		a.run();
+	}
+
+	public final void testBindConnect() throws IOException {
+		class Server extends Connection {
+			public void postInit() {
+				// TODO: get peername here and check if the port is 33333
+				// doesnt seem like peername is impl yet?
+				System.out.println("post init!");
+			}
+		};
+
+		Application a = new Application();
+		a.addTimer(0, new Timer() {
+			public void fire() {
+				application.startServer(new InetSocketAddress("localhost", 20000), new DefaultConnectionFactory());
+			}
+		});
+		a.addTimer(500, new Timer() {
+			public void fire() {
+				application.bindConnect("localhost", 33333, "localhost", 20000, new Connection());
+			}
+		});
+
+		a.run();
+	}
+
+	class C1 extends Connection {
+		Application application;
+		public C1 (Application a) {
+			application = a;
+		}
+		public void postInit() {
+			application.stop();
+		}
+	}
+	@Test
+	public final void test3() {
+		final Application a = new Application();
+		a.run (new Runnable() {
+			public void run() {
+				a.connect("www.bayshorenetworks.com", 80, new C1(a));
+			}
+		});
+	}
+	
+	
+
+}