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