mirror of
https://github.com/kemko/nomad.git
synced 2026-01-03 17:05:43 +03:00
36 lines
854 B
Go
36 lines
854 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package allocrunner
|
|
|
|
import (
|
|
"context"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/nomad/client/config"
|
|
)
|
|
|
|
// upstreamAllocsHook waits for a PrevAllocWatcher to exit before allowing
|
|
// an allocation to be executed
|
|
type upstreamAllocsHook struct {
|
|
allocWatcher config.PrevAllocWatcher
|
|
logger log.Logger
|
|
}
|
|
|
|
func newUpstreamAllocsHook(logger log.Logger, allocWatcher config.PrevAllocWatcher) *upstreamAllocsHook {
|
|
h := &upstreamAllocsHook{
|
|
allocWatcher: allocWatcher,
|
|
}
|
|
h.logger = logger.Named(h.Name())
|
|
return h
|
|
}
|
|
|
|
func (h *upstreamAllocsHook) Name() string {
|
|
return "await_previous_allocations"
|
|
}
|
|
|
|
func (h *upstreamAllocsHook) Prerun() error {
|
|
// Wait for a previous alloc - if any - to terminate
|
|
return h.allocWatcher.Wait(context.Background())
|
|
}
|