packer-cn/packer/plugin/builder.go

47 lines
794 B
Go
Raw Normal View History

package plugin
import (
2017-04-04 16:39:01 -04:00
"github.com/hashicorp/packer/packer"
"log"
)
type cmdBuilder struct {
builder packer.Builder
client *Client
}
2013-11-02 23:51:26 -04:00
func (b *cmdBuilder) Prepare(config ...interface{}) ([]string, error) {
defer func() {
r := recover()
b.checkExit(r, nil)
}()
return b.builder.Prepare(config...)
}
func (b *cmdBuilder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packer.Artifact, error) {
defer func() {
r := recover()
b.checkExit(r, nil)
}()
2013-06-10 01:00:47 -04:00
return b.builder.Run(ui, hook, cache)
}
func (b *cmdBuilder) Cancel() {
defer func() {
r := recover()
b.checkExit(r, nil)
}()
b.builder.Cancel()
}
func (c *cmdBuilder) checkExit(p interface{}, cb func()) {
if c.client.Exited() && cb != nil {
cb()
} else if p != nil && !Killed {
log.Panic(p)
}
}