##// END OF EJS Templates
pool refactoring
cin -
r117:8beee0d11de6 v2
parent child
Show More
@@ -0,0 +1,90
1 using System;
2 using Implab.Parallels;
3 using System.Threading;
4 using System.Diagnostics;
5 using System.Diagnostics.CodeAnalysis;
6
7 namespace Implab {
8 public abstract class DisposablePool<T> : IDisposable {
9 readonly int m_size;
10 readonly MTQueue<T> m_queue = new MTQueue<T>();
11
12 [SuppressMessage("Microsoft.Design", "CA1000:DoNotDeclareStaticMembersOnGenericTypes")]
13 static readonly bool _isValueType = typeof(T).IsValueType;
14
15 bool m_disposed;
16
17 int m_count;
18
19 protected DisposablePool(int size) {
20 m_size = size;
21 }
22
23 protected DisposablePool() : this(Environment.ProcessorCount+1) {
24 }
25
26 public T Allocate() {
27 if (m_disposed)
28 throw new ObjectDisposedException(ToString());
29
30 T instance;
31 if (m_queue.TryDequeue(out instance)) {
32 Interlocked.Decrement(ref m_count);
33 } else {
34 instance = CreateInstance();
35 Debug.Assert(!Object.Equals(instance, default(T)) || _isValueType);
36 }
37 return instance;
38 }
39
40 protected abstract T CreateInstance();
41
42 protected virtual void CleanupInstance(T instance) {
43 }
44
45 public void Release(T instance) {
46 if ( Object.Equals(instance,default(T)) && !_isValueType)
47 return;
48
49 Thread.MemoryBarrier();
50 if (m_count < m_size && !m_disposed) {
51 Interlocked.Increment(ref m_count);
52
53 CleanupInstance(instance);
54
55 m_queue.Enqueue(instance);
56
57 // пока элемент возвращался в кеш, была начата операция освобождения всего кеша
58 // и возможно уже законцена, в таком случае следует извлечь элемент обратно и
59 // освободить его. Если операция освобождения кеша еще не заврешилась, то будет
60 // изъят и освобожден произвольный элемен, что не повлияет на ход всего процесса.
61 if (m_disposed && m_queue.TryDequeue(out instance) && instance is IDisposable)
62 ((IDisposable)instance).Dispose() ;
63
64 } else {
65 if (instance is IDisposable)
66 ((IDisposable)instance).Dispose();
67 }
68 }
69
70 protected virtual void Dispose(bool disposing) {
71 if (disposing) {
72 m_disposed = true;
73 T instance;
74 while (m_queue.TryDequeue(out instance))
75 if (instance is IDisposable)
76 ((IDisposable)instance).Dispose();
77 }
78 }
79
80 #region IDisposable implementation
81
82 public void Dispose() {
83 Dispose(true);
84 GC.SuppressFinalize(this);
85 }
86
87 #endregion
88 }
89 }
90
@@ -1,225 +1,225
1 1 <?xml version="1.0" encoding="utf-8"?>
2 2 <Project DefaultTargets="Build" ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
3 3 <PropertyGroup>
4 4 <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
5 5 <Platform Condition=" '$(Platform)' == '' ">AnyCPU</Platform>
6 6 <ProjectGuid>{F550F1F8-8746-4AD0-9614-855F4C4B7F05}</ProjectGuid>
7 7 <OutputType>Library</OutputType>
8 8 <RootNamespace>Implab</RootNamespace>
9 9 <AssemblyName>Implab</AssemblyName>
10 10 </PropertyGroup>
11 11 <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
12 12 <DebugSymbols>true</DebugSymbols>
13 13 <DebugType>full</DebugType>
14 14 <Optimize>false</Optimize>
15 15 <OutputPath>bin\Debug</OutputPath>
16 16 <DefineConstants>TRACE;DEBUG;</DefineConstants>
17 17 <ErrorReport>prompt</ErrorReport>
18 18 <WarningLevel>4</WarningLevel>
19 19 <ConsolePause>false</ConsolePause>
20 20 <RunCodeAnalysis>true</RunCodeAnalysis>
21 21 </PropertyGroup>
22 22 <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
23 23 <DebugType>full</DebugType>
24 24 <Optimize>true</Optimize>
25 25 <OutputPath>bin\Release</OutputPath>
26 26 <ErrorReport>prompt</ErrorReport>
27 27 <WarningLevel>4</WarningLevel>
28 28 <ConsolePause>false</ConsolePause>
29 29 </PropertyGroup>
30 30 <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug 4.5|AnyCPU' ">
31 31 <DebugSymbols>true</DebugSymbols>
32 32 <DebugType>full</DebugType>
33 33 <Optimize>false</Optimize>
34 34 <OutputPath>bin\Debug</OutputPath>
35 35 <DefineConstants>TRACE;DEBUG;NET_4_5</DefineConstants>
36 36 <ErrorReport>prompt</ErrorReport>
37 37 <WarningLevel>4</WarningLevel>
38 38 <RunCodeAnalysis>true</RunCodeAnalysis>
39 39 <ConsolePause>false</ConsolePause>
40 40 </PropertyGroup>
41 41 <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release 4.5|AnyCPU' ">
42 42 <Optimize>true</Optimize>
43 43 <OutputPath>bin\Release</OutputPath>
44 44 <ErrorReport>prompt</ErrorReport>
45 45 <WarningLevel>4</WarningLevel>
46 46 <ConsolePause>false</ConsolePause>
47 47 <DefineConstants>NET_4_5</DefineConstants>
48 48 </PropertyGroup>
49 49 <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'DebugMono|AnyCPU' ">
50 50 <DebugSymbols>true</DebugSymbols>
51 51 <DebugType>full</DebugType>
52 52 <Optimize>false</Optimize>
53 53 <OutputPath>bin\Debug</OutputPath>
54 54 <DefineConstants>TRACE;DEBUG;NET_4_5;MONO</DefineConstants>
55 55 <ErrorReport>prompt</ErrorReport>
56 56 <WarningLevel>4</WarningLevel>
57 57 <RunCodeAnalysis>true</RunCodeAnalysis>
58 58 <ConsolePause>false</ConsolePause>
59 59 </PropertyGroup>
60 60 <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'ReleaseMono|AnyCPU' ">
61 61 <Optimize>true</Optimize>
62 62 <OutputPath>bin\Release</OutputPath>
63 63 <DefineConstants>NET_4_5;MONO;</DefineConstants>
64 64 <ErrorReport>prompt</ErrorReport>
65 65 <WarningLevel>4</WarningLevel>
66 66 <ConsolePause>false</ConsolePause>
67 67 </PropertyGroup>
68 68 <ItemGroup>
69 69 <Reference Include="System" />
70 70 <Reference Include="System.Xml" />
71 71 </ItemGroup>
72 72 <ItemGroup>
73 73 <Compile Include="Component.cs" />
74 74 <Compile Include="CustomEqualityComparer.cs" />
75 75 <Compile Include="Diagnostics\ConsoleTraceListener.cs" />
76 76 <Compile Include="Diagnostics\EventText.cs" />
77 77 <Compile Include="Diagnostics\IEventTextFormatter.cs" />
78 78 <Compile Include="Diagnostics\LogChannel.cs" />
79 79 <Compile Include="Diagnostics\LogicalOperation.cs" />
80 80 <Compile Include="Diagnostics\TextFileListener.cs" />
81 81 <Compile Include="Diagnostics\TextListenerBase.cs" />
82 82 <Compile Include="Diagnostics\TraceLog.cs" />
83 83 <Compile Include="Diagnostics\TraceEvent.cs" />
84 84 <Compile Include="Diagnostics\TraceEventType.cs" />
85 85 <Compile Include="Disposable.cs" />
86 86 <Compile Include="ICancellable.cs" />
87 87 <Compile Include="IProgressHandler.cs" />
88 88 <Compile Include="IProgressNotifier.cs" />
89 89 <Compile Include="IPromiseT.cs" />
90 90 <Compile Include="IPromise.cs" />
91 91 <Compile Include="IServiceLocator.cs" />
92 92 <Compile Include="ITaskController.cs" />
93 93 <Compile Include="JSON\JSONElementContext.cs" />
94 94 <Compile Include="JSON\JSONElementType.cs" />
95 95 <Compile Include="JSON\JSONGrammar.cs" />
96 96 <Compile Include="JSON\JSONParser.cs" />
97 97 <Compile Include="JSON\JSONScanner.cs" />
98 98 <Compile Include="JSON\JsonTokenType.cs" />
99 99 <Compile Include="JSON\JSONWriter.cs" />
100 100 <Compile Include="JSON\JSONXmlReader.cs" />
101 101 <Compile Include="JSON\JSONXmlReaderOptions.cs" />
102 102 <Compile Include="JSON\StringTranslator.cs" />
103 103 <Compile Include="Parallels\DispatchPool.cs" />
104 104 <Compile Include="Parallels\ArrayTraits.cs" />
105 105 <Compile Include="Parallels\MTQueue.cs" />
106 106 <Compile Include="Parallels\WorkerPool.cs" />
107 107 <Compile Include="Parsing\Alphabet.cs" />
108 108 <Compile Include="Parsing\AlphabetBase.cs" />
109 109 <Compile Include="Parsing\AltToken.cs" />
110 110 <Compile Include="Parsing\BinaryToken.cs" />
111 111 <Compile Include="Parsing\CatToken.cs" />
112 112 <Compile Include="Parsing\CDFADefinition.cs" />
113 113 <Compile Include="Parsing\DFABuilder.cs" />
114 114 <Compile Include="Parsing\DFADefinitionBase.cs" />
115 115 <Compile Include="Parsing\DFAStateDescriptor.cs" />
116 116 <Compile Include="Parsing\DFAutomaton.cs" />
117 117 <Compile Include="Parsing\EDFADefinition.cs" />
118 118 <Compile Include="Parsing\EmptyToken.cs" />
119 119 <Compile Include="Parsing\EndToken.cs" />
120 120 <Compile Include="Parsing\EnumAlphabet.cs" />
121 121 <Compile Include="Parsing\Grammar.cs" />
122 122 <Compile Include="Parsing\IAlphabet.cs" />
123 123 <Compile Include="Parsing\IDFADefinition.cs" />
124 124 <Compile Include="Parsing\IVisitor.cs" />
125 125 <Compile Include="Parsing\ParserException.cs" />
126 126 <Compile Include="Parsing\Scanner.cs" />
127 127 <Compile Include="Parsing\StarToken.cs" />
128 128 <Compile Include="Parsing\SymbolToken.cs" />
129 129 <Compile Include="Parsing\Token.cs" />
130 <Compile Include="SafePool.cs" />
131 130 <Compile Include="ServiceLocator.cs" />
132 131 <Compile Include="TaskController.cs" />
133 132 <Compile Include="ProgressInitEventArgs.cs" />
134 133 <Compile Include="Properties\AssemblyInfo.cs" />
135 134 <Compile Include="Promise.cs" />
136 135 <Compile Include="Parallels\AsyncPool.cs" />
137 136 <Compile Include="Safe.cs" />
138 137 <Compile Include="ValueEventArgs.cs" />
139 138 <Compile Include="PromiseExtensions.cs" />
140 139 <Compile Include="TransientPromiseException.cs" />
141 140 <Compile Include="SyncContextPromise.cs" />
142 <Compile Include="ObjectPool.cs" />
143 141 <Compile Include="Diagnostics\OperationContext.cs" />
144 142 <Compile Include="Diagnostics\TraceContext.cs" />
145 143 <Compile Include="Diagnostics\LogEventArgs.cs" />
146 144 <Compile Include="Diagnostics\LogEventArgsT.cs" />
147 145 <Compile Include="Diagnostics\Extensions.cs" />
148 146 <Compile Include="IComponentContainer.cs" />
149 147 <Compile Include="PromiseEventType.cs" />
150 148 <Compile Include="Parallels\MTCustomQueue.cs" />
151 149 <Compile Include="Parallels\MTCustomQueueNode.cs" />
152 150 <Compile Include="ComponentContainer.cs" />
151 <Compile Include="DisposablePool.cs" />
152 <Compile Include="ObjectPool.cs" />
153 153 </ItemGroup>
154 154 <Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
155 155 <ItemGroup />
156 156 <ProjectExtensions>
157 157 <MonoDevelop>
158 158 <Properties>
159 159 <Policies>
160 160 <CSharpFormattingPolicy IndentSwitchBody="True" NamespaceBraceStyle="EndOfLine" ClassBraceStyle="EndOfLine" InterfaceBraceStyle="EndOfLine" StructBraceStyle="EndOfLine" EnumBraceStyle="EndOfLine" MethodBraceStyle="EndOfLine" ConstructorBraceStyle="EndOfLine" DestructorBraceStyle="EndOfLine" BeforeMethodDeclarationParentheses="False" BeforeMethodCallParentheses="False" BeforeConstructorDeclarationParentheses="False" NewLineBeforeConstructorInitializerColon="NewLine" NewLineAfterConstructorInitializerColon="SameLine" BeforeIndexerDeclarationBracket="False" BeforeDelegateDeclarationParentheses="False" NewParentheses="False" SpacesBeforeBrackets="False" inheritsSet="Mono" inheritsScope="text/x-csharp" scope="text/x-csharp" />
161 161 <TextStylePolicy FileWidth="120" EolMarker="Unix" inheritsSet="VisualStudio" inheritsScope="text/plain" scope="text/x-csharp" />
162 162 <DotNetNamingPolicy DirectoryNamespaceAssociation="PrefixedHierarchical" ResourceNamePolicy="MSBuild" />
163 163 <TextStylePolicy FileWidth="120" TabsToSpaces="False" inheritsSet="VisualStudio" inheritsScope="text/plain" scope="application/xml" />
164 164 <XmlFormattingPolicy inheritsSet="Mono" inheritsScope="application/xml" scope="application/xml" />
165 165 <TextStylePolicy FileWidth="120" TabsToSpaces="False" inheritsSet="VisualStudio" inheritsScope="text/plain" scope="text/plain" />
166 166 <NameConventionPolicy>
167 167 <Rules>
168 168 <NamingRule Name="Namespaces" AffectedEntity="Namespace" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
169 169 <NamingRule Name="Types" AffectedEntity="Class, Struct, Enum, Delegate" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
170 170 <NamingRule Name="Interfaces" AffectedEntity="Interface" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True">
171 171 <RequiredPrefixes>
172 172 <String>I</String>
173 173 </RequiredPrefixes>
174 174 </NamingRule>
175 175 <NamingRule Name="Attributes" AffectedEntity="CustomAttributes" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True">
176 176 <RequiredSuffixes>
177 177 <String>Attribute</String>
178 178 </RequiredSuffixes>
179 179 </NamingRule>
180 180 <NamingRule Name="Event Arguments" AffectedEntity="CustomEventArgs" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True">
181 181 <RequiredSuffixes>
182 182 <String>EventArgs</String>
183 183 </RequiredSuffixes>
184 184 </NamingRule>
185 185 <NamingRule Name="Exceptions" AffectedEntity="CustomExceptions" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True">
186 186 <RequiredSuffixes>
187 187 <String>Exception</String>
188 188 </RequiredSuffixes>
189 189 </NamingRule>
190 190 <NamingRule Name="Methods" AffectedEntity="Methods" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
191 191 <NamingRule Name="Static Readonly Fields" AffectedEntity="ReadonlyField" VisibilityMask="Internal, Protected, Public" NamingStyle="CamelCase" IncludeInstanceMembers="False" IncludeStaticEntities="True" />
192 192 <NamingRule Name="Fields (Non Private)" AffectedEntity="Field" VisibilityMask="Internal, Public" NamingStyle="CamelCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
193 193 <NamingRule Name="ReadOnly Fields (Non Private)" AffectedEntity="ReadonlyField" VisibilityMask="Internal, Public" NamingStyle="CamelCase" IncludeInstanceMembers="True" IncludeStaticEntities="False" />
194 194 <NamingRule Name="Fields (Private)" AffectedEntity="Field, ReadonlyField" VisibilityMask="Private, Protected" NamingStyle="CamelCase" IncludeInstanceMembers="True" IncludeStaticEntities="False">
195 195 <RequiredPrefixes>
196 196 <String>m_</String>
197 197 </RequiredPrefixes>
198 198 </NamingRule>
199 199 <NamingRule Name="Static Fields (Private)" AffectedEntity="Field" VisibilityMask="Private" NamingStyle="CamelCase" IncludeInstanceMembers="False" IncludeStaticEntities="True">
200 200 <RequiredPrefixes>
201 201 <String>_</String>
202 202 </RequiredPrefixes>
203 203 </NamingRule>
204 204 <NamingRule Name="ReadOnly Fields (Private)" AffectedEntity="ReadonlyField" VisibilityMask="Private, Protected" NamingStyle="CamelCase" IncludeInstanceMembers="True" IncludeStaticEntities="False">
205 205 <RequiredPrefixes>
206 206 <String>m_</String>
207 207 </RequiredPrefixes>
208 208 </NamingRule>
209 209 <NamingRule Name="Constant Fields" AffectedEntity="ConstantField" VisibilityMask="VisibilityMask" NamingStyle="AllUpper" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
210 210 <NamingRule Name="Properties" AffectedEntity="Property" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
211 211 <NamingRule Name="Events" AffectedEntity="Event" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
212 212 <NamingRule Name="Enum Members" AffectedEntity="EnumMember" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
213 213 <NamingRule Name="Parameters" AffectedEntity="Parameter, LocalVariable" VisibilityMask="VisibilityMask" NamingStyle="CamelCase" IncludeInstanceMembers="True" IncludeStaticEntities="True" />
214 214 <NamingRule Name="Type Parameters" AffectedEntity="TypeParameter" VisibilityMask="VisibilityMask" NamingStyle="PascalCase" IncludeInstanceMembers="True" IncludeStaticEntities="True">
215 215 <RequiredPrefixes>
216 216 <String>T</String>
217 217 </RequiredPrefixes>
218 218 </NamingRule>
219 219 </Rules>
220 220 </NameConventionPolicy>
221 221 </Policies>
222 222 </Properties>
223 223 </MonoDevelop>
224 224 </ProjectExtensions>
225 225 </Project> No newline at end of file
@@ -1,90 +1,60
1 using System;
2 using Implab.Parallels;
3 using System.Threading;
4 using System.Diagnostics;
5 using System.Diagnostics.CodeAnalysis;
6
7 namespace Implab {
8 public abstract class ObjectPool<T> : IDisposable {
9 readonly int m_size;
10 readonly MTQueue<T> m_queue = new MTQueue<T>();
11
12 [SuppressMessage("Microsoft.Design", "CA1000:DoNotDeclareStaticMembersOnGenericTypes")]
13 static readonly bool _isValueType = typeof(T).IsValueType;
14
15 bool m_disposed;
16
17 int m_count;
18
19 protected ObjectPool(int size) {
20 m_size = size;
21 }
22
23 protected ObjectPool() : this(Environment.ProcessorCount+1) {
24 }
25
26 public T Allocate() {
27 if (m_disposed)
28 throw new ObjectDisposedException(ToString());
29
30 T instance;
31 if (m_queue.TryDequeue(out instance)) {
32 Interlocked.Decrement(ref m_count);
33 } else {
34 instance = CreateInstance();
35 Debug.Assert(!Object.Equals(instance, default(T)) || _isValueType);
36 }
37 return instance;
38 }
39
40 protected abstract T CreateInstance();
41
42 protected virtual void CleanupInstance(T instance) {
43 }
44
45 public void Release(T instance) {
46 if ( Object.Equals(instance,default(T)) && !_isValueType)
47 return;
48
49 Thread.MemoryBarrier();
50 if (m_count < m_size && !m_disposed) {
51 Interlocked.Increment(ref m_count);
52
53 CleanupInstance(instance);
54
55 m_queue.Enqueue(instance);
56
57 // пока элемент возвращался в кеш, была начата операция освобождения всего кеша
58 // и возможно уже законцена, в таком случае следует извлечь элемент обратно и
59 // освободить его. Если операция освобождения кеша еще не заврешилась, то будет
60 // изъят и освобожден произвольный элемен, что не повлияет на ход всего процесса.
61 if (m_disposed && m_queue.TryDequeue(out instance) && instance is IDisposable)
62 ((IDisposable)instance).Dispose() ;
63
64 } else {
65 if (instance is IDisposable)
66 ((IDisposable)instance).Dispose();
67 }
68 }
69
70 protected virtual void Dispose(bool disposing) {
71 if (disposing) {
72 m_disposed = true;
73 T instance;
74 while (m_queue.TryDequeue(out instance))
75 if (instance is IDisposable)
76 ((IDisposable)instance).Dispose();
77 }
78 }
79
80 #region IDisposable implementation
81
82 public void Dispose() {
83 Dispose(true);
84 GC.SuppressFinalize(this);
85 }
86
87 #endregion
88 }
89 }
90
1 using Implab.Parallels;
2 using System;
3 using System.Threading;
4
5 namespace Implab {
6 /// <summary>
7 /// Базовый класс для создания пулов объектов.
8 /// </summary>
9 /// <remarks>
10 /// <para>Пул объектов позволяет многократно использовать один и тотже объект,
11 /// что актуально для объектов, создание которых требует существенных ресурсов.
12 /// Пул объектов использует слабые ссылки, чтобы не препятствовать освобождению
13 /// ресурсов и создает новые объекты при необходимости.</para>
14 /// <para>
15 /// Наследники должны реализовывать метод <see cref="CreateInstance()"/> для создания
16 /// новых экземпляров.
17 /// </para>
18 /// <para>Пул поддерживает обращения сразу из нескольких потоков.</para>
19 /// </remarks>
20 public abstract class ObjectPool<T> where T : class {
21 readonly MTQueue<WeakReference> m_queue = new MTQueue<WeakReference>();
22 readonly int m_size;
23 int m_count = 0;
24
25 protected ObjectPool() : this(Environment.ProcessorCount+1) {
26
27 }
28
29 protected ObjectPool(int size) {
30 Safe.ArgumentInRange(size,1,size,"size");
31
32 m_size = size;
33 }
34
35 protected abstract T CreateInstance();
36
37 protected virtual void CleanupInstance(T instance) {
38 }
39
40 public T Allocate() {
41 WeakReference reference;
42 while (m_queue.TryDequeue(out reference)) {
43 Interlocked.Decrement(ref m_count);
44 object instance = reference.Target;
45 if (instance == null)
46 continue;
47 return (T)instance;
48 }
49 return CreateInstance();
50 }
51
52 public void Release(T instance) {
53 if (m_count < m_size && instance != null) {
54 Interlocked.Increment(ref m_count);
55 CleanupInstance(instance);
56 m_queue.Enqueue(new WeakReference(instance));
57 }
58 }
59 }
60 }
@@ -1,248 +1,248
1 1 using System;
2 2 using System.Collections.Generic;
3 3
4 4 namespace Implab {
5 5 /// <summary>
6 6 /// Коллекция сервисов, позволяет регистрировать и получать сервисы.
7 7 /// </summary>
8 public class ServiceLocator: Component, IServiceLocator, IServiceProvider {
8 public class ServiceLocator: Disposable, IServiceLocator, IServiceProvider {
9 9 // запись о сервисе
10 10 struct ServiceEntry : IDisposable {
11 11 public object service; // сервис
12 12 public bool shared; // признак того, что сервис НЕ нужно освобождать
13 13 public Func<object> activator; // активатор сервиса при первом обращении
14 14 public Action<object> cleanup; // функция для очистки сервиса
15 15 public List<Type> associated; // ссылки на текущую запись
16 16 public Type origin; // ссылка на оригинальную запись о сервисе
17 17
18 18 #region IDisposable implementation
19 19
20 20 public void Dispose() {
21 21 if (shared)
22 22 return;
23 23 if (cleanup != null) {
24 24 if (service != null)
25 25 cleanup(service);
26 26 } else
27 27 Safe.Dispose(service);
28 28 }
29 29
30 30 #endregion
31 31 }
32 32
33 33 // словарь существующих сервисов
34 34 readonly Dictionary<Type, ServiceEntry> m_services = new Dictionary<Type,ServiceEntry>();
35 35
36 36 /// <summary>
37 37 /// Получает объект предоставляющий сервис <typeparamref name="T"/>.
38 38 /// </summary>
39 39 /// <typeparam name="T">Тип запрашиваемого сервиса</typeparam>
40 40 /// <returns>Объект, реализующий сервис</returns>
41 41 /// <exception cref="KeyNotFoundException">Сервис не зарегистрирован</exception>
42 42 public T GetService<T>() {
43 43 object result;
44 44 if (TryGetService(typeof(T), out result))
45 45 return (T)result;
46 46 throw new ApplicationException (String.Format ("{0} doesn't provide {1} service", this, typeof(T)));
47 47 }
48 48
49 49
50 50 /// <summary>
51 51 /// Пытается получить указанный сервис, в случае, если компонента не предоставляет требуемый сервис
52 52 /// не возникает исключений.
53 53 /// </summary>
54 54 /// <typeparam name="T">Тип требуемого сервиса.</typeparam>
55 55 /// <param name="service">Объект реализующий сервис, или <c>default(T)</c> если такового нет.</param>
56 56 /// <returns><c>true</c> - сервис найден, <c>false</c> - сервис не зарегистрирован.</returns>
57 57 public bool TryGetService<T>(out T service) {
58 58 object result;
59 59 if (TryGetService(typeof(T), out result)) {
60 60 service = (T)result;
61 61 return true;
62 62 }
63 63 service = default(T);
64 64 return false;
65 65 }
66 66
67 67 /// <summary>
68 68 /// Получает объект предоставляющий сервис <paramref name="serviceType"/>
69 69 /// </summary>
70 70 /// <param name="serviceType">Тип запрашиваемого сервиса</param>
71 71 /// <returns>Объект, реализующий сервис</returns>
72 72 /// <exception cref="KeyNotFoundException">Сервис не зарегистрирован</exception>
73 73 public object GetService(Type serviceType) {
74 74 object result;
75 75 if (TryGetService(serviceType, out result))
76 76 return result;
77 77 throw new ApplicationException (String.Format ("{0} doesn't provide {1} service", this, serviceType));
78 78 }
79 79
80 80 /// <summary>
81 81 /// Пытается получить требуемый сервис или совместимый с ним.
82 82 /// </summary>
83 83 /// <returns><c>true</c>, если сервис был найден, <c>false</c> в противном случае..</returns>
84 84 /// <param name="serviceType">Тип запрашиваемого сервиса.</param>
85 85 /// <param name="service">Искомый сервис.</param>
86 86 public virtual bool TryGetService(Type serviceType, out object service) {
87 87 Safe.ArgumentNotNull(serviceType, "serviceType");
88 88 AssertNotDisposed();
89 89
90 90 ServiceEntry se;
91 91 if (!m_services.TryGetValue(serviceType, out se)) {
92 92 // ищем ближайщий объект, реализующий нужный сервис
93 93 Type pt = null;
94 94 foreach (var t in m_services.Keys)
95 95 if (serviceType.IsAssignableFrom(t) && (pt == null || t.IsAssignableFrom(pt)))
96 96 pt = t;
97 97
98 98 if (pt == null) {
99 99 // нет нужного сервиса
100 100 service = null;
101 101 return false;
102 102 }
103 103
104 104 var pe = m_services[pt];
105 105
106 106 // найденная запись может ссылаться на оригинальную запись с сервисом
107 107 if(pe.origin != null) {
108 108 pt = pe.origin;
109 109 pe = m_services[pt];
110 110 }
111 111
112 112 // добавляем список с обратными ссылками
113 113 if (pe.associated == null)
114 114 pe.associated = new List<Type>();
115 115
116 116 pe.associated.Add(serviceType);
117 117
118 118 // обновляем родительскую запись
119 119 m_services[pt] = pe;
120 120
121 121 // создаем запись со ссылкой
122 122 se = new ServiceEntry {
123 123 service = pe.service,
124 124 origin = pt,
125 125 shared = true // предотвращаем множественные попытки освобождения
126 126 };
127 127
128 128 m_services[serviceType] = se;
129 129 }
130 130
131 131 // запись содержит в себе информацию о сервисе
132 132 if (se.service != null) {
133 133 service = se.service;
134 134 return true;
135 135 }
136 136
137 137 // текущая запись является ссылкой
138 138 if (se.origin != null) {
139 139 se.service = GetService(se.origin);
140 140 m_services[serviceType] = se;
141 141 service = se.service;
142 142 return true;
143 143 }
144 144
145 145 // текущая запись не является ссылкой и не имеет информации о сервисе
146 146 // она должна сожержать информацию об активации
147 147 if (se.activator != null) {
148 148 se.service = se.activator();
149 149
150 150 m_services[serviceType] = se;
151 151
152 152 service = se.service;
153 153 return true;
154 154 }
155 155
156 156 service = null;
157 157 return false;
158 158 }
159 159
160 160 /// <summary>
161 161 /// Регистрирует фабрику для активации сервиса по первому требованию.
162 162 /// </summary>
163 163 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
164 164 /// <param name="activator">Фабрика для создания/получения объекта, предоставляющего сервис.</param>
165 165 /// <param name = "cleanup">Метод для освобождения экземпляра сервиса, будет вызыван при освобождении сервис-локатора.</param>
166 166 /// <remarks>При освобождении сервис-локатора, сервисы полученные в результате активации также будут освобождены.</remarks>
167 167 public void Register<T>(Func<T> activator, Action<T> cleanup) {
168 168 Safe.ArgumentNotNull(activator, "activator");
169 169
170 170 AssertNotDisposed();
171 171
172 172 Unregister(typeof(T));
173 173
174 174 var serviceEntry = new ServiceEntry();
175 175 serviceEntry.activator = () => activator();
176 176 if (cleanup != null)
177 177 serviceEntry.cleanup = instance => cleanup((T)instance);
178 178 m_services[typeof(T)] = serviceEntry;
179 179 }
180 180
181 181 public void Register<T>(Func<T> activator) {
182 182 Register(activator, null);
183 183 }
184 184
185 185 /// <summary>
186 186 /// Регистрирует объект, предоставляющий сервис.
187 187 /// </summary>
188 188 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
189 189 /// <param name="service">Объект, предоставляющий сервис.</param>
190 190 /// <exception cref="InvalidOperationException">Указанный сервис уже зарегистрирован.</exception>
191 191 /// <remarks>Сервис-локатором не управляет временем жизни объекта для зарегистрированного сервиса.</remarks>
192 192 public void Register<T>(T service) {
193 193 Register(service, true);
194 194 }
195 195
196 196 /// <summary>
197 197 /// Регистрирует объект, предоставляющий сервис. Повторная регистрация отменяет уже существующую.
198 198 /// </summary>
199 199 /// <typeparam name="T">Тип регистрируемого сервиса.</typeparam>
200 200 /// <param name="service">Объект, предоставляющий сервис.</param>
201 201 /// <param name="shared">Признак того, что объект является разделяемым и сервис-локатор не должен его освобождать.</param>
202 202 public void Register<T>(T service, bool shared) {
203 203 Safe.ArgumentNotNull(service, "service");
204 204
205 205 AssertNotDisposed();
206 206
207 207 Unregister(typeof(T));
208 208
209 209 m_services[typeof(T)] = new ServiceEntry { service = service, shared = shared };
210 210 }
211 211
212 212 public void Unregister(Type serviceType) {
213 213 Safe.ArgumentNotNull(serviceType, "serviceType");
214 214
215 215 AssertNotDisposed();
216 216
217 217 ServiceEntry se;
218 218 if (m_services.TryGetValue(serviceType, out se)) {
219 219 if (se.origin != null) {
220 220 var pe = m_services[se.origin];
221 221 pe.associated.Remove(serviceType);
222 222 }
223 223 // освобождаем ресурсы
224 224 se.Dispose();
225 225 m_services.Remove(serviceType);
226 226
227 227 // убираем связанные записи
228 228 if (se.associated != null)
229 229 foreach (var item in se.associated)
230 230 m_services.Remove(item);
231 231 }
232 232 }
233 233
234 234 /// <summary>
235 235 /// Освобождает зарегистрированные сервисы (которые требуется освобоить).
236 236 /// </summary>
237 237 /// <param name="disposing">Призанак того, что нужно освободить ресурсы.</param>
238 238 protected override void Dispose(bool disposing) {
239 239 if (disposing) {
240 240
241 241 foreach (var entry in m_services.Values)
242 242 entry.Dispose();
243 243
244 244 }
245 245 base.Dispose(disposing);
246 246 }
247 247 }
248 248 } No newline at end of file
1 NO CONTENT: file was removed
General Comments 0
You need to be logged in to leave comments. Login now