func (p *dockerProvisioner) SetUnitStatus(unit provision.Unit, status provision.Status) error { container, err := p.GetContainer(unit.Name) if err != nil { return err } if unit.AppName != "" && container.AppName != unit.AppName { return errors.New("wrong app name") } err = container.SetStatus(p, status.String(), true) if err != nil { return err } return p.checkContainer(container) }
func (p *dockerProvisioner) RegisterUnit(unit provision.Unit, customData map[string]interface{}) error { container, err := p.GetContainer(unit.Name) if err != nil { return err } if container.Status == provision.StatusBuilding.String() { if container.BuildingImage != "" && customData != nil { return saveImageCustomData(container.BuildingImage, customData) } return nil } err = container.SetStatus(p, provision.StatusStarted.String(), true) if err != nil { return err } return p.checkContainer(container) }