-
Notifications
You must be signed in to change notification settings - Fork 234
/
http_client.h
651 lines (613 loc) · 32.9 KB
/
http_client.h
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
// Copyright 2020-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
// * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
// * Neither the name of NVIDIA CORPORATION nor the names of its
// contributors may be used to endorse or promote products derived
// from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
// OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#pragma once
#include <map>
#include <memory>
#include "common.h"
#include "ipc.h"
namespace triton { namespace client {
class HttpInferRequest;
/// The key-value map type to be included in the request
/// as custom headers.
typedef std::map<std::string, std::string> Headers;
/// The key-value map type to be included as URL parameters.
typedef std::map<std::string, std::string> Parameters;
// The options for authorizing and authenticating SSL/TLS connections.
struct HttpSslOptions {
enum CERTTYPE { CERT_PEM = 0, CERT_DER = 1 };
enum KEYTYPE {
KEY_PEM = 0,
KEY_DER = 1
// TODO TMA-1645: Support loading private key from crypto engine
// KEY_ENG = 2
};
explicit HttpSslOptions()
: verify_peer(1), verify_host(2), cert_type(CERTTYPE::CERT_PEM),
key_type(KEYTYPE::KEY_PEM)
{
}
// This option determines whether curl verifies the authenticity of the peer's
// certificate. A value of 1 means curl verifies; 0 (zero) means it does not.
// Default value is 1. See here for more details:
// https://curl.se/libcurl/c/CURLOPT_SSL_VERIFYPEER.html
long verify_peer;
// This option determines whether libcurl verifies that the server cert is for
// the server it is known as. The default value for this option is 2 which
// means that certificate must indicate that the server is the server to which
// you meant to connect, or the connection fails. See here for more details:
// https://curl.se/libcurl/c/CURLOPT_SSL_VERIFYHOST.html
long verify_host;
// File holding one or more certificates to verify the peer with. If not
// specified, client will look for the system path where cacert bundle is
// assumed to be stored, as established at build time. See here for more
// information: https://curl.se/libcurl/c/CURLOPT_CAINFO.html
std::string ca_info;
// The format of client certificate. By default it is CERT_PEM. See here for
// more details: https://curl.se/libcurl/c/CURLOPT_SSLCERTTYPE.html
CERTTYPE cert_type;
// The file name of your client certificate. See here for more details:
// https://curl.se/libcurl/c/CURLOPT_SSLCERT.html
std::string cert;
// The format of the private key. By default it is KEY_PEM. See here for more
// details: https://curl.se/libcurl/c/CURLOPT_SSLKEYTYPE.html.
KEYTYPE key_type;
// The private key. See here for more details:
// https://curl.se/libcurl/c/CURLOPT_SSLKEY.html.
std::string key;
};
//==============================================================================
/// An InferenceServerHttpClient object is used to perform any kind of
/// communication with the InferenceServer using HTTP protocol. None
/// of the methods of InferenceServerHttpClient are thread safe. The
/// class is intended to be used by a single thread and simultaneously
/// calling different methods with different threads is not supported
/// and will cause undefined behavior.
///
/// \code
/// std::unique_ptr<InferenceServerHttpClient> client;
/// InferenceServerHttpClient::Create(&client, "localhost:8000");
/// bool live;
/// client->IsServerLive(&live);
/// ...
/// ...
/// \endcode
///
class InferenceServerHttpClient : public InferenceServerClient {
public:
enum class CompressionType { NONE, DEFLATE, GZIP };
~InferenceServerHttpClient();
/// Generate a request body for inference using the supplied 'inputs' and
/// requesting the outputs specified by 'outputs'.
/// \param request_body Returns the generated inference request body
/// \param header_length Returns the length of the inference header.
/// \param options The options for inference request.
/// \param inputs The vector of InferInput describing the model inputs.
/// \param outputs The vector of InferRequestedOutput describing how the
/// output must be returned. If not provided then all the outputs in the
/// model config will be returned as default settings.
/// \return Error object indicating success or failure of the
/// request.
static Error GenerateRequestBody(
std::vector<char>* request_body, size_t* header_length,
const InferOptions& options, const std::vector<InferInput*>& inputs,
const std::vector<const InferRequestedOutput*>& outputs =
std::vector<const InferRequestedOutput*>());
/// Generate a InferResult object from the given 'response_body'.
/// \param result Returns the generated InferResult object.
/// \param response_body The inference response from the server
/// \param header_length The length of the inference header if the header
/// does not occupy the whole response body. 0 indicates that
/// the whole response body is the inference response header.
/// \return Error object indicating success or failure of the
/// request.
static Error ParseResponseBody(
InferResult** result, const std::vector<char>& response_body,
const size_t header_length = 0);
/// Create a client that can be used to communicate with the server.
/// \param client Returns a new InferenceServerHttpClient object.
/// \param server_url The inference server name, port, optional
/// scheme and optional base path in the following format:
/// <scheme://>host:port/<base-path>.
/// \param verbose If true generate verbose output when contacting
/// the inference server.
/// \param ssl_options Specifies the settings for configuring
/// SSL encryption and authorization. Providing these options
/// do not ensure that SSL/TLS will be used in communication.
/// The use of SSL/TLS depends entirely on the server endpoint.
/// These options will be ignored if the server_url does not
/// expose `https://` scheme.
/// \return Error object indicating success or failure.
static Error Create(
std::unique_ptr<InferenceServerHttpClient>* client,
const std::string& server_url, bool verbose = false,
const HttpSslOptions& ssl_options = HttpSslOptions());
/// Contact the inference server and get its liveness.
/// \param live Returns whether the server is live or not.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \return Error object indicating success or failure of the request.
Error IsServerLive(
bool* live, const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get its readiness.
/// \param ready Returns whether the server is ready or not.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error IsServerReady(
bool* ready, const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the readiness of specified model.
/// \param ready Returns whether the specified model is ready or not.
/// \param model_name The name of the model to check for readiness.
/// \param model_version The version of the model to check for readiness.
/// The default value is an empty string which means then the server will
/// choose a version based on the model and internal policy.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error IsModelReady(
bool* ready, const std::string& model_name,
const std::string& model_version = "", const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get its metadata.
/// \param server_metadata Returns JSON representation of the
/// metadata as a string.
/// \param headers Optional map specifying additional HTTP headers to
/// include in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error ServerMetadata(
std::string* server_metadata, const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the metadata of specified model.
/// \param model_metadata Returns JSON representation of model
/// metadata as a string.
/// \param model_name The name of the model to get metadata.
/// \param model_version The version of the model to get metadata.
/// The default value is an empty string which means then the server will
/// choose a version based on the model and internal policy.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error ModelMetadata(
std::string* model_metadata, const std::string& model_name,
const std::string& model_version = "", const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the configuration of specified model.
/// \param model_config Returns JSON representation of model
/// configuration as a string.
/// \param model_name The name of the model to get configuration.
/// \param model_version The version of the model to get configuration.
/// The default value is an empty string which means then the server will
/// choose a version based on the model and internal policy.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error ModelConfig(
std::string* model_config, const std::string& model_name,
const std::string& model_version = "", const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the index of model repository
/// contents.
/// \param repository_index Returns JSON representation of the
/// repository index as a string.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error ModelRepositoryIndex(
std::string* repository_index, const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Request the inference server to load or reload specified model.
/// \param model_name The name of the model to be loaded or reloaded.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \param config Optional JSON representation of a model config provided for
/// the load request, if provided, this config will be used for
/// loading the model.
/// \param files Optional map specifying file path (with "file:"
/// prefix) in the override model directory to the file content.
/// The files will form the model directory that the model
/// will be loaded from. If specified, 'config' must be provided to be
/// the model configuration of the override model directory.
/// \return Error object indicating success or failure of the request.
Error LoadModel(
const std::string& model_name, const Headers& headers = Headers(),
const Parameters& query_params = Parameters(),
const std::string& config = std::string(),
const std::map<std::string, std::vector<char>>& files = {});
/// Request the inference server to unload specified model.
/// \param model_name The name of the model to be unloaded.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error UnloadModel(
const std::string& model_name, const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the inference statistics for the
/// specified model name and version.
/// \param infer_stat Returns the JSON representation of the
/// inference statistics as a string.
/// \param model_name The name of the model to get inference statistics. The
/// default value is an empty string which means statistics of all models will
/// be returned in the response.
/// \param model_version The version of the model to get inference statistics.
/// The default value is an empty string which means then the server will
/// choose a version based on the model and internal policy.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error ModelInferenceStatistics(
std::string* infer_stat, const std::string& model_name = "",
const std::string& model_version = "", const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Update the trace settings for the specified model name, or global trace
/// settings if model name is not given.
/// \param response Returns the JSON representation of the updated trace
/// settings as a string.
/// \param model_name The name of the model to update trace settings. The
/// default value is an empty string which means the global trace settings
/// will be updated.
/// \param settings The new trace setting values. Only the settings listed
/// will be updated. If a trace setting is listed in the map with an empty
/// string, that setting will be cleared.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error UpdateTraceSettings(
std::string* response, const std::string& model_name = "",
const std::map<std::string, std::vector<std::string>>& settings =
std::map<std::string, std::vector<std::string>>(),
const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Get the trace settings for the specified model name, or global trace
/// settings if model name is not given.
/// \param settings Returns the JSON representation of the trace
/// settings as a string.
/// \param model_name The name of the model to get trace settings. The
/// default value is an empty string which means the global trace settings
/// will be returned.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error GetTraceSettings(
std::string* settings, const std::string& model_name = "",
const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the status for requested system
/// shared memory.
/// \param status Returns the JSON representation of the system
/// shared memory status as a string.
/// \param region_name The name of the region to query status. The default
/// value is an empty string, which means that the status of all active system
/// shared memory will be returned.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error SystemSharedMemoryStatus(
std::string* status, const std::string& region_name = "",
const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Request the server to register a system shared memory with the provided
/// details.
/// \param name The name of the region to register.
/// \param key The key of the underlying memory object that contains the
/// system shared memory region.
/// \param byte_size The size of the system shared memory region, in bytes.
/// \param offset Offset, in bytes, within the underlying memory object to
/// the start of the system shared memory region. The default value is zero.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request
Error RegisterSystemSharedMemory(
const std::string& name, const std::string& key, const size_t byte_size,
const size_t offset = 0, const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Request the server to unregister a system shared memory with the
/// specified name.
/// \param name The name of the region to unregister. The default value is
/// empty string which means all the system shared memory regions will be
/// unregistered.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request
Error UnregisterSystemSharedMemory(
const std::string& name = "", const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Contact the inference server and get the status for requested CUDA
/// shared memory.
/// \param status Returns the JSON representation of the CUDA shared
/// memory status as a string.
/// \param region_name The name of the region to query status. The default
/// value is an empty string, which means that the status of all active CUDA
/// shared memory will be returned.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request.
Error CudaSharedMemoryStatus(
std::string* status, const std::string& region_name = "",
const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Request the server to register a CUDA shared memory with the provided
/// details.
/// \param name The name of the region to register.
/// \param cuda_shm_handle The cudaIPC handle for the memory object.
/// \param device_id The GPU device ID on which the cudaIPC handle was
/// created.
/// \param byte_size The size of the CUDA shared memory region, in
/// bytes.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request
Error RegisterCudaSharedMemory(
const std::string& name, const cudaIpcMemHandle_t& cuda_shm_handle,
const size_t device_id, const size_t byte_size,
const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Request the server to unregister a CUDA shared memory with the
/// specified name.
/// \param name The name of the region to unregister. The default value is
/// empty string which means all the CUDA shared memory regions will be
/// unregistered.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \return Error object indicating success or failure of the request
Error UnregisterCudaSharedMemory(
const std::string& name = "", const Headers& headers = Headers(),
const Parameters& query_params = Parameters());
/// Run synchronous inference on server.
/// \param result Returns the result of inference.
/// \param options The options for inference request.
/// \param inputs The vector of InferInput describing the model inputs.
/// \param outputs The vector of InferRequestedOutput describing how the
/// output must be returned. If not provided then all the outputs in the
/// model config will be returned as default settings.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \param request_compression_algorithm Optional HTTP compression algorithm
/// to use for the request body on client side. Currently supports DEFLATE,
/// GZIP and NONE. By default, no compression is used.
/// \param response_compression_algorithm Optional HTTP compression algorithm
/// to request for the response body. Note that the response may not be
/// compressed if the server does not support the specified algorithm.
/// Currently supports DEFLATE, GZIP and NONE. By default, no compression
/// is used.
/// \return Error object indicating success or failure of the
/// request.
Error Infer(
InferResult** result, const InferOptions& options,
const std::vector<InferInput*>& inputs,
const std::vector<const InferRequestedOutput*>& outputs =
std::vector<const InferRequestedOutput*>(),
const Headers& headers = Headers(),
const Parameters& query_params = Parameters(),
const CompressionType request_compression_algorithm =
CompressionType::NONE,
const CompressionType response_compression_algorithm =
CompressionType::NONE);
/// Run asynchronous inference on server.
/// Once the request is completed, the InferResult pointer will be passed to
/// the provided 'callback' function. Upon the invocation of callback
/// function, the ownership of InferResult object is transferred to the
/// function caller. It is then the caller's choice on either retrieving the
/// results inside the callback function or deferring it to a different thread
/// so that the client is unblocked. In order to prevent memory leak, user
/// must ensure this object gets deleted.
/// Note: InferInput::AppendRaw() or InferInput::SetSharedMemory() calls do
/// not copy the data buffers but hold the pointers to the data directly.
/// It is advisable to not to disturb the buffer contents until the respective
/// callback is invoked.
/// \param callback The callback function to be invoked on request completion.
/// \param options The options for inference request.
/// \param inputs The vector of InferInput describing the model inputs.
/// \param outputs The vector of InferRequestedOutput describing how the
/// output must be returned. If not provided then all the outputs in the
/// model config will be returned as default settings.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \param request_compression_algorithm Optional HTTP compression algorithm
/// to use for the request body on client side. Currently supports DEFLATE,
/// GZIP and NONE. By default, no compression is used.
/// \param response_compression_algorithm Optional HTTP compression algorithm
/// to request for the response body. Note that the response may not be
/// compressed if the server does not support the specified algorithm.
/// Currently supports DEFLATE, GZIP and NONE. By default, no compression
/// is used.
/// \return Error object indicating success
/// or failure of the request.
Error AsyncInfer(
OnCompleteFn callback, const InferOptions& options,
const std::vector<InferInput*>& inputs,
const std::vector<const InferRequestedOutput*>& outputs =
std::vector<const InferRequestedOutput*>(),
const Headers& headers = Headers(),
const Parameters& query_params = Parameters(),
const CompressionType request_compression_algorithm =
CompressionType::NONE,
const CompressionType response_compression_algorithm =
CompressionType::NONE);
/// Run multiple synchronous inferences on server.
/// \param results Returns the results of the inferences.
/// \param options The options for each inference request, one set of
/// options may be provided and it will be used for all inference requests.
/// \param inputs The vector of InferInput objects describing the model inputs
/// for each inference request.
/// \param outputs Optional vector of InferRequestedOutput objects describing
/// how the output must be returned. If not provided then all the outputs in
/// the model config will be returned as default settings. And one set of
/// outputs may be provided and it will be used for all inference requests.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \param request_compression_algorithm Optional HTTP compression algorithm
/// to use for the request body on client side. Currently supports DEFLATE,
/// GZIP and NONE. By default, no compression is used.
/// \param response_compression_algorithm Optional HTTP compression algorithm
/// to request for the response body. Note that the response may not be
/// compressed if the server does not support the specified algorithm.
/// Currently supports DEFLATE, GZIP and NONE. By default, no compression
/// is used.
/// \return Error object indicating success or failure of the
/// request.
Error InferMulti(
std::vector<InferResult*>* results,
const std::vector<InferOptions>& options,
const std::vector<std::vector<InferInput*>>& inputs,
const std::vector<std::vector<const InferRequestedOutput*>>& outputs =
std::vector<std::vector<const InferRequestedOutput*>>(),
const Headers& headers = Headers(),
const Parameters& query_params = Parameters(),
const CompressionType request_compression_algorithm =
CompressionType::NONE,
const CompressionType response_compression_algorithm =
CompressionType::NONE);
/// Run multiple asynchronous inferences on server.
/// Once all the requests are completed, the vector of InferResult pointers
/// will be passed to the provided 'callback' function. Upon the invocation
/// of callback function, the ownership of the InferResult objects are
/// transferred to the function caller. It is then the caller's choice on
/// either retrieving the results inside the callback function or deferring it
/// to a different thread so that the client is unblocked. In order to
/// prevent memory leak, user must ensure these objects get deleted.
/// Note: InferInput::AppendRaw() or InferInput::SetSharedMemory() calls do
/// not copy the data buffers but hold the pointers to the data directly.
/// It is advisable to not to disturb the buffer contents until the respective
/// callback is invoked.
/// \param callback The callback function to be invoked on the completion of
/// all requests.
/// \param options The options for each inference request, one set of
/// option may be provided and it will be used for all inference requests.
/// \param inputs The vector of InferInput objects describing the model inputs
/// for each inference request.
/// \param outputs Optional vector of InferRequestedOutput objects describing
/// how the output must be returned. If not provided then all the outputs in
/// the model config will be returned as default settings. And one set of
/// outputs may be provided and it will be used for all inference requests.
/// \param headers Optional map specifying additional HTTP headers to include
/// in request.
/// \param query_params Optional map specifying parameters that must be
/// included with URL query.
/// \param request_compression_algorithm Optional HTTP compression algorithm
/// to use for the request body on client side. Currently supports DEFLATE,
/// GZIP and NONE. By default, no compression is used.
/// \param response_compression_algorithm Optional HTTP compression algorithm
/// to request for the response body. Note that the response may not be
/// compressed if the server does not support the specified algorithm.
/// Currently supports DEFLATE, GZIP and NONE. By default, no compression
/// is used.
/// \return Error object indicating success
/// or failure of the request.
Error AsyncInferMulti(
OnMultiCompleteFn callback, const std::vector<InferOptions>& options,
const std::vector<std::vector<InferInput*>>& inputs,
const std::vector<std::vector<const InferRequestedOutput*>>& outputs =
std::vector<std::vector<const InferRequestedOutput*>>(),
const Headers& headers = Headers(),
const Parameters& query_params = Parameters(),
const CompressionType request_compression_algorithm =
CompressionType::NONE,
const CompressionType response_compression_algorithm =
CompressionType::NONE);
private:
InferenceServerHttpClient(
const std::string& url, bool verbose, const HttpSslOptions& ssl_options);
Error PreRunProcessing(
void* curl, std::string& request_uri, const InferOptions& options,
const std::vector<InferInput*>& inputs,
const std::vector<const InferRequestedOutput*>& outputs,
const Headers& headers, const Parameters& query_params,
const CompressionType request_compression_algorithm,
const CompressionType response_compression_algorithm,
std::shared_ptr<HttpInferRequest>& request);
void AsyncTransfer();
Error Get(
std::string& request_uri, const Headers& headers,
const Parameters& query_params, std::string* response,
long* http_code = nullptr);
Error Post(
std::string& request_uri, const std::string& request,
const Headers& headers, const Parameters& query_params,
std::string* response);
static size_t ResponseHandler(
void* contents, size_t size, size_t nmemb, void* userp);
static size_t InferRequestProvider(
void* contents, size_t size, size_t nmemb, void* userp);
static size_t InferResponseHeaderHandler(
void* contents, size_t size, size_t nmemb, void* userp);
static size_t InferResponseHandler(
void* contents, size_t size, size_t nmemb, void* userp);
// The server url
const std::string url_;
// The options for authorizing and authenticating SSL/TLS connections
HttpSslOptions ssl_options_;
using AsyncReqMap = std::map<uintptr_t, std::shared_ptr<HttpInferRequest>>;
// curl easy handle shared for all synchronous requests
void* easy_handle_;
// curl multi handle for processing asynchronous requests
void* multi_handle_;
// map to record new asynchronous requests with pointer to easy handle
// or tag id as key
AsyncReqMap new_async_requests_;
};
}} // namespace triton::client