forked from concourse/concourse
/
workers.go
94 lines (78 loc) · 2.05 KB
/
workers.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
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
package concourse
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
"time"
"github.com/chenbh/concourse/v6/atc"
"github.com/chenbh/concourse/v6/go-concourse/concourse/internal"
"github.com/tedsuo/rata"
)
type PruneWorkerError struct {
atc.PruneWorkerResponseBody
}
func (e PruneWorkerError) Error() string {
return e.Stderr
}
func (client *client) ListWorkers() ([]atc.Worker, error) {
var workers []atc.Worker
err := client.connection.Send(internal.Request{
RequestName: atc.ListWorkers,
}, &internal.Response{
Result: &workers,
})
return workers, err
}
func (client *client) SaveWorker(worker atc.Worker, ttl *time.Duration) (*atc.Worker, error) {
buffer := &bytes.Buffer{}
err := json.NewEncoder(buffer).Encode(worker)
if err != nil {
return nil, fmt.Errorf("Unable to marshal worker: %s", err)
}
params := rata.Params{}
if ttl != nil {
params["ttl"] = ttl.String()
}
var savedWorker *atc.Worker
err = client.connection.Send(internal.Request{
RequestName: atc.RegisterWorker,
Body: buffer,
Params: params,
}, &internal.Response{
Result: &savedWorker,
})
return savedWorker, err
}
func (client *client) PruneWorker(workerName string) error {
params := rata.Params{"worker_name": workerName}
err := client.connection.Send(internal.Request{
RequestName: atc.PruneWorker,
Params: params,
Header: http.Header{
"Content-Type": {"application/json"},
},
}, nil)
if unexpectedResponseError, ok := err.(internal.UnexpectedResponseError); ok {
if unexpectedResponseError.StatusCode == http.StatusBadRequest {
var pruneWorkerErr PruneWorkerError
err = json.Unmarshal([]byte(unexpectedResponseError.Body), &pruneWorkerErr)
if err != nil {
return err
}
return pruneWorkerErr
}
}
return err
}
func (client *client) LandWorker(workerName string) error {
params := rata.Params{"worker_name": workerName}
err := client.connection.Send(internal.Request{
RequestName: atc.LandWorker,
Params: params,
Header: http.Header{
"Content-Type": {"application/json"},
},
}, nil)
return err
}