implab/ImplabNet Commit - r86:b33832ab0262

ServiceLocator: added a cleanup callback to the service registration method

cin -

r86:b33832ab0262 v2

parent child

Context file:

r86:b33832ab0262

Collapse all files

Implab/ServiceLocator.cs +32 -17

              using System;
              using System.Collections.Generic;
-             using System.Linq;
-             using System.Web;
-             using System.Diagnostics;
              namespace Implab {
                  /// <summary>
                  /// Коллекция сервисов, позволяет регистрировать и получать сервисы.
                  /// </summary>
                  public class ServiceLocator: Component, IServiceLocator, IServiceProvider {
-                     // запись об сервисе
-                     struct ServiceEntry {
+                     // запись о сервисе
+                     struct ServiceEntry : IDisposable {
                          public object service; // сервис
                          public bool shared; // признак того, что сервис НЕ нужно освобождать
                          public Func<object> activator; // активатор сервиса при первом обращении
+                         public Action<object> cleanup; // функция для очистки сервиса
                          public List<Type> associated; // ссылки на текущую запись
                          public Type origin; // ссылка на оригинальную запись о сервисе
+                         #region IDisposable implementation
+                         public void Dispose() {
+                             if (shared)
+                                 return;
+                             if (cleanup != null)
+                                 cleanup(service);
+                             else {
+                                 var d = service as IDisposable;
+                                 if (d != null)
+                                     d.Dispose();
+                             }
+                         }
+                         #endregion
                      }
                      // словарь существующих сервисов
              		/// <param name="serviceType">Тип запрашиваемого сервиса.</param>
              		/// <param name="service">Искомый сервис.</param>
              		public virtual bool TryGetService(Type serviceType, out object service) {
-                         if (serviceType == null)
-                             throw new ArgumentNullException("serviceType");
+                         Safe.ArgumentNotNull(serviceType, "serviceType");
                          AssertNotDisposed();
                          ServiceEntry se;
                      /// </summary>
                      /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
                      /// <param name="activator">Фабрика для создания/получения объекта, предоставляющего сервис.</param>
+                     /// <param name = "cleanup">Метод для освобождения экземпляра сервиса, будет вызыван при освобождении сервис-локатора.</param>
                      /// <exception cref="InvalidOperationException">Указанный сервис уже зарегистрирован.</exception>
                      /// <remarks>При освобождении сервис-локатора, сервисы полученные в результате активации также будут освобождены.</remarks>
-                     public void Register<T>(Func<T> activator) {
-                         if (activator == null)
-                             throw new ArgumentNullException("activator");
+                     public void Register<T>(Func<T> activator, Action<T> cleanup) {
+                         Safe.ArgumentNotNull(activator, "activator");
                          AssertNotDisposed();
                          Unregister(typeof(T));
                          m_services[typeof(T)] = new ServiceEntry {
-                             activator = () => activator() as object
+                             activator = () => activator(),
+                             cleanup = instance => cleanup((T)instance)
                          };
                      }
+                     public void Register<T>(Func<T> activator) {
+                         Register(activator, null);
+                     }
                      /// <summary>
                      /// Регистрирует объект, предоставляющий сервис.
                      /// </summary>
                      /// <param name="shared">Признак того, что объект является разделяемым и сервис-локатор не должен его освобождать.</param>
                      /// <exception cref="InvalidOperationException">Указанный сервис уже зарегистрирован.</exception>
                      public void Register<T>(T service, bool shared) {
-                         if (service == null)
-                             throw new ArgumentNullException("service");
+                         Safe.ArgumentNotNull(service, "service");
                          AssertNotDisposed();
                      }
                      public void Unregister(Type serviceType) {
-                         if (serviceType == null)
-                             throw new ArgumentNullException("serviceType");
+                         Safe.ArgumentNotNull(serviceType, "serviceType");
                          AssertNotDisposed();
                          if (disposing) {
                              foreach (var entry in m_services.Values)
-                                 if (!entry.shared && entry.service is IDisposable)
-                                     ((IDisposable)entry.service).Dispose();
+                                 entry.Dispose();
                          }
                          base.Dispose(disposing);

General Comments 0

Write
Preview

You need to be logged in to leave comments. Login now

No TODOs yet

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages