-
Notifications
You must be signed in to change notification settings - Fork 0
/
staff_group.go
66 lines (55 loc) · 1.7 KB
/
staff_group.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
package pkg
import (
mapset "github.com/deckarep/golang-set/v2"
"github.com/pkg/errors"
)
type StaffGroup struct {
Name string
SubGroups []StaffSubGroup
}
func (sg *StaffGroup) PopulateMemberData(members map[uint]Member, assigned *mapset.Set[uint]) (err error) {
var subgroups []StaffSubGroup
for _, group := range sg.SubGroups {
err = group.PopulateMemberData(members, assigned)
if err != nil {
// TODO: Instead of halting on error, continue to populate and return a slice of errors
err = errors.WithStack(err)
return err
}
subgroups = append(subgroups, group)
}
sg.SubGroups = subgroups
return nil
}
type StaffSubGroup struct {
Name string
Leader DutyAssignment
DirectReports []DutyAssignment
}
func (ssg *StaffSubGroup) PopulateMemberData(members map[uint]Member, assigned *mapset.Set[uint]) (err error) {
if ssg.Leader.Assignee != nil {
leader, ok := members[ssg.Leader.Assignee.CAPID]
if !ok {
// TODO: Instead of halting on error, continue to populate and return a slice of errors
err = errors.Errorf("no member found with CAPID %d", ssg.Leader.Assignee.CAPID)
return err
}
ssg.Leader.Assignee = &leader
(*assigned).Add(leader.CAPID)
}
var directReports []DutyAssignment
for _, report := range ssg.DirectReports {
if report.Assignee != nil {
member, ok := members[report.Assignee.CAPID]
if !ok {
// TODO: Instead of halting on error, continue to populate and return a slice of errors
return errors.Errorf("no member found with CAPID %d", ssg.Leader.Assignee.CAPID)
}
report.Assignee = &member
(*assigned).Add(member.CAPID)
directReports = append(directReports, report)
}
}
ssg.DirectReports = directReports
return nil
}