Mercurial > pub > ImplabNet

comparison Implab/Components/ServiceLocator.cs @ 152:240aa6994018 v2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
component model refactoring
author cin
date Thu, 11 Feb 2016 01:56:27 +0300
parents
children bdfdba6b645b
comparison
equal deleted inserted replaced
151:ec91a6dfa5b3 152:240aa6994018
1 using System;
2 using System.Collections.Generic;
3
4 namespace Implab.Components {
5 /// <summary>
6 /// Коллекция сервисов, позволяет регистрировать и получать сервисы.
7 /// </summary>
8 public class ServiceLocator: Disposable, IServiceLocator, IServiceProvider {
9 // запись о сервисе
10 struct ServiceEntry : IDisposable {
11 public object service; // сервис
12 public bool shared; // признак того, что сервис НЕ нужно освобождать
13 public Func<object> activator; // активатор сервиса при первом обращении
14 public Action<object> cleanup; // функция для очистки сервиса
15 public List<Type> associated; // ссылки на текущую запись
16 public Type origin; // ссылка на оригинальную запись о сервисе
17
18 #region IDisposable implementation
19
20 public void Dispose() {
21 if (shared)
22 return;
23 if (cleanup != null) {
24 if (service != null)
25 cleanup(service);
26 } else
27 Safe.Dispose(service);
28 }
29
30 #endregion
31 }
32
33 // словарь существующих сервисов
34 readonly Dictionary<Type, ServiceEntry> m_services = new Dictionary<Type,ServiceEntry>();
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>() {
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)));
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) {
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;
65 }
66
67 /// <summary>
68 /// Получает объект предоставляющий сервис <paramref name="serviceType"/>
69 /// </summary>
70 /// <param name="serviceType">Тип запрашиваемого сервиса</param>
71 /// <returns>Объект, реализующий сервис</returns>
72 /// <exception cref="KeyNotFoundException">Сервис не зарегистрирован</exception>
73 public object GetService(Type serviceType) {
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));
78 }
79
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) {
87 Safe.ArgumentNotNull(serviceType, "serviceType");
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;
97
98 if (pt == null) {
99 // нет нужного сервиса
100 service = null;
101 return false;
102 }
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 // запись содержит в себе информацию о сервисе
132 if (se.service != null) {
133 service = se.service;
134 return true;
135 }
136
137 // текущая запись является ссылкой
138 if (se.origin != null) {
139 se.service = GetService(se.origin);
140 m_services[serviceType] = se;
141 service = se.service;
142 return true;
143 }
144
145 // текущая запись не является ссылкой и не имеет информации о сервисе
146 // она должна сожержать информацию об активации
147 if (se.activator != null) {
148 se.service = se.activator();
149
150 m_services[serviceType] = se;
151
152 service = se.service;
153 return true;
154 }
155
156 service = null;
157 return false;
158 }
159
160 /// <summary>
161 /// Регистрирует фабрику для активации сервиса по первому требованию.
162 /// </summary>
163 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
164 /// <param name="activator">Фабрика для создания/получения объекта, предоставляющего сервис.</param>
165 /// <param name = "cleanup">Метод для освобождения экземпляра сервиса, будет вызыван при освобождении сервис-локатора.</param>
166 /// <remarks>При освобождении сервис-локатора, сервисы полученные в результате активации также будут освобождены.</remarks>
167 public void Register<T>(Func<T> activator, Action<T> cleanup) {
168 Safe.ArgumentNotNull(activator, "activator");
169
170 AssertNotDisposed();
171
172 Unregister(typeof(T));
173
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;
179 }
180
181 public void Register<T>(Func<T> activator) {
182 Register(activator, null);
183 }
184
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>
197 /// Регистрирует объект, предоставляющий сервис. Повторная регистрация отменяет уже существующую.
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) {
203 Safe.ArgumentNotNull(service, "service");
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) {
213 Safe.ArgumentNotNull(serviceType, "serviceType");
214
215 AssertNotDisposed();
216
217 ServiceEntry se;
218 if (m_services.TryGetValue(serviceType, out se)) {
219 if (se.origin != null) {
220 var pe = m_services[se.origin];
221 pe.associated.Remove(serviceType);
222 }
223 // освобождаем ресурсы
224 se.Dispose();
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)
242 entry.Dispose();
243
244 }
245 base.Dispose(disposing);
246 }
247 }
248 }