|
152
|
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 } |
