forked from remind101/empire
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tasks.go
61 lines (51 loc) · 1.36 KB
/
tasks.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
package empire
import (
"fmt"
"time"
"github.com/remind101/empire/pkg/constraints"
"github.com/remind101/empire/scheduler"
"golang.org/x/net/context"
)
// Task represents a running process.
type Task struct {
Name string
Type string
Command string
State string
UpdatedAt time.Time
Constraints Constraints
}
type tasksService struct {
*Empire
}
func (s *tasksService) Tasks(ctx context.Context, app *App) ([]*Task, error) {
var tasks []*Task
instances, err := s.Scheduler.Instances(ctx, app.ID)
if err != nil {
return tasks, err
}
for _, i := range instances {
tasks = append(tasks, taskFromInstance(i))
}
return tasks, nil
}
// taskFromInstance converts a scheduler.Instance into a Task.
// It pulls some of its data from empire specific environment variables if they have been set.
// Once ECS supports this data natively, we can stop doing this.
func taskFromInstance(i *scheduler.Instance) *Task {
version := i.Process.Env["EMPIRE_RELEASE"]
if version == "" {
version = "v0"
}
return &Task{
Name: fmt.Sprintf("%s.%s.%s", version, i.Process.Type, i.ID),
Type: string(i.Process.Type),
Command: i.Process.Command,
Constraints: Constraints{
CPUShare: constraints.CPUShare(i.Process.CPUShares),
Memory: constraints.Memory(i.Process.MemoryLimit),
},
State: i.State,
UpdatedAt: i.UpdatedAt,
}
}