graphvent/manager.go

package main

import (
  "fmt"
  "errors"
)

type EventManager struct {
  dag_nodes map[string]Resource
  root_event Event
  aborts []chan error
}

// root_event's requirements must be in dag_nodes, and dag_nodes must be ordered by dependency(children first)
func NewEventManager(root_event Event, dag_nodes []Resource) * EventManager {

  manager := &EventManager{
    dag_nodes: map[string]Resource{},
    root_event: nil,
    aborts: []chan error{},
  }

  // Construct the DAG
  for _, resource := range dag_nodes {
    err := manager.AddResource(resource)
    if err != nil {
      log.Logf("manager", "Failed to add %s to EventManager: %s", resource.Name(), err)
      return nil
    }
  }

  err := manager.AddEvent(nil, root_event, nil)
  if err != nil {
    log.Logf("manager", "Failed to add %s to EventManager as root_event: %s", root_event.Name(), err)
  }

  return manager;
}

// Init to all resources(in a thread to handle reconnections), and start the first event
func (manager * EventManager) Run() error {
  log.Logf("manager", "MANAGER_START")

  abort := make(chan error, 1)
  go func(abort chan error, manager * EventManager) {
    <- abort
    for _, c := range(manager.aborts) {
      c <- nil
    }
  }(abort, manager)

  err := LockResources(manager.root_event)
  if err != nil {
    log.Logf("manager", "MANAGER_LOCK_ERR: %s", err)
    abort <- nil
    return err
  }

  err = RunEvent(manager.root_event)
  abort <- nil
  if err != nil {
    log.Logf("manager", "MANAGER_RUN_ERR: %s", err)
    return err
  }

  err = FinishEvent(manager.root_event)
  if err != nil {
    log.Logf("manager", "MANAGER_FINISH_ERR: %s", err)
    return err
  }
  log.Logf("manager", "MANAGER_DONE")

  return nil
}

func (manager * EventManager) FindResource(id string) Resource {
  resource, exists := manager.dag_nodes[id]
  if exists == false {
    return nil
  }

  return resource
}

func (manager * EventManager) FindEvent(id string) Event {
  event := FindChild(manager.root_event, id)

  return event
}

func (manager * EventManager) AddResource(resource Resource) error {
  log.Logf("manager", "Adding resource %s", resource.Name())
  _, exists := manager.dag_nodes[resource.ID()]
  if exists == true {
    error_str := fmt.Sprintf("%s is already in the resource DAG, cannot add again", resource.Name())
    return errors.New(error_str)
  }

  for _, child := range resource.Children() {
    _, exists := manager.dag_nodes[child.ID()]
    if exists == false {
      error_str := fmt.Sprintf("%s is not in the resource DAG, cannot add %s to DAG", child.Name(), resource.Name())
      return errors.New(error_str)
    }
  }
  manager.dag_nodes[resource.ID()] = resource
  abort := make(chan error, 1)
  abort_used := resource.Init(abort)
  if abort_used == true {
    manager.aborts = append(manager.aborts, abort)
  }
  for _, child := range resource.Children() {
    AddParent(child, resource)
  }
  return nil
}

// Check that the node doesn't already exist in the tree
// Check the the selected parent exists in the tree
// Check that required resources exist in the DAG
// Check that created resources don't exist in the DAG
// Add resources created by the event to the DAG
// Add child to parent
func (manager * EventManager) CheckResources(event Event) error {
  if event == nil {
    return errors.New("Cannot check nil event for resources")
  }

  for _, r := range(event.RequiredResources()) {
    res_found := false
    for _, res := range(manager.dag_nodes) {
      if res.ID() == r.ID() {
        res_found = true
      }
    }
    if res_found == false {
      return errors.New(fmt.Sprintf("Failed to find %s in the resource forest for %s", r.Name(), event.Name()))
    }
  }

  for _, c := range(event.Children()) {
    err := manager.CheckResources(c)
    if err != nil {
      return err
    }
  }

  return nil
}

func (manager * EventManager) AddDoneResources(event Event) {
  if event == nil {
    return
  }

  done_resource := event.DoneResource()
  _, exists := manager.dag_nodes[done_resource.ID()]
  if exists == false {
    manager.AddResource(done_resource)
  }

  for _, child := range(event.Children()) {
    manager.AddDoneResources(child)
  }
}

func (manager * EventManager) AddEvent(parent Event, child Event, info EventInfo) error {
  if child == nil {
    return errors.New("Cannot add nil Event to EventManager")
  }

  err := manager.CheckResources(child)
  if err != nil {
    return fmt.Errorf("Failed to add event to event manager: %w", err)
  }

  manager.AddDoneResources(child)

  if manager.root_event == nil {
    if parent != nil {
      return fmt.Errorf("EventManager has no root, so can't add event to parent")
    } else {
      manager.root_event = child
      return nil
    }
  } else {
    if parent == nil {
      return fmt.Errorf("Replacing root event not implemented")
    } else if FindChild(manager.root_event, parent.ID()) == nil {
      return fmt.Errorf("Parent does not exists in event tree")
    } else if FindChild(manager.root_event, child.ID()) != nil {
      return fmt.Errorf("Child already exists in event tree")
    } else {
      AddChild(parent, child, info)
    }
  }
      return nil
}