mirror of
https://github.com/safing/portbase
synced 2025-09-01 18:19:57 +00:00
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
// Copyright Safing ICS Technologies GmbH. Use of this source code is governed by the AGPL license that can be found in the LICENSE file.
|
|
|
|
package modules
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/tevino/abool"
|
|
)
|
|
|
|
var (
|
|
startComplete = abool.NewBool(false)
|
|
|
|
modulesLock sync.Mutex
|
|
modules = make(map[string]*Module)
|
|
modulesOrder []*Module
|
|
|
|
// ErrCleanExit is returned by Start() when the program is interrupted before starting. This can happen for example, when using the "--help" flag.
|
|
ErrCleanExit = errors.New("clean exit requested")
|
|
)
|
|
|
|
// Module represents a module.
|
|
type Module struct {
|
|
Name string
|
|
Active *abool.AtomicBool
|
|
inTransition bool
|
|
|
|
prep func() error
|
|
start func() error
|
|
stop func() error
|
|
|
|
dependencies []string
|
|
}
|
|
|
|
// Register registers a new module.
|
|
func Register(name string, prep, start, stop func() error, dependencies ...string) *Module {
|
|
newModule := &Module{
|
|
Name: name,
|
|
Active: abool.NewBool(false),
|
|
prep: prep,
|
|
start: start,
|
|
stop: stop,
|
|
dependencies: dependencies,
|
|
}
|
|
modulesLock.Lock()
|
|
defer modulesLock.Unlock()
|
|
modulesOrder = append(modulesOrder, newModule)
|
|
modules[name] = newModule
|
|
return newModule
|
|
}
|