##// END OF EJS Templates
ServiceLocator: fixed services cleanup
cin -
r88:c4877ae77cee v2
parent child
Show More
@@ -1,249 +1,248
1 1 using System;
2 2 using System.Collections.Generic;
3 3
4 4 namespace Implab {
5 5 /// <summary>
6 6 /// Коллекция сервисов, позволяет регистрировать и получать сервисы.
7 7 /// </summary>
8 8 public class ServiceLocator: Component, IServiceLocator, IServiceProvider {
9 9 // запись о сервисе
10 10 struct ServiceEntry : IDisposable {
11 11 public object service; // сервис
12 12 public bool shared; // признак того, что сервис НЕ нужно освобождать
13 13 public Func<object> activator; // активатор сервиса при первом обращении
14 14 public Action<object> cleanup; // функция для очистки сервиса
15 15 public List<Type> associated; // ссылки на текущую запись
16 16 public Type origin; // ссылка на оригинальную запись о сервисе
17 17
18 18 #region IDisposable implementation
19 19
20 20 public void Dispose() {
21 21 if (shared)
22 22 return;
23 if (cleanup != null)
24 cleanup(service);
25 else {
26 var d = service as IDisposable;
27 if (d != null)
28 d.Dispose();
29 }
23 if (cleanup != null) {
24 if (service != null)
25 cleanup(service);
26 } else
27 Safe.Dispose(service);
30 28 }
31 29
32 30 #endregion
33 31 }
34 32
35 33 // словарь существующих сервисов
36 34 readonly Dictionary<Type, ServiceEntry> m_services = new Dictionary<Type,ServiceEntry>();
37 35
38 36 /// <summary>
39 37 /// Получает объект предоставляющий сервис <typeparamref name="T"/>.
40 38 /// </summary>
41 39 /// <typeparam name="T">Тип запрашиваемого сервиса</typeparam>
42 40 /// <returns>Объект, реализующий сервис</returns>
43 41 /// <exception cref="KeyNotFoundException">Сервис не зарегистрирован</exception>
44 42 public T GetService<T>() {
45 43 object result;
46 44 if (TryGetService(typeof(T), out result))
47 45 return (T)result;
48 46 throw new ApplicationException (String.Format ("{0} doesn't provide {1} service", this, typeof(T)));
49 47 }
50 48
51 49
52 50 /// <summary>
53 51 /// Пытается получить указанный сервис, в случае, если компонента не предоставляет требуемый сервис
54 52 /// не возникает исключений.
55 53 /// </summary>
56 54 /// <typeparam name="T">Тип требуемого сервиса.</typeparam>
57 55 /// <param name="service">Объект реализующий сервис, или <c>default(T)</c> если такового нет.</param>
58 56 /// <returns><c>true</c> - сервис найден, <c>false</c> - сервис не зарегистрирован.</returns>
59 57 public bool TryGetService<T>(out T service) {
60 58 object result;
61 59 if (TryGetService(typeof(T), out result)) {
62 60 service = (T)result;
63 61 return true;
64 62 }
65 63 service = default(T);
66 64 return false;
67 65 }
68 66
69 67 /// <summary>
70 68 /// Получает объект предоставляющий сервис <paramref name="serviceType"/>
71 69 /// </summary>
72 70 /// <param name="serviceType">Тип запрашиваемого сервиса</param>
73 71 /// <returns>Объект, реализующий сервис</returns>
74 72 /// <exception cref="KeyNotFoundException">Сервис не зарегистрирован</exception>
75 73 public object GetService(Type serviceType) {
76 74 object result;
77 75 if (TryGetService(serviceType, out result))
78 76 return result;
79 77 throw new ApplicationException (String.Format ("{0} doesn't provide {1} service", this, serviceType));
80 78 }
81 79
82 80 /// <summary>
83 81 /// Пытается получить требуемый сервис или совместимый с ним.
84 82 /// </summary>
85 83 /// <returns><c>true</c>, если сервис был найден, <c>false</c> в противном случае..</returns>
86 84 /// <param name="serviceType">Тип запрашиваемого сервиса.</param>
87 85 /// <param name="service">Искомый сервис.</param>
88 86 public virtual bool TryGetService(Type serviceType, out object service) {
89 87 Safe.ArgumentNotNull(serviceType, "serviceType");
90 88 AssertNotDisposed();
91 89
92 90 ServiceEntry se;
93 91 if (!m_services.TryGetValue(serviceType, out se)) {
94 92 // ищем ближайщий объект, реализующий нужный сервис
95 93 Type pt = null;
96 94 foreach (var t in m_services.Keys)
97 95 if (serviceType.IsAssignableFrom(t) && (pt == null || t.IsAssignableFrom(pt)))
98 96 pt = t;
99 97
100 98 if (pt == null) {
101 99 // нет нужного сервиса
102 100 service = null;
103 101 return false;
104 102 }
105 103
106 104 var pe = m_services[pt];
107 105
108 106 // найденная запись может ссылаться на оригинальную запись с сервисом
109 107 if(pe.origin != null) {
110 108 pt = pe.origin;
111 109 pe = m_services[pt];
112 110 }
113 111
114 112 // добавляем список с обратными ссылками
115 113 if (pe.associated == null)
116 114 pe.associated = new List<Type>();
117 115
118 116 pe.associated.Add(serviceType);
119 117
120 118 // обновляем родительскую запись
121 119 m_services[pt] = pe;
122 120
123 121 // создаем запись со ссылкой
124 122 se = new ServiceEntry {
125 123 service = pe.service,
126 124 origin = pt,
127 125 shared = true // предотвращаем множественные попытки освобождения
128 126 };
129 127
130 128 m_services[serviceType] = se;
131 129 }
132 130
133 131 // запись содержит в себе информацию о сервисе
134 132 if (se.service != null) {
135 133 service = se.service;
136 134 return true;
137 135 }
138 136
139 137 // текущая запись является ссылкой
140 138 if (se.origin != null) {
141 139 se.service = GetService(se.origin);
142 140 m_services[serviceType] = se;
143 141 service = se.service;
144 142 return true;
145 143 }
146 144
147 145 // текущая запись не является ссылкой и не имеет информации о сервисе
148 146 // она должна сожержать информацию об активации
149 147 if (se.activator != null) {
150 148 se.service = se.activator();
151 149
152 150 m_services[serviceType] = se;
153 151
154 152 service = se.service;
155 153 return true;
156 154 }
157 155
158 156 service = null;
159 157 return false;
160 158 }
161 159
162 160 /// <summary>
163 161 /// Регистрирует фабрику для активации сервиса по первому требованию.
164 162 /// </summary>
165 163 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
166 164 /// <param name="activator">Фабрика для создания/получения объекта, предоставляющего сервис.</param>
167 165 /// <param name = "cleanup">Метод для освобождения экземпляра сервиса, будет вызыван при освобождении сервис-локатора.</param>
168 166 /// <remarks>При освобождении сервис-локатора, сервисы полученные в результате активации также будут освобождены.</remarks>
169 167 public void Register<T>(Func<T> activator, Action<T> cleanup) {
170 168 Safe.ArgumentNotNull(activator, "activator");
171 169
172 170 AssertNotDisposed();
173 171
174 172 Unregister(typeof(T));
175 173
176 m_services[typeof(T)] = new ServiceEntry {
177 activator = () => activator(),
178 cleanup = instance => cleanup((T)instance)
179 };
174 var serviceEntry = new ServiceEntry();
175 serviceEntry.activator = () => activator();
176 if (cleanup != null)
177 serviceEntry.cleanup = instance => cleanup((T)instance);
178 m_services[typeof(T)] = serviceEntry;
180 179 }
181 180
182 181 public void Register<T>(Func<T> activator) {
183 182 Register(activator, null);
184 183 }
185 184
186 185 /// <summary>
187 186 /// Регистрирует объект, предоставляющий сервис.
188 187 /// </summary>
189 188 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
190 189 /// <param name="service">Объект, предоставляющий сервис.</param>
191 190 /// <exception cref="InvalidOperationException">Указанный сервис уже зарегистрирован.</exception>
192 191 /// <remarks>Сервис-локатором не управляет временем жизни объекта для зарегистрированного сервиса.</remarks>
193 192 public void Register<T>(T service) {
194 193 Register(service, true);
195 194 }
196 195
197 196 /// <summary>
198 197 /// Регистрирует объект, предоставляющий сервис. Повторная регистрация отменяет уже существующую.
199 198 /// </summary>
200 199 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
201 200 /// <param name="service">Объект, предоставляющий сервис.</param>
202 201 /// <param name="shared">Признак того, что объект является разделяемым и сервис-локатор не должен его освобождать.</param>
203 202 public void Register<T>(T service, bool shared) {
204 203 Safe.ArgumentNotNull(service, "service");
205 204
206 205 AssertNotDisposed();
207 206
208 207 Unregister(typeof(T));
209 208
210 209 m_services[typeof(T)] = new ServiceEntry { service = service, shared = shared };
211 210 }
212 211
213 212 public void Unregister(Type serviceType) {
214 213 Safe.ArgumentNotNull(serviceType, "serviceType");
215 214
216 215 AssertNotDisposed();
217 216
218 217 ServiceEntry se;
219 218 if (m_services.TryGetValue(serviceType, out se)) {
220 219 if (se.origin != null) {
221 220 var pe = m_services[se.origin];
222 221 pe.associated.Remove(serviceType);
223 222 }
224 223 // освобождаем ресурсы
225 224 se.Dispose();
226 225 m_services.Remove(serviceType);
227 226
228 227 // убираем связанные записи
229 228 if (se.associated != null)
230 229 foreach (var item in se.associated)
231 230 m_services.Remove(item);
232 231 }
233 232 }
234 233
235 234 /// <summary>
236 235 /// Освобождает зарегистрированные сервисы (которые требуется освобоить).
237 236 /// </summary>
238 237 /// <param name="disposing">Призанак того, что нужно освободить ресурсы.</param>
239 238 protected override void Dispose(bool disposing) {
240 239 if (disposing) {
241 240
242 241 foreach (var entry in m_services.Values)
243 242 entry.Dispose();
244 243
245 244 }
246 245 base.Dispose(disposing);
247 246 }
248 247 }
249 248 } No newline at end of file
General Comments 0
You need to be logged in to leave comments. Login now