@@ -2,6 +2,7 @@ package http
2
2
3
3
import (
4
4
"bytes"
5
+ "context"
5
6
"errors"
6
7
"fmt"
7
8
"io"
@@ -70,7 +71,7 @@ func (krc *KuberayAPIServerClient) CreateComputeTemplate(request *api.CreateComp
70
71
return nil , nil , fmt .Errorf ("failed to marshal api.ComputeTemplate to JSON: %w" , err )
71
72
}
72
73
73
- httpRequest , err := http .NewRequest ( "POST" , createURL , bytes .NewReader (bytez ))
74
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "POST" , createURL , bytes .NewReader (bytez ))
74
75
if err != nil {
75
76
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , createURL , err )
76
77
}
@@ -99,7 +100,7 @@ func (krc *KuberayAPIServerClient) DeleteComputeTemplate(request *api.DeleteComp
99
100
// Finds a specific compute template by its name and namespace.
100
101
func (krc * KuberayAPIServerClient ) GetComputeTemplate (request * api.GetComputeTemplateRequest ) (* api.ComputeTemplate , * rpcStatus.Status , error ) {
101
102
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/compute_templates/" + request .Name
102
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
103
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
103
104
if err != nil {
104
105
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
105
106
}
@@ -120,7 +121,7 @@ func (krc *KuberayAPIServerClient) GetComputeTemplate(request *api.GetComputeTem
120
121
// GetAllComputeTemplates finds all compute templates in all namespaces.
121
122
func (krc * KuberayAPIServerClient ) GetAllComputeTemplates () (* api.ListAllComputeTemplatesResponse , * rpcStatus.Status , error ) {
122
123
getURL := krc .baseURL + "/apis/v1/compute_templates"
123
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
124
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
124
125
if err != nil {
125
126
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
126
127
}
@@ -141,7 +142,7 @@ func (krc *KuberayAPIServerClient) GetAllComputeTemplates() (*api.ListAllCompute
141
142
// GetAllComputeTemplatesInNamespace Finds all compute templates in a given namespace.
142
143
func (krc * KuberayAPIServerClient ) GetAllComputeTemplatesInNamespace (request * api.ListComputeTemplatesRequest ) (* api.ListComputeTemplatesResponse , * rpcStatus.Status , error ) {
143
144
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/compute_templates"
144
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
145
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
145
146
if err != nil {
146
147
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
147
148
}
@@ -168,7 +169,7 @@ func (krc *KuberayAPIServerClient) CreateCluster(request *api.CreateClusterReque
168
169
return nil , nil , fmt .Errorf ("failed to marshal api.Cluster to JSON: %w" , err )
169
170
}
170
171
171
- httpRequest , err := http .NewRequest ( "POST" , createURL , bytes .NewReader (bytez ))
172
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "POST" , createURL , bytes .NewReader (bytez ))
172
173
if err != nil {
173
174
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , createURL , err )
174
175
}
@@ -196,7 +197,7 @@ func (krc *KuberayAPIServerClient) DeleteCluster(request *api.DeleteClusterReque
196
197
// GetCluster finds a specific Cluster by ID.
197
198
func (krc * KuberayAPIServerClient ) GetCluster (request * api.GetClusterRequest ) (* api.Cluster , * rpcStatus.Status , error ) {
198
199
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/clusters/" + request .Name
199
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
200
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
200
201
if err != nil {
201
202
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
202
203
}
@@ -217,7 +218,7 @@ func (krc *KuberayAPIServerClient) GetCluster(request *api.GetClusterRequest) (*
217
218
// ListCluster finds all clusters in a given namespace.
218
219
func (krc * KuberayAPIServerClient ) ListClusters (request * api.ListClustersRequest ) (* api.ListClustersResponse , * rpcStatus.Status , error ) {
219
220
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/clusters"
220
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
221
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
221
222
if err != nil {
222
223
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
223
224
}
@@ -243,7 +244,7 @@ func (krc *KuberayAPIServerClient) ListClusters(request *api.ListClustersRequest
243
244
// ListAllClusters finds all Clusters in all namespaces. Supports pagination, and sorting on certain fields.
244
245
func (krc * KuberayAPIServerClient ) ListAllClusters (request * api.ListAllClustersRequest ) (* api.ListAllClustersResponse , * rpcStatus.Status , error ) {
245
246
getURL := krc .baseURL + "/apis/v1/clusters"
246
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
247
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
247
248
if err != nil {
248
249
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
249
250
}
@@ -274,7 +275,7 @@ func (krc *KuberayAPIServerClient) CreateRayJob(request *api.CreateRayJobRequest
274
275
return nil , nil , fmt .Errorf ("failed to marshal api.Cluster to JSON: %w" , err )
275
276
}
276
277
277
- httpRequest , err := http .NewRequest ( "POST" , createURL , bytes .NewReader (bytez ))
278
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "POST" , createURL , bytes .NewReader (bytez ))
278
279
if err != nil {
279
280
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , createURL , err )
280
281
}
@@ -296,7 +297,7 @@ func (krc *KuberayAPIServerClient) CreateRayJob(request *api.CreateRayJobRequest
296
297
// GetRayJob finds a specific job by its name and namespace.
297
298
func (krc * KuberayAPIServerClient ) GetRayJob (request * api.GetRayJobRequest ) (* api.RayJob , * rpcStatus.Status , error ) {
298
299
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/jobs/" + request .Name
299
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
300
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
300
301
if err != nil {
301
302
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
302
303
}
@@ -317,7 +318,7 @@ func (krc *KuberayAPIServerClient) GetRayJob(request *api.GetRayJobRequest) (*ap
317
318
// Finds all job in a given namespace.
318
319
func (krc * KuberayAPIServerClient ) ListRayJobs (request * api.ListRayJobsRequest ) (* api.ListRayJobsResponse , * rpcStatus.Status , error ) {
319
320
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/jobs"
320
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
321
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
321
322
if err != nil {
322
323
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
323
324
}
@@ -343,7 +344,7 @@ func (krc *KuberayAPIServerClient) ListRayJobs(request *api.ListRayJobsRequest)
343
344
// ListAllRayJobs Finds all job in all namespaces.
344
345
func (krc * KuberayAPIServerClient ) ListAllRayJobs () (* api.ListAllRayJobsResponse , * rpcStatus.Status , error ) {
345
346
getURL := krc .baseURL + "/apis/v1/jobs"
346
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
347
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
347
348
if err != nil {
348
349
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
349
350
}
@@ -375,7 +376,7 @@ func (krc *KuberayAPIServerClient) CreateRayService(request *api.CreateRayServic
375
376
return nil , nil , fmt .Errorf ("failed to marshal api.Cluster to JSON: %w" , err )
376
377
}
377
378
378
- httpRequest , err := http .NewRequest ( "POST" , createURL , bytes .NewReader (bytez ))
379
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "POST" , createURL , bytes .NewReader (bytez ))
379
380
if err != nil {
380
381
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , createURL , err )
381
382
}
@@ -402,7 +403,7 @@ func (krc *KuberayAPIServerClient) UpdateRayService(request *api.UpdateRayServic
402
403
return nil , nil , fmt .Errorf ("failed to marshal api.Cluster to JSON: %w" , err )
403
404
}
404
405
405
- httpRequest , err := http .NewRequest ( "PUT" , updateURL , bytes .NewReader (bytez ))
406
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "PUT" , updateURL , bytes .NewReader (bytez ))
406
407
if err != nil {
407
408
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , updateURL , err )
408
409
}
@@ -424,7 +425,7 @@ func (krc *KuberayAPIServerClient) UpdateRayService(request *api.UpdateRayServic
424
425
// Find a specific ray serve by name and namespace.
425
426
func (krc * KuberayAPIServerClient ) GetRayService (request * api.GetRayServiceRequest ) (* api.RayService , * rpcStatus.Status , error ) {
426
427
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/services/" + request .Name
427
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
428
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
428
429
if err != nil {
429
430
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
430
431
}
@@ -445,7 +446,7 @@ func (krc *KuberayAPIServerClient) GetRayService(request *api.GetRayServiceReque
445
446
// Finds all ray services in a given namespace. Supports pagination, and sorting on certain fields.
446
447
func (krc * KuberayAPIServerClient ) ListRayServices (request * api.ListRayServicesRequest ) (* api.ListRayServicesResponse , * rpcStatus.Status , error ) {
447
448
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/services"
448
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
449
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
449
450
if err != nil {
450
451
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
451
452
}
@@ -466,7 +467,7 @@ func (krc *KuberayAPIServerClient) ListRayServices(request *api.ListRayServicesR
466
467
// Finds all ray services in a given namespace. Supports pagination, and sorting on certain fields.
467
468
func (krc * KuberayAPIServerClient ) ListAllRayServices () (* api.ListAllRayServicesResponse , * rpcStatus.Status , error ) {
468
469
getURL := krc .baseURL + "/apis/v1/services"
469
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
470
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
470
471
if err != nil {
471
472
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
472
473
}
@@ -498,7 +499,7 @@ func (krc *KuberayAPIServerClient) SubmitRayJob(request *api.SubmitRayJobRequest
498
499
return nil , nil , fmt .Errorf ("failed to marshal api.Cluster to JSON: %w" , err )
499
500
}
500
501
501
- httpRequest , err := http .NewRequest ( "POST" , createURL , bytes .NewReader (bytez ))
502
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "POST" , createURL , bytes .NewReader (bytez ))
502
503
if err != nil {
503
504
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , createURL , err )
504
505
}
@@ -520,7 +521,7 @@ func (krc *KuberayAPIServerClient) SubmitRayJob(request *api.SubmitRayJobRequest
520
521
// GetRayJobDetails. Get details about specific job on a given cluster.
521
522
func (krc * KuberayAPIServerClient ) GetRayJobDetails (request * api.GetJobDetailsRequest ) (* api.JobSubmissionInfo , * rpcStatus.Status , error ) {
522
523
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/jobsubmissions/" + request .Clustername + "/" + request .Submissionid
523
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
524
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
524
525
if err != nil {
525
526
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
526
527
}
@@ -541,7 +542,7 @@ func (krc *KuberayAPIServerClient) GetRayJobDetails(request *api.GetJobDetailsRe
541
542
// GetRayJobLog. Get log for a specific job on a given cluster.
542
543
func (krc * KuberayAPIServerClient ) GetRayJobLog (request * api.GetJobLogRequest ) (* api.GetJobLogReply , * rpcStatus.Status , error ) {
543
544
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/jobsubmissions/" + request .Clustername + "/log/" + request .Submissionid
544
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
545
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
545
546
if err != nil {
546
547
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
547
548
}
@@ -562,7 +563,7 @@ func (krc *KuberayAPIServerClient) GetRayJobLog(request *api.GetJobLogRequest) (
562
563
// ListRayJobsCluster. List Ray jobs on a given cluster.
563
564
func (krc * KuberayAPIServerClient ) ListRayJobsCluster (request * api.ListJobDetailsRequest ) (* api.ListJobSubmissionInfo , * rpcStatus.Status , error ) {
564
565
getURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/jobsubmissions/" + request .Clustername
565
- httpRequest , err := http .NewRequest ( "GET" , getURL , nil )
566
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "GET" , getURL , nil )
566
567
if err != nil {
567
568
return nil , nil , fmt .Errorf ("failed to create http request for url '%s': %w" , getURL , err )
568
569
}
@@ -584,7 +585,7 @@ func (krc *KuberayAPIServerClient) ListRayJobsCluster(request *api.ListJobDetail
584
585
func (krc * KuberayAPIServerClient ) StopRayJob (request * api.StopRayJobSubmissionRequest ) (* rpcStatus.Status , error ) {
585
586
createURL := krc .baseURL + "/apis/v1/namespaces/" + request .Namespace + "/jobsubmissions/" + request .Clustername + "/" + request .Submissionid
586
587
587
- httpRequest , err := http .NewRequest ( "POST" , createURL , nil )
588
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "POST" , createURL , nil )
588
589
if err != nil {
589
590
return nil , fmt .Errorf ("failed to create http request for url '%s': %w" , createURL , err )
590
591
}
@@ -606,7 +607,7 @@ func (krc *KuberayAPIServerClient) DeleteRayJobCluster(request *api.DeleteRayJob
606
607
}
607
608
608
609
func (krc * KuberayAPIServerClient ) doDelete (deleteURL string ) (* rpcStatus.Status , error ) {
609
- httpRequest , err := http .NewRequest ( "DELETE" , deleteURL , nil )
610
+ httpRequest , err := http .NewRequestWithContext ( context . TODO (), "DELETE" , deleteURL , nil )
610
611
if err != nil {
611
612
return nil , fmt .Errorf ("failed to create http request for url '%s': %w" , deleteURL , err )
612
613
}
0 commit comments