diff options
Diffstat (limited to 'vendor/codeberg.org/gruf/go-runners/service.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-runners/service.go | 29 |
1 files changed, 28 insertions, 1 deletions
diff --git a/vendor/codeberg.org/gruf/go-runners/service.go b/vendor/codeberg.org/gruf/go-runners/service.go index c0f878c45..68e8ea384 100644 --- a/vendor/codeberg.org/gruf/go-runners/service.go +++ b/vendor/codeberg.org/gruf/go-runners/service.go @@ -15,7 +15,7 @@ type Service struct { mu sync.Mutex // mu protects state changes } -// Run will run the supplied function until completion, use given context to propagate cancel. +// Run will run the supplied function until completion, using given context to propagate cancel. // Immediately returns false if the Service is already running, and true after completed run. func (svc *Service) Run(fn func(context.Context)) bool { // Attempt to start the svc @@ -39,6 +39,33 @@ func (svc *Service) Run(fn func(context.Context)) bool { return true } +// GoRun will run the supplied function until completion in a goroutine, using given context to +// propagate cancel. Immediately returns boolean indicating success, or that service is already running. +func (svc *Service) GoRun(fn func(context.Context)) bool { + // Attempt to start the svc + ctx, ok := svc.doStart() + if !ok { + return false + } + + go func() { + defer func() { + // unlock single wait + svc.wait.Unlock() + + // ensure stopped + svc.Stop() + }() + + // Run user func + if fn != nil { + fn(ctx) + } + }() + + return true +} + // Stop will attempt to stop the service, cancelling the running function's context. Immediately // returns false if not running, and true only after Service is fully stopped. func (svc *Service) Stop() bool { |