mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
764 B
Go
49 lines
764 B
Go
package bootstrap
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type BootstrapConf func(*Bootstrap)
|
|
|
|
func WithContext(ctx context.Context) BootstrapConf {
|
|
return func(b *Bootstrap) {
|
|
b.ctx = ctx
|
|
}
|
|
}
|
|
|
|
func WithTask(f ...BootstrapFunc) BootstrapConf {
|
|
return func(b *Bootstrap) {
|
|
b.task = append(b.task, f...)
|
|
}
|
|
}
|
|
|
|
type Bootstrap struct {
|
|
task []BootstrapFunc
|
|
ctx context.Context
|
|
}
|
|
|
|
func New(conf ...BootstrapConf) *Bootstrap {
|
|
b := &Bootstrap{}
|
|
for _, c := range conf {
|
|
c(b)
|
|
}
|
|
return b
|
|
}
|
|
|
|
type BootstrapFunc func(context.Context) error
|
|
|
|
func (b *Bootstrap) Add(f ...BootstrapFunc) *Bootstrap {
|
|
b.task = append(b.task, f...)
|
|
return b
|
|
}
|
|
|
|
func (b *Bootstrap) Run() error {
|
|
for _, f := range b.task {
|
|
if err := f(b.ctx); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|