|
1 | | -using System.Collections.Concurrent; |
| 1 | +using System.Buffers; |
| 2 | +using System.Collections.Concurrent; |
2 | 3 | using System.Diagnostics.CodeAnalysis; |
3 | 4 | using System.Reflection; |
4 | 5 |
|
@@ -94,49 +95,61 @@ public bool TryTransform<TSource, TTarget>(TSource source, IGremlinQueryEnvironm |
94 | 95 | private IConverter<TStaticSource, TTarget> GetUnifiedConverter<TStaticSource, TActualSource, TTarget>(IGremlinQueryEnvironment environment) |
95 | 96 | where TActualSource : TStaticSource |
96 | 97 | { |
| 98 | + var tupleCount = 0; |
97 | 99 | var deferTransformers = _deferTransformers; |
98 | | - var list = new List<(IConverter<TActualSource, TTarget> converter, TransformerImpl overridden)>(); |
| 100 | + var arrayPool = ArrayPool<(IConverter<TActualSource, TTarget> converter, TransformerImpl overridden)>.Shared; |
99 | 101 |
|
100 | | - for (var i = _converterFactories.Count - 1; i >= 0; i--) |
| 102 | + var tuples = arrayPool |
| 103 | + .Rent(_converterFactories.Count); |
| 104 | + |
| 105 | + try |
101 | 106 | { |
102 | | - if (_converterFactories[i].TryCreate<TActualSource, TTarget>(environment) is { } converter) |
| 107 | + for (var i = _converterFactories.Count - 1; i >= 0; i--) |
103 | 108 | { |
104 | | - TransformerImpl? deferTransformer; |
105 | | - |
106 | | - while (true) |
| 109 | + if (_converterFactories[i].TryCreate<TActualSource, TTarget>(environment) is { } converter) |
107 | 110 | { |
108 | | - if (deferTransformers is { } existingDeferTransformers) |
| 111 | + TransformerImpl? deferTransformer; |
| 112 | + |
| 113 | + while (true) |
109 | 114 | { |
110 | | - if (existingDeferTransformers[i] is { } existingDeferTransformer) |
| 115 | + if (deferTransformers is { } existingDeferTransformers) |
111 | 116 | { |
112 | | - deferTransformer = existingDeferTransformer; |
| 117 | + if (existingDeferTransformers[i] is { } existingDeferTransformer) |
| 118 | + { |
| 119 | + deferTransformer = existingDeferTransformer; |
| 120 | + |
| 121 | + break; |
| 122 | + } |
| 123 | + else |
| 124 | + { |
| 125 | + deferTransformer = new TransformerImpl(_converterFactories[..i], this); |
113 | 126 |
|
114 | | - break; |
| 127 | + if (Interlocked.CompareExchange(ref deferTransformers[i], deferTransformer, null) is { } otherDeferTransformer) |
| 128 | + deferTransformer = otherDeferTransformer; |
| 129 | + |
| 130 | + break; |
| 131 | + } |
115 | 132 | } |
116 | 133 | else |
117 | 134 | { |
118 | | - deferTransformer = new TransformerImpl(_converterFactories[..i], this); |
119 | | - |
120 | | - if (Interlocked.CompareExchange(ref deferTransformers[i], deferTransformer, null) is { } otherDeferTransformer) |
121 | | - deferTransformer = otherDeferTransformer; |
| 135 | + deferTransformers = new TransformerImpl[_converterFactories.Count]; |
122 | 136 |
|
123 | | - break; |
| 137 | + if (Interlocked.CompareExchange(ref _deferTransformers, deferTransformers, null) is { } otherDeferTransformers) |
| 138 | + deferTransformers = otherDeferTransformers; |
124 | 139 | } |
125 | 140 | } |
126 | | - else |
127 | | - { |
128 | | - deferTransformers = new TransformerImpl[_converterFactories.Count]; |
129 | 141 |
|
130 | | - if (Interlocked.CompareExchange(ref _deferTransformers, deferTransformers, null) is { } otherDeferTransformers) |
131 | | - deferTransformers = otherDeferTransformers; |
132 | | - } |
| 142 | + tuples[tupleCount++] = (converter, deferTransformer); |
133 | 143 | } |
134 | | - |
135 | | - list.Add((converter, deferTransformer)); |
136 | 144 | } |
137 | | - } |
138 | 145 |
|
139 | | - return new UnifiedConverter<TStaticSource, TActualSource, TTarget>([.. list], _recurse); |
| 146 | + return new UnifiedConverter<TStaticSource, TActualSource, TTarget>([.. tuples[0..tupleCount]], _recurse); |
| 147 | + } |
| 148 | + finally |
| 149 | + { |
| 150 | + arrayPool |
| 151 | + .Return(tuples); |
| 152 | + } |
140 | 153 | } |
141 | 154 | } |
142 | 155 |
|
|
0 commit comments