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