monitor.go

type Monitor struct {
  tasks map[string]*task
}

type task struct {
  *ecs.Task
  status schedulerStatus
}

func New() *Monitor {
  m := Monitor{tasks: map[string]task{}}
  go m.run()
  return &m
}

func (m *Monitor) run() {
  for range time.Tick(frequency) {
    m.tick()
  }
}

func (m *Monitor) tick() error {
  var taskIDs []string
  for taskID := range m.tasks {
    taskIDs = append(taskIDs, taskID)
  }
  
  resp := aws.DescribeTasks(taskIDs)
  for _, task := range resp.Tasks {
    if task.Done {
      m.tasks[task.ID] = getStatus(task.Containers[0])
    }
  }
}

func (m *Monitor) Watch(ecsTask *ecs.Task) {
  m.tasks[ecsTask.ID] = task{Task: ecsTask}
}

func (m *Monitor) GetStatus() map[string]*status {
  return m.tasks
}