Merge pull request #1668 from alexanderkyte/bug1856
[mono.git] / mcs / class / corlib / System.Runtime.Remoting.Lifetime / LeaseManager.cs
1 //
2 // System.Runtime.Remoting.Identity.cs
3 //
4 // Author: Lluis Sanchez Gual (lluis@ideary.com)
5 //
6 // (C) 2003, Lluis Sanchez Gual
7 //
8
9 //
10 // Copyright (C) 2004 Novell, Inc (http://www.novell.com)
11 //
12 // Permission is hereby granted, free of charge, to any person obtaining
13 // a copy of this software and associated documentation files (the
14 // "Software"), to deal in the Software without restriction, including
15 // without limitation the rights to use, copy, modify, merge, publish,
16 // distribute, sublicense, and/or sell copies of the Software, and to
17 // permit persons to whom the Software is furnished to do so, subject to
18 // the following conditions:
19 // 
20 // The above copyright notice and this permission notice shall be
21 // included in all copies or substantial portions of the Software.
22 // 
23 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
24 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
27 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
28 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
29 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
30 //
31
32 using System;
33 using System.Threading;
34 using System.Collections;
35 using System.Runtime.Remoting;
36
37 namespace System.Runtime.Remoting.Lifetime
38 {
39         internal class LeaseManager
40         {
41                 ArrayList _objects = new ArrayList();
42                 Timer _timer = null;
43
44                 public void SetPollTime (TimeSpan timeSpan)
45                 {
46                         lock (_objects.SyncRoot) 
47                         {
48                                 if (_timer != null)
49                                         _timer.Change (timeSpan,timeSpan);
50                         }
51                 }
52
53                 public void TrackLifetime (ServerIdentity identity)
54                 {
55                         lock (_objects.SyncRoot)
56                         {
57                                 identity.Lease.Activate();
58                                 _objects.Add (identity);
59
60                                 if (_timer == null) StartManager();
61                         }
62                 }
63
64                 public void StopTrackingLifetime (ServerIdentity identity)
65                 {
66                         lock (_objects.SyncRoot)
67                         {
68                                 _objects.Remove (identity);
69                         }
70                 }
71
72                 public void StartManager()
73                 {
74                         _timer = new Timer (new TimerCallback (ManageLeases), null, LifetimeServices.LeaseManagerPollTime,LifetimeServices.LeaseManagerPollTime);
75                 }
76
77                 public void StopManager()
78                 {
79                         Timer t = _timer;
80                         _timer = null;
81                         if (t != null) t.Dispose();
82                 }
83
84                 public void ManageLeases(object state)
85                 {
86                         lock (_objects.SyncRoot)
87                         {
88                                 int n=0;
89                                 while (n < _objects.Count)
90                                 {
91                                         ServerIdentity ident = (ServerIdentity)_objects[n];
92                                         ident.Lease.UpdateState();
93                                         if (ident.Lease.CurrentState == LeaseState.Expired)
94                                         {
95                                                 _objects.RemoveAt (n);
96                                                 ident.OnLifetimeExpired ();
97                                         }
98                                         else
99                                                 n++;
100                                 }
101
102                                 if (_objects.Count == 0) 
103                                         StopManager();
104                         }
105                 }
106         }
107 }