From jboss-remoting-commits at lists.jboss.org Tue Aug 18 22:02:25 2009
Content-Type: multipart/mixed; boundary="===============2319501879131834850=="
MIME-Version: 1.0
From: jboss-remoting-commits at lists.jboss.org
To: jboss-remoting-commits at lists.jboss.org
Subject: [jboss-remoting-commits] JBoss Remoting SVN: r5369 -
 remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket.
Date: Tue, 18 Aug 2009 22:02:25 -0400
Message-ID: <200908190202.n7J22PZP005856@svn01.web.mwc.hst.phx2.redhat.com>

--===============2319501879131834850==
Content-Type: text/plain; charset="utf-8"
MIME-Version: 1.0
Content-Transfer-Encoding: quoted-printable

Author: ron.sigal(a)jboss.com
Date: 2009-08-18 22:02:25 -0400 (Tue, 18 Aug 2009)
New Revision: 5369

Added:
   remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/Time=
dOutputStream.java
Log:
JBREM-1120: New OutputStream for output timeout facility.

Added: remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/=
TimedOutputStream.java
=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=
=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=
=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D
--- remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/Tim=
edOutputStream.java	                        (rev 0)
+++ remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/Tim=
edOutputStream.java	2009-08-19 02:02:25 UTC (rev 5369)
@@ -0,0 +1,160 @@
+/*
+* JBoss, Home of Professional Open Source
+* Copyright 2009, JBoss Inc., and individual contributors as indicated
+* by the @authors tag. See the copyright.txt in the distribution for a
+* full listing of individual contributors.
+*
+* This is free software; you can redistribute it and/or modify it
+* under the terms of the GNU Lesser General Public License as
+* published by the Free Software Foundation; either version 2.1 of
+* the License, or (at your option) any later version.
+*
+* This software is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+* Lesser General Public License for more details.
+*
+* You should have received a copy of the GNU Lesser General Public
+* License along with this software; if not, write to the Free
+* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+*/
+
+package org.jboss.remoting.transport.socket;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.Timer;
+import java.util.TimerTask;
+
+import org.jboss.logging.Logger;
+
+/**
+ * @author <a href=3D"ron.sigal(a)jboss.com">Ron Sigal</a>
+ * @version $Rev$
+ * <p>
+ * Copyright April 22, 2009
+ * </p>
+ */
+public class TimedOutputStream extends OutputStream
+{
+   static private Timer timer =3D new Timer(true);
+   static private Logger log =3D Logger.getLogger(TimedOutputStream.class);
+   =

+   private OutputStream os;
+   private int outputTimeout;
+   private OutputTimerTask timerTask;
+   private Object lock =3D new Object();
+   =

+   public TimedOutputStream(OutputStream os, int outputTimeout)
+   {
+      this.os =3D os;
+      this.outputTimeout =3D outputTimeout;
+   }
+   =

+   public void close() throws IOException
+   {
+      os.close();
+   }
+   =

+   public void write(int b) throws IOException
+   {
+      synchronized (lock)
+      {
+         if (timerTask =3D=3D null)
+         {
+            try
+            {
+               timerTask =3D new OutputTimerTask(this);
+               timer.schedule(timerTask, outputTimeout);
+               if (log.isTraceEnabled()) log.trace("scheduled OutputTimerT=
ask: " + outputTimeout);
+            }
+            catch (IllegalStateException e)
+            {
+               timer =3D new Timer(true);
+               timer.schedule(new OutputTimerTask(this), outputTimeout);
+               if (log.isTraceEnabled()) log.trace("scheduled OutputTimerT=
ask: " + outputTimeout);
+            }
+         }
+      }
+      =

+      try
+      {
+         os.write(b);
+      }
+      finally
+      {
+         synchronized (lock)
+         {
+            timerTask.cancel();
+            timerTask =3D null;
+         }
+      }
+   }
+   =

+   public void write(byte b[], int off, int len) throws IOException
+   {
+      synchronized (lock)
+      {
+         if (timerTask =3D=3D null)
+         {
+            try
+            {
+               timerTask =3D new OutputTimerTask(this);
+               timer.schedule(timerTask, outputTimeout);
+               if (log.isTraceEnabled()) log.trace(this + " scheduled " + =
timerTask + ": " + outputTimeout);
+            }
+            catch (IllegalStateException e)
+            {
+//               timer =3D new Timer("TimedOutputStreamTimer", true);
+               timer =3D new Timer(true);
+               timer.schedule(new OutputTimerTask(this), outputTimeout);
+               if (log.isTraceEnabled()) log.trace(this + " scheduled " + =
timerTask + ": " + outputTimeout);
+            }
+         }
+      }
+      =

+      try
+      {
+         os.write(b, off, len);
+      }
+      finally
+      {
+         synchronized (lock)
+         {
+            timerTask.cancel();
+            timerTask =3D null;
+         }
+      }
+   }
+   =

+   static class OutputTimerTask extends TimerTask
+   {
+      private TimedOutputStream tos;
+      =

+      public OutputTimerTask(TimedOutputStream tos)
+      {
+         this.tos =3D tos;
+      }
+      =

+      public void run()
+      {
+         try
+         {
+            log.debug(this + " closing: " + tos);
+            tos.close();
+            tos =3D null;
+         }
+         catch (IOException e)
+         {
+            log.debug("unable to close " + tos);
+         }
+      }
+      =

+      public boolean cancel()
+      {
+         tos =3D null;
+         return super.cancel();
+      }
+   }
+}


--===============2319501879131834850==--