-
Notifications
You must be signed in to change notification settings - Fork 66
/
data_source_worker_pools.go
49 lines (40 loc) · 1.41 KB
/
data_source_worker_pools.go
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
package octopusdeploy
import (
"context"
"time"
"github.com/OctopusDeploy/go-octopusdeploy/v2/pkg/client"
"github.com/OctopusDeploy/go-octopusdeploy/v2/pkg/workerpools"
"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
)
func dataSourceWorkerPools() *schema.Resource {
return &schema.Resource{
Description: "Provides information about existing worker pools.",
ReadContext: dataSourceWorkerPoolsRead,
Schema: getWorkerPoolDataSchema(),
}
}
func dataSourceWorkerPoolsRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
query := workerpools.WorkerPoolsQuery{
IDs: expandArray(d.Get("ids").([]interface{})),
PartialName: d.Get("partial_name").(string),
Skip: d.Get("skip").(int),
Take: d.Get("take").(int),
}
client := m.(*client.Client)
workerPools, err := workerpools.Get(client, d.Get("space_id").(string), query)
if err != nil {
return diag.FromErr(err)
}
flattenedWorkerPools := []interface{}{}
for _, workerPool := range workerPools.Items {
workerPoolResource, err := workerpools.ToWorkerPoolResource(workerPool)
if err != nil {
return diag.FromErr(err)
}
flattenedWorkerPools = append(flattenedWorkerPools, flattenWorkerPool(workerPoolResource))
}
d.Set("worker_pools", flattenedWorkerPools)
d.SetId("Worker Pools " + time.Now().UTC().String())
return nil
}