Skip to content

Commit 603e586

Browse files
committed
addiong options on the client calls for transform start/stop
1 parent 8c2fb9a commit 603e586

File tree

1 file changed

+19
-7
lines changed

1 file changed

+19
-7
lines changed

internal/clients/elasticsearch/index.go

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -560,13 +560,13 @@ func PutTransform(ctx context.Context, apiClient *clients.ApiClient, transform *
560560
return diag.FromErr(err)
561561
}
562562

563-
opts := []func(*esapi.TransformPutTransformRequest){
563+
putOptions := []func(*esapi.TransformPutTransformRequest){
564564
esClient.TransformPutTransform.WithContext(ctx),
565565
esClient.TransformPutTransform.WithDeferValidation(params.DeferValidation),
566566
esClient.TransformPutTransform.WithTimeout(params.Timeout),
567567
}
568568

569-
res, err := esClient.TransformPutTransform(bytes.NewReader(transformBytes), transform.Name, opts...)
569+
res, err := esClient.TransformPutTransform(bytes.NewReader(transformBytes), transform.Name, putOptions...)
570570
if err != nil {
571571
return diag.FromErr(err)
572572
}
@@ -577,7 +577,11 @@ func PutTransform(ctx context.Context, apiClient *clients.ApiClient, transform *
577577
}
578578

579579
if params.Enabled {
580-
_, err := esClient.TransformStartTransform(transform.Name)
580+
startOptions := []func(*esapi.TransformStartTransformRequest){
581+
esClient.TransformStartTransform.WithContext(ctx),
582+
esClient.TransformStartTransform.WithTimeout(params.Timeout),
583+
}
584+
_, err := esClient.TransformStartTransform(transform.Name, startOptions...)
581585
if err != nil {
582586
return diag.FromErr(err)
583587
}
@@ -636,13 +640,13 @@ func UpdateTransform(ctx context.Context, apiClient *clients.ApiClient, transfor
636640
return diag.FromErr(err)
637641
}
638642

639-
opts := []func(*esapi.TransformUpdateTransformRequest){
643+
updateOptions := []func(*esapi.TransformUpdateTransformRequest){
640644
esClient.TransformUpdateTransform.WithContext(ctx),
641645
esClient.TransformUpdateTransform.WithDeferValidation(params.DeferValidation),
642646
esClient.TransformUpdateTransform.WithTimeout(params.Timeout),
643647
}
644648

645-
res, err := esClient.TransformUpdateTransform(bytes.NewReader(transformBytes), transform.Name, opts...)
649+
res, err := esClient.TransformUpdateTransform(bytes.NewReader(transformBytes), transform.Name, updateOptions...)
646650
if err != nil {
647651
return diag.FromErr(err)
648652
}
@@ -653,12 +657,20 @@ func UpdateTransform(ctx context.Context, apiClient *clients.ApiClient, transfor
653657
}
654658

655659
if params.Enabled {
656-
_, err := esClient.TransformStartTransform(transform.Name)
660+
startOptions := []func(*esapi.TransformStartTransformRequest){
661+
esClient.TransformStartTransform.WithContext(ctx),
662+
esClient.TransformStartTransform.WithTimeout(params.Timeout),
663+
}
664+
_, err := esClient.TransformStartTransform(transform.Name, startOptions...)
657665
if err != nil {
658666
return diag.FromErr(err)
659667
}
660668
} else {
661-
_, err := esClient.TransformStopTransform(transform.Name)
669+
stopOptions := []func(*esapi.TransformStopTransformRequest){
670+
esClient.TransformStopTransform.WithContext(ctx),
671+
esClient.TransformStopTransform.WithTimeout(params.Timeout),
672+
}
673+
_, err := esClient.TransformStopTransform(transform.Name, stopOptions...)
662674
if err != nil {
663675
return diag.FromErr(err)
664676
}

0 commit comments

Comments
 (0)