packer-cn/common/step_provision.go

64 lines
1.3 KiB
Go
Raw Normal View History

2013-07-16 02:43:01 -04:00
package common
import (
"context"
2013-07-16 02:43:01 -04:00
"log"
2013-08-31 02:28:31 -04:00
"time"
2018-01-22 20:21:10 -05:00
"github.com/hashicorp/packer/helper/multistep"
2018-01-22 20:21:10 -05:00
"github.com/hashicorp/packer/packer"
2013-07-16 02:43:01 -04:00
)
// StepProvision runs the provisioners.
//
// Uses:
// communicator packer.Communicator
// hook packer.Hook
// ui packer.Ui
//
// Produces:
// <nothing>
type StepProvision struct {
Comm packer.Communicator
}
func (s *StepProvision) Run(_ context.Context, state multistep.StateBag) multistep.StepAction {
comm := s.Comm
if comm == nil {
raw, ok := state.Get("communicator").(packer.Communicator)
if ok {
comm = raw.(packer.Communicator)
}
}
2013-08-31 15:17:59 -04:00
hook := state.Get("hook").(packer.Hook)
ui := state.Get("ui").(packer.Ui)
2013-07-16 02:43:01 -04:00
2013-08-31 02:28:31 -04:00
// Run the provisioner in a goroutine so we can continually check
// for cancellations...
2013-07-16 02:43:01 -04:00
log.Println("Running the provision hook")
2013-08-31 02:28:31 -04:00
errCh := make(chan error, 1)
go func() {
errCh <- hook.Run(packer.HookProvision, ui, comm, nil)
2013-08-31 02:28:31 -04:00
}()
for {
select {
case err := <-errCh:
if err != nil {
2013-08-31 15:17:59 -04:00
state.Put("error", err)
2013-08-31 02:28:31 -04:00
return multistep.ActionHalt
}
2013-07-16 02:43:01 -04:00
2013-08-31 02:28:31 -04:00
return multistep.ActionContinue
case <-time.After(1 * time.Second):
2013-08-31 15:17:59 -04:00
if _, ok := state.GetOk(multistep.StateCancelled); ok {
log.Println("Cancelling provisioning due to interrupt...")
2013-08-31 02:28:31 -04:00
hook.Cancel()
return multistep.ActionHalt
}
}
}
2013-07-16 02:43:01 -04:00
}
2013-08-31 15:17:59 -04:00
func (*StepProvision) Cleanup(multistep.StateBag) {}