forked from MassTransit/MassTransit
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMessageRequestClient.cs
More file actions
74 lines (68 loc) · 3.65 KB
/
MessageRequestClient.cs
File metadata and controls
74 lines (68 loc) · 3.65 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
// Copyright 2007-2016 Chris Patterson, Dru Sellers, Travis Smith, et. al.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use
// this file except in compliance with the License. You may obtain a copy of the
// License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
namespace MassTransit
{
using System;
using System.Threading;
using System.Threading.Tasks;
using GreenPipes;
using Pipeline;
/// <summary>
/// Implements a request client that uses specified endpoint to send the request and return the
/// response task.
/// </summary>
/// <typeparam name="TRequest">The request message type</typeparam>
/// <typeparam name="TResponse">The response message type</typeparam>
public class MessageRequestClient<TRequest, TResponse> :
RequestClient<TRequest, TResponse>
where TRequest : class
where TResponse : class
{
readonly Lazy<Task<ISendEndpoint>> _requestEndpoint;
/// <summary>
/// Creates a message request client for the bus and endpoint specified
/// </summary>
/// <param name="bus">The bus instance</param>
/// <param name="serviceAddress">The service endpoint address</param>
/// <param name="timeout">The request timeout</param>
/// <param name="timeToLive">The time that the request will live for</param>
/// <param name="callback"></param>
public MessageRequestClient(IBus bus, Uri serviceAddress, TimeSpan timeout, TimeSpan? timeToLive = default(TimeSpan?),
Action<SendContext<TRequest>> callback = null)
: base(bus, bus.Address, timeout, timeToLive, callback)
{
_requestEndpoint = new Lazy<Task<ISendEndpoint>>(async () => await bus.GetSendEndpoint(serviceAddress).ConfigureAwait(false));
}
/// <summary>
/// Creates a message request client for the bus and endpoint specified
/// </summary>
/// <param name="sendEndpointProvider"></param>
/// <param name="connector">The bus instance</param>
/// <param name="responseAddress">The response address of the connector</param>
/// <param name="serviceAddress">The service endpoint address</param>
/// <param name="timeout">The request timeout</param>
/// <param name="timeToLive">The time that the request will live for</param>
/// <param name="callback"></param>
public MessageRequestClient(ISendEndpointProvider sendEndpointProvider, IRequestPipeConnector connector, Uri responseAddress, Uri serviceAddress,
TimeSpan timeout, TimeSpan? timeToLive = default(TimeSpan?), Action<SendContext<TRequest>> callback = null)
: base(connector, responseAddress, timeout, timeToLive, callback)
{
_requestEndpoint = new Lazy<Task<ISendEndpoint>>(async () => await sendEndpointProvider.GetSendEndpoint(serviceAddress).ConfigureAwait(false));
}
protected override async Task SendRequest(TRequest request, IPipe<SendContext<TRequest>> requestPipe, CancellationToken cancellationToken)
{
var endpoint = await _requestEndpoint.Value.ConfigureAwait(false);
await endpoint.Send(request, requestPipe, cancellationToken).ConfigureAwait(false);
}
}
}