/
multiresponse.go
113 lines (99 loc) · 2.75 KB
/
multiresponse.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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
* === This file is part of ALICE O² ===
*
* Copyright 2018 CERN and copyright holders of ALICE O².
* Author: Teo Mrnjavac <teo.mrnjavac@cern.ch>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* In applying this license CERN does not waive the privileges and
* immunities granted to it by virtue of its status as an
* Intergovernmental Organization or submit itself to any jurisdiction.
*/
package controlcommands
import (
"errors"
"fmt"
"strings"
)
type MesosCommandMultiResponse struct {
MesosCommandResponseBase
responses map[MesosCommandTarget]MesosCommandResponse
}
func (m *MesosCommandMultiResponse) GetResponses() map[MesosCommandTarget]MesosCommandResponse {
if m == nil {
return nil
}
return m.responses
}
func (m *MesosCommandMultiResponse) IsMultiResponse() bool {
return true
}
func (m *MesosCommandMultiResponse) GetResponseSenders() []MesosCommandTarget {
if m != nil {
senders := make([]MesosCommandTarget, len(m.responses))
i := 0
for k, _ := range m.responses {
senders[i] = k
i++
}
return senders
}
return nil
}
func (m *MesosCommandMultiResponse) Err() error {
if m == nil {
return errors.New("nil response")
}
errs := make(map[MesosCommandTarget]error, 0)
for k, v := range m.responses {
if v.Err() != nil {
errs[k] = v.Err()
}
}
return errors.New(strings.Join(func() (out []string) {
for k, v := range errs {
if v != nil && len(strings.TrimSpace(v.Error())) != 0 {
out = append(out, fmt.Sprintf("[task %s] %s", k.TaskId.Value, v.Error()))
}
}
return
}(), "\n"))
}
func (m *MesosCommandMultiResponse) Errors() map[MesosCommandTarget]error {
errMap := make(map[MesosCommandTarget]error)
if m != nil {
for k, v := range m.responses {
if v.Err() != nil {
errMap[k] = v.Err()
}
}
return errMap
}
return nil
}
func consolidateResponses(command MesosCommand, responses map[MesosCommandTarget]MesosCommandResponse) MesosCommandResponse {
if len(responses) == 0 {
return nil
}
if len(responses) == 1 {
for _, v := range responses {
return v
}
}
return &MesosCommandMultiResponse{
MesosCommandResponseBase: *NewMesosCommandResponse(command, nil),
responses: responses,
}
}