| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using System.Diagnostics; |
| | | 4 | | |
| | | 5 | | namespace Ice.Internal; |
| | | 6 | | |
| | | 7 | | internal sealed class BatchRequestI : BatchRequest |
| | | 8 | | { |
| | | 9 | | internal BatchRequestI(BatchRequestQueue queue) => _queue = queue; |
| | | 10 | | |
| | | 11 | | internal void reset(ObjectPrx proxy, string operation, int size) |
| | | 12 | | { |
| | | 13 | | _proxy = proxy; |
| | | 14 | | _operation = operation; |
| | | 15 | | _size = size; |
| | | 16 | | } |
| | | 17 | | |
| | | 18 | | public void enqueue() => _queue.enqueueBatchRequest(_proxy); |
| | | 19 | | |
| | | 20 | | public ObjectPrx getProxy() => _proxy; |
| | | 21 | | |
| | | 22 | | public string getOperation() => _operation; |
| | | 23 | | |
| | | 24 | | public int getSize() => _size; |
| | | 25 | | |
| | | 26 | | private readonly BatchRequestQueue _queue; |
| | | 27 | | private ObjectPrx _proxy; |
| | | 28 | | private string _operation; |
| | | 29 | | private int _size; |
| | | 30 | | } |
| | | 31 | | |
| | | 32 | | internal sealed class BatchRequestQueue |
| | | 33 | | { |
| | 1 | 34 | | internal BatchRequestQueue(Instance instance, bool datagram) |
| | | 35 | | { |
| | 1 | 36 | | InitializationData initData = instance.initializationData(); |
| | 1 | 37 | | _interceptor = initData.batchRequestInterceptor; |
| | 1 | 38 | | _batchStreamInUse = false; |
| | 1 | 39 | | _batchRequestNum = 0; |
| | 1 | 40 | | _batchStream = |
| | 1 | 41 | | new OutputStream(Protocol.currentProtocolEncoding, instance.defaultsAndOverrides().defaultFormat); |
| | 1 | 42 | | _batchStream.writeBlob(Protocol.requestBatchHdr); |
| | 1 | 43 | | _batchMarker = _batchStream.size(); |
| | 1 | 44 | | _request = new BatchRequestI(this); |
| | | 45 | | |
| | 1 | 46 | | _maxSize = instance.batchAutoFlushSize(); |
| | 1 | 47 | | if (_maxSize > 0 && datagram) |
| | | 48 | | { |
| | 1 | 49 | | int udpSndSize = initData.properties.getPropertyAsIntWithDefault( |
| | 1 | 50 | | "Ice.UDP.SndSize", |
| | 1 | 51 | | 65535 - _udpOverhead); |
| | 1 | 52 | | if (udpSndSize < _maxSize) |
| | | 53 | | { |
| | 1 | 54 | | _maxSize = udpSndSize; |
| | | 55 | | } |
| | | 56 | | } |
| | 1 | 57 | | } |
| | | 58 | | |
| | | 59 | | internal void prepareBatchRequest(OutputStream os) |
| | | 60 | | { |
| | 1 | 61 | | lock (_mutex) |
| | | 62 | | { |
| | 1 | 63 | | if (_exception != null) |
| | | 64 | | { |
| | 0 | 65 | | throw _exception; |
| | | 66 | | } |
| | | 67 | | |
| | | 68 | | // This is similar to a mutex lock in that the stream is only "locked" while marshaling. |
| | 1 | 69 | | while (_batchStreamInUse) |
| | | 70 | | { |
| | 0 | 71 | | Monitor.Wait(_mutex); |
| | | 72 | | } |
| | 1 | 73 | | _batchStreamInUse = true; |
| | 1 | 74 | | _batchStream.swap(os); |
| | 1 | 75 | | } |
| | 1 | 76 | | } |
| | | 77 | | |
| | | 78 | | internal void finishBatchRequest(OutputStream os, ObjectPrx proxy, string operation) |
| | | 79 | | { |
| | 1 | 80 | | lock (_mutex) |
| | | 81 | | { |
| | | 82 | | // Bring the request stream back and become the owner so this thread's own auto-flush below can |
| | | 83 | | // re-enter swap(). swap() lets only _batchStreamOwner through while the stream is in use, and only |
| | | 84 | | // after passing that gate does it read the queue's bookkeeping; no other thread can touch the |
| | | 85 | | // queue, so the mutations below need no further synchronization. |
| | | 86 | | Debug.Assert(_batchStreamInUse); |
| | 1 | 87 | | _batchStream.swap(os); |
| | 1 | 88 | | _batchStreamOwner = Thread.CurrentThread; |
| | 1 | 89 | | } |
| | | 90 | | |
| | | 91 | | try |
| | | 92 | | { |
| | 1 | 93 | | if (_maxSize > 0 && _batchStream.size() >= _maxSize) |
| | | 94 | | { |
| | 1 | 95 | | _ = proxy.ice_flushBatchRequestsAsync(); // Auto flush |
| | | 96 | | } |
| | | 97 | | |
| | | 98 | | Debug.Assert(_batchMarker < _batchStream.size()); |
| | 1 | 99 | | if (_interceptor != null) |
| | | 100 | | { |
| | 1 | 101 | | _request.reset(proxy, operation, _batchStream.size() - _batchMarker); |
| | 1 | 102 | | _interceptor(_request, _batchRequestNum, _batchMarker); |
| | | 103 | | } |
| | | 104 | | else |
| | | 105 | | { |
| | 1 | 106 | | bool? compress = ((ObjectPrxHelperBase)proxy).iceReference().getCompressOverride(); |
| | 1 | 107 | | if (compress is not null) |
| | | 108 | | { |
| | 1 | 109 | | _batchCompress |= compress.Value; |
| | | 110 | | } |
| | 1 | 111 | | _batchMarker = _batchStream.size(); |
| | 1 | 112 | | ++_batchRequestNum; |
| | | 113 | | } |
| | 1 | 114 | | } |
| | | 115 | | finally |
| | | 116 | | { |
| | 1 | 117 | | lock (_mutex) |
| | | 118 | | { |
| | 1 | 119 | | _batchStream.resize(_batchMarker); |
| | 1 | 120 | | _batchStreamInUse = false; |
| | 1 | 121 | | _batchStreamOwner = null; |
| | 1 | 122 | | Monitor.PulseAll(_mutex); |
| | 1 | 123 | | } |
| | 1 | 124 | | } |
| | 1 | 125 | | } |
| | | 126 | | |
| | | 127 | | internal void abortBatchRequest(OutputStream os) |
| | | 128 | | { |
| | 0 | 129 | | lock (_mutex) |
| | | 130 | | { |
| | 0 | 131 | | if (_batchStreamInUse) |
| | | 132 | | { |
| | 0 | 133 | | _batchStream.swap(os); |
| | 0 | 134 | | _batchStream.resize(_batchMarker); |
| | 0 | 135 | | _batchStreamInUse = false; |
| | 0 | 136 | | Monitor.PulseAll(_mutex); |
| | | 137 | | } |
| | 0 | 138 | | } |
| | 0 | 139 | | } |
| | | 140 | | |
| | | 141 | | internal int swap(OutputStream os, out bool compress) |
| | | 142 | | { |
| | 1 | 143 | | lock (_mutex) |
| | | 144 | | { |
| | | 145 | | // Only the thread that currently owns the in-use stream may bypass the wait, to run its own |
| | | 146 | | // auto-flush; every other caller waits for the in-progress batch request to finish. |
| | 1 | 147 | | if (_batchStreamOwner != Thread.CurrentThread) |
| | | 148 | | { |
| | 1 | 149 | | while (_batchStreamInUse) |
| | | 150 | | { |
| | 0 | 151 | | Monitor.Wait(_mutex); |
| | | 152 | | } |
| | | 153 | | } |
| | | 154 | | |
| | | 155 | | // Read the bookkeeping only after passing the gate above: finishBatchRequest mutates these scalars |
| | | 156 | | // without the lock while it owns the in-use stream, so reading _batchRequestNum before the wait |
| | | 157 | | // would race those writes. |
| | 1 | 158 | | if (_batchRequestNum == 0) |
| | | 159 | | { |
| | 1 | 160 | | compress = false; |
| | 1 | 161 | | return 0; |
| | | 162 | | } |
| | | 163 | | |
| | 1 | 164 | | byte[] lastRequest = null; |
| | 1 | 165 | | if (_batchMarker < _batchStream.size()) |
| | | 166 | | { |
| | 1 | 167 | | lastRequest = new byte[_batchStream.size() - _batchMarker]; |
| | 1 | 168 | | Buffer buffer = _batchStream.getBuffer(); |
| | 1 | 169 | | buffer.b.position(_batchMarker); |
| | 1 | 170 | | buffer.b.get(lastRequest); |
| | 1 | 171 | | _batchStream.resize(_batchMarker); |
| | | 172 | | } |
| | | 173 | | |
| | 1 | 174 | | int requestNum = _batchRequestNum; |
| | 1 | 175 | | compress = _batchCompress; |
| | 1 | 176 | | _batchStream.swap(os); |
| | | 177 | | |
| | | 178 | | // |
| | | 179 | | // Reset the batch. |
| | | 180 | | // |
| | 1 | 181 | | _batchRequestNum = 0; |
| | 1 | 182 | | _batchCompress = false; |
| | 1 | 183 | | _batchStream.writeBlob(Protocol.requestBatchHdr); |
| | 1 | 184 | | _batchMarker = _batchStream.size(); |
| | 1 | 185 | | if (lastRequest != null) |
| | | 186 | | { |
| | 1 | 187 | | _batchStream.writeBlob(lastRequest); |
| | | 188 | | } |
| | 1 | 189 | | return requestNum; |
| | | 190 | | } |
| | 1 | 191 | | } |
| | | 192 | | |
| | | 193 | | internal void destroy(LocalException ex) |
| | | 194 | | { |
| | 1 | 195 | | lock (_mutex) |
| | | 196 | | { |
| | 1 | 197 | | _exception = ex; |
| | 1 | 198 | | } |
| | 1 | 199 | | } |
| | | 200 | | |
| | | 201 | | internal void enqueueBatchRequest(ObjectPrx proxy) |
| | | 202 | | { |
| | | 203 | | Debug.Assert(_batchMarker < _batchStream.size()); |
| | 1 | 204 | | bool? compress = ((ObjectPrxHelperBase)proxy).iceReference().getCompressOverride(); |
| | 1 | 205 | | if (compress is not null) |
| | | 206 | | { |
| | 1 | 207 | | _batchCompress |= compress.Value; |
| | | 208 | | } |
| | 1 | 209 | | _batchMarker = _batchStream.size(); |
| | 1 | 210 | | ++_batchRequestNum; |
| | 1 | 211 | | } |
| | | 212 | | |
| | 1 | 213 | | private readonly object _mutex = new(); |
| | | 214 | | |
| | | 215 | | private readonly System.Action<BatchRequest, int, int> _interceptor; |
| | | 216 | | private readonly OutputStream _batchStream; |
| | | 217 | | private bool _batchStreamInUse; |
| | | 218 | | |
| | | 219 | | // While finishBatchRequest holds the in-use stream, this is its thread: the only thread allowed to |
| | | 220 | | // re-enter swap() (for its own auto-flush). A null owner means no thread may flush the in-use stream, so |
| | | 221 | | // other threads wait for _batchStreamInUse to clear. |
| | | 222 | | private Thread _batchStreamOwner; |
| | | 223 | | |
| | | 224 | | private int _batchRequestNum; |
| | | 225 | | private int _batchMarker; |
| | | 226 | | private bool _batchCompress; |
| | | 227 | | private readonly BatchRequestI _request; |
| | | 228 | | private LocalException _exception; |
| | | 229 | | private readonly int _maxSize; |
| | | 230 | | private const int _udpOverhead = 20 + 8; |
| | | 231 | | } |