micropelago/pmux
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Introduce StartAfterFunc

Browse Source
This commit is contained in:
Brian Picciano 2022-10-20 20:52:52 +02:00
parent d0696182e1
commit 30159d80b2
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pmuxlib/proc.go
View File

@ -48,6 +48,11 @@ type ProcessConfig struct {
// NoRestartOn indicates which exit codes should result in the process not // NoRestartOn indicates which exit codes should result in the process not
// being restarted any further. // being restarted any further.
NoRestartOn []int `yaml:"noRestartOn"` NoRestartOn []int `yaml:"noRestartOn"`
// StartAfterFunc will cause the starting of the process to be blocked until
// this function returns. If used with RunProcessOnce and an error is
// returned, RunProcessOnce will return that error.
StartAfterFunc func(context.Context) error
} }
func (cfg ProcessConfig) withDefaults() ProcessConfig { func (cfg ProcessConfig) withDefaults() ProcessConfig {
@ -104,6 +109,12 @@ func RunProcessOnce(
cfg = cfg.withDefaults() cfg = cfg.withDefaults()
if cfg.StartAfterFunc != nil {
if err := cfg.StartAfterFunc(ctx); err != nil {
return -1, err
}
}
var wg sync.WaitGroup var wg sync.WaitGroup
fwdOutPipe := func(logger Logger, r io.Reader) { fwdOutPipe := func(logger Logger, r io.Reader) {