public void Unsub(SubscriptionInfo subscriptionInfo) { if (subscriptionInfo == null) { throw new ArgumentNullException(nameof(subscriptionInfo)); } if (!TryRemoveSubscription(subscriptionInfo)) { return; } if (!IsConnected) { return; } ThrowIfDisposed(); ThrowIfNotConnected(); _connection.WithWriteLock((writer, arg) => { UnsubCmd.Write(writer, arg.Id, arg.MaxMessages); writer.Flush(); }, subscriptionInfo); }
public async Task UnsubAsync(SubscriptionInfo subscriptionInfo) { if (subscriptionInfo == null) { throw new ArgumentNullException(nameof(subscriptionInfo)); } if (!TryRemoveSubscription(subscriptionInfo)) { return; } if (!IsConnected) { return; } ThrowIfDisposed(); ThrowIfNotConnected(); await _connection.WithWriteLockAsync(async (writer, arg) => { await UnsubCmd.WriteAsync(writer, arg.Id.AsMemory(), arg.MaxMessages).ConfigureAwait(false); await writer.FlushAsync().ConfigureAwait(false); }, subscriptionInfo).ConfigureAwait(false); }
private async Task <MsgOp> DoRequestAsync(string subject, ReadOnlyMemory <byte> body, CancellationToken cancellationToken = default) { var replyToSubject = UniqueId.Generate(); var taskComp = new TaskCompletionSource <MsgOp>(); using var _ = MsgOpStream .WhereSubjectMatches(replyToSubject) .SubscribeSafe(msg => taskComp.SetResult(msg), ex => taskComp.SetException(ex)); using var cts = cancellationToken == default ? new CancellationTokenSource(_connectionInfo.RequestTimeoutMs) : CancellationTokenSource.CreateLinkedTokenSource(_cancellation.Token, cancellationToken); await using var __ = cts.Token.Register(() => taskComp.SetCanceled()).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); await _connection.WithWriteLockAsync(async (writer, arg) => { var(subjectIn, bodyIn, replyToIn) = arg; var sid = UniqueId.Generate().AsMemory(); await SubCmd.WriteAsync(writer, replyToIn, sid, ReadOnlyMemory <char> .Empty).ConfigureAwait(false); await UnsubCmd.WriteAsync(writer, sid, 1).ConfigureAwait(false); await writer.FlushAsync().ConfigureAwait(false); await PubCmd.WriteAsync(writer, subjectIn, replyToIn, bodyIn).ConfigureAwait(false); await writer.FlushAsync().ConfigureAwait(false); }, Tuple.Create(subject.AsMemory(), body, replyToSubject.AsMemory())).ConfigureAwait(false); return(await taskComp.Task.ConfigureAwait(false)); }
public async Task UnsubAsync(SubscriptionInfo subscriptionInfo) { EnsureArg.IsNotNull(subscriptionInfo, nameof(subscriptionInfo)); ThrowIfDisposed(); if (!TryRemoveSubscription(subscriptionInfo)) { return; } if (!IsConnected) { return; } var cmdPayload = UnsubCmd.Generate(subscriptionInfo.Id, subscriptionInfo.MaxMessages); await _connection.WithWriteLockAsync(async writer => { if (!IsConnected) { return; } await writer.WriteAsync(cmdPayload).ConfigureAwait(false); await writer.FlushAsync().ConfigureAwait(false); }).ConfigureAwait(false); }
public void Unsub(SubscriptionInfo subscriptionInfo) { EnsureArg.IsNotNull(subscriptionInfo, nameof(subscriptionInfo)); ThrowIfDisposed(); if (!TryRemoveSubscription(subscriptionInfo)) { return; } if (!IsConnected) { return; } var cmdPayload = UnsubCmd.Generate(subscriptionInfo.Id, subscriptionInfo.MaxMessages); _connection.WithWriteLock(writer => { if (!IsConnected) { return; } writer.Write(cmdPayload); writer.Flush(); }); }
private async Task DoSubAsync(SubscriptionInfo subscriptionInfo) => await _connection.WithWriteLockAsync(async writer => { await writer.WriteAsync(SubCmd.Generate(subscriptionInfo.Subject, subscriptionInfo.Id, subscriptionInfo.QueueGroup)).ConfigureAwait(false); if (subscriptionInfo.MaxMessages.HasValue) { await writer.WriteAsync(UnsubCmd.Generate(subscriptionInfo.Id, subscriptionInfo.MaxMessages)).ConfigureAwait(false); } await writer.FlushAsync().ConfigureAwait(false); }).ConfigureAwait(false);
private void DoSub(SubscriptionInfo subscriptionInfo) => _connection.WithWriteLock(writer => { writer.Write(SubCmd.Generate(subscriptionInfo.Subject, subscriptionInfo.Id, subscriptionInfo.QueueGroup)); if (subscriptionInfo.MaxMessages.HasValue) { writer.Write(UnsubCmd.Generate(subscriptionInfo.Id, subscriptionInfo.MaxMessages)); } writer.Flush(); });
private void DoSub(SubscriptionInfo subscriptionInfo) => _connection.WithWriteLock((writer, arg) => { var sid = arg.Id.AsSpan(); SubCmd.Write(writer, arg.Subject, sid, arg.QueueGroup); if (arg.MaxMessages.HasValue) { UnsubCmd.Write(writer, sid, arg.MaxMessages); } writer.Flush(); }, subscriptionInfo);
private async Task <MsgOp> DoRequestAsync(string subject, byte[] body, int?timeoutMs) { var requestReplyAddress = $"{Guid.NewGuid():N}"; var pubCmd = PubCmd.Generate(subject, body, requestReplyAddress); var taskComp = new TaskCompletionSource <MsgOp>(); var requestSubscription = MsgOpStream.Where(msg => msg.Subject == requestReplyAddress).Subscribe( msg => taskComp.SetResult(msg), ex => taskComp.SetException(ex)); var subscriptionInfo = new SubscriptionInfo(requestReplyAddress, maxMessages: 1); var subCmd = SubCmd.Generate(subscriptionInfo.Subject, subscriptionInfo.Id); var unsubCmd = UnsubCmd.Generate(subscriptionInfo.Id, subscriptionInfo.MaxMessages); await _connection.WithWriteLockAsync(async writer => { await writer.WriteAsync(subCmd).ConfigureAwait(false); await writer.WriteAsync(unsubCmd).ConfigureAwait(false); await writer.FlushAsync().ConfigureAwait(false); await writer.WriteAsync(pubCmd).ConfigureAwait(false); await writer.FlushAsync().ConfigureAwait(false); }).ConfigureAwait(false); Task.WaitAny(new[] { Task.Delay(timeoutMs ?? _connectionInfo.RequestTimeoutMs), taskComp.Task }, _cancellation.Token); if (!taskComp.Task.IsCompleted) { taskComp.SetException(NatsException.RequestTimedOut()); } return(await taskComp.Task .ContinueWith(t => { requestSubscription?.Dispose(); if (!t.IsFaulted) { return t.Result; } var ex = t.Exception?.GetBaseException() ?? t.Exception; if (ex == null) { return t.Result; } _logger.Error("Exception while performing request.", ex); throw ex; }) .ConfigureAwait(false)); }
private Task DoSubAsync(SubscriptionInfo subscriptionInfo) => _connection.WithWriteLockAsync(async(writer, arg) => { var sid = arg.Id.AsMemory(); await SubCmd.WriteAsync(writer, arg.Subject.AsMemory(), sid, arg.QueueGroup.AsMemory()) .ConfigureAwait(false); if (arg.MaxMessages.HasValue) { await UnsubCmd.WriteAsync(writer, sid, arg.MaxMessages).ConfigureAwait(false); } await writer.FlushAsync().ConfigureAwait(false); }, subscriptionInfo);
private Task DoSubAsync(SubscriptionInfo[] subscriptionInfos) => _connection.WithWriteLockAsync(async(writer, arg) => { foreach (var subscriptionInfo in arg) { var sid = subscriptionInfo.Id.AsMemory(); await SubCmd.WriteAsync(writer, subscriptionInfo.Subject.AsMemory(), sid, subscriptionInfo.QueueGroup.AsMemory()) .ConfigureAwait(false); if (subscriptionInfo.MaxMessages.HasValue) { await UnsubCmd.WriteAsync(writer, sid, subscriptionInfo.MaxMessages).ConfigureAwait(false); } await writer.FlushAsync().ConfigureAwait(false); } }, subscriptionInfos);