mcs/class/System.Web/System.Web/TimeoutManager.cs

   1 //
   2 // System.Web.TimeoutManager
   3 //
   4 // Authors:
   5 //      Gonzalo Paniagua Javier (gonzalo@ximian.com)
   6 //
   7 // (C) 2003 Novell, Inc (http://www.novell.com)
   8 //
   9
  10 //
  11 // Permission is hereby granted, free of charge, to any person obtaining
  12 // a copy of this software and associated documentation files (the
  13 // "Software"), to deal in the Software without restriction, including
  14 // without limitation the rights to use, copy, modify, merge, publish,
  15 // distribute, sublicense, and/or sell copies of the Software, and to
  16 // permit persons to whom the Software is furnished to do so, subject to
  17 // the following conditions:
  18 //
  19 // The above copyright notice and this permission notice shall be
  20 // included in all copies or substantial portions of the Software.
  21 //
  22 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  23 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  24 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  25 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  26 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  27 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  28 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  29 //
  30
  31 using System;
  32 using System.Collections;
  33 using System.Threading;
  34 using System.Web.Configuration;
  35
  36 namespace System.Web
  37 {
  38         class StepTimeout
  39         {
  40         }
  41
  42         class TimeoutManager
  43         {
  44                 object this_lock = new object();
  45
  46                 Timer timer;
  47                 Hashtable contexts;
  48
  49                 public TimeoutManager ()
  50                 {
  51                         contexts = Hashtable.Synchronized (new Hashtable ());
  52                         timer = new Timer (new TimerCallback (CheckTimeouts), null, 0, 15000);
  53
  54                         /*
  55                          * To prevent the compiler from issuing a warning, we need to keep a
  56                          * reference to the timer so it is not disposed
  57                          */
  58                         if (timer == null)
  59                                 timer = null;
  60                 }
  61
  62                 public void Add (HttpContext context)
  63                 {
  64                         object value = contexts [context];
  65                         if (value == null) {
  66                                 value = Thread.CurrentThread;
  67                         } else if (value is Thread) {
  68                                 ArrayList list = new ArrayList ();
  69                                 list.Add (value);
  70                                 list.Add (Thread.CurrentThread);
  71                                 value = list;
  72                         } else {
  73                                 ArrayList list = (ArrayList) value;
  74                                 list.Add (Thread.CurrentThread);
  75                                 value = list;
  76                         }
  77
  78                         lock (this_lock) {
  79                                 contexts [context] = value;
  80                         }
  81                 }
  82
  83                 public Thread Remove (HttpContext context)
  84                 {
  85                         object value = contexts [context];
  86                         if (value == null)
  87                                 return null;
  88
  89                         if (value is Thread) {
  90                                 lock (this_lock) {
  91                                         contexts.Remove (context);
  92                                 }
  93                                 return (Thread) value;
  94                         }
  95
  96                         ArrayList list = (ArrayList) value;
  97                         Thread result = null;
  98                         if (list.Count > 0) {
  99                                 result = (Thread) list [list.Count - 1];
 100                                 list.RemoveAt (list.Count - 1);
 101                         }
 102
 103                         if (list.Count == 0) {
 104                                 lock (this_lock) {
 105                                         contexts.Remove (context);
 106                                 }
 107                         }
 108
 109                         return result;
 110                 }
 111
 112                 void CheckTimeouts (object state)
 113                 {
 114                         if (contexts.Count == 0) {
 115                                 return;
 116                         }
 117
 118                         DateTime now = DateTime.UtcNow;
 119                         ArrayList clist = new ArrayList ();
 120
 121                         lock (this_lock) { // The lock prevents Keys enumerator from being out of synch
 122                                 clist.AddRange (contexts.Keys);
 123                         }
 124
 125                         foreach (HttpContext context in clist) {
 126                                 if (!context.CheckIfTimeout (now))
 127                                         continue;
 128
 129                                 Thread thread = Remove (context);
 130                                 if (thread != null) // Only if context is removed right after the lock
 131                                         thread.Abort (new StepTimeout ());
 132                         }
 133                 }
 134         }
 135 }
 136