mirror of
https://github.com/hashicorp/packer.git
synced 2026-04-21 14:20:40 -04:00
19055df3ec
* Updating the license from MPL to Business Source License Going forward, this project will be licensed under the Business Source License v1.1. Please see our blog post for more details at https://hashi.co/bsl-blog, FAQ at https://hashi.co/license-faq, and details of the license at www.hashicorp.com/bsl. * Update copyright file headers to BUSL-1.1 --------- Co-authored-by: hashicorp-copywrite[bot] <110428419+hashicorp-copywrite[bot]@users.noreply.github.com>
40 lines
837 B
Go
40 lines
837 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package command
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
|
|
)
|
|
|
|
func handleTermInterrupt(ui packersdk.Ui) (context.Context, func()) {
|
|
ctx, cancelCtx := context.WithCancel(context.Background())
|
|
// Handle interrupts for this build
|
|
sigCh := make(chan os.Signal, 1)
|
|
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
|
|
cleanup := func() {
|
|
cancelCtx()
|
|
signal.Stop(sigCh)
|
|
close(sigCh)
|
|
}
|
|
go func() {
|
|
select {
|
|
case sig := <-sigCh:
|
|
if sig == nil {
|
|
// context got cancelled and this closed chan probably
|
|
// triggered first
|
|
return
|
|
}
|
|
ui.Error(fmt.Sprintf("Cancelling build after receiving %s", sig))
|
|
cancelCtx()
|
|
case <-ctx.Done():
|
|
}
|
|
}()
|
|
return ctx, cleanup
|
|
}
|