-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
UniqueRequestIDPolicy.ts
46 lines (41 loc) · 1.25 KB
/
UniqueRequestIDPolicy.ts
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
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
import {
BaseRequestPolicy,
generateUuid,
HttpOperationResponse,
RequestPolicy,
RequestPolicyOptions,
WebResource
} from "@azure/ms-rest-js";
import { HeaderConstants } from "../utils/constants";
/**
* UniqueRequestIDPolicy generates an UUID as x-ms-request-id header value.
*
* @class UniqueRequestIDPolicy
* @extends {BaseRequestPolicy}
*/
export class UniqueRequestIDPolicy extends BaseRequestPolicy {
/**
* Creates an instance of UniqueRequestIDPolicy.
* @param {RequestPolicy} nextPolicy
* @param {RequestPolicyOptions} options
* @memberof UniqueRequestIDPolicy
*/
constructor(nextPolicy: RequestPolicy, options: RequestPolicyOptions) {
super(nextPolicy, options);
}
/**
* Sends request.
*
* @param {WebResource} request
* @returns {Promise<HttpOperationResponse>}
* @memberof UniqueRequestIDPolicy
*/
public async sendRequest(request: WebResource): Promise<HttpOperationResponse> {
if (!request.headers.contains(HeaderConstants.X_MS_CLIENT_REQUEST_ID)) {
request.headers.set(HeaderConstants.X_MS_CLIENT_REQUEST_ID, generateUuid());
}
return this._nextPolicy.sendRequest(request);
}
}