From 4d2ab8e6dd645d345fa26d8a067ad6dc14fc1fce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Paul=20B=C3=BCtow?= Date: Sat, 22 Feb 2020 10:34:19 +0000 Subject: serverhandler understands background jobs better --- internal/server/background/background.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'internal/server/background') diff --git a/internal/server/background/background.go b/internal/server/background/background.go index d31c1f2..537ccbb 100644 --- a/internal/server/background/background.go +++ b/internal/server/background/background.go @@ -2,6 +2,7 @@ package background import ( "context" + "fmt" "sync" "github.com/mimecast/dtail/internal/io/logger" @@ -23,12 +24,18 @@ func NewBackground() *Background { } } -func (b Background) Add(name string, cancel context.CancelFunc, done <-chan struct{}) { +func (b Background) Add(name string, cancel context.CancelFunc, done <-chan struct{}) error { b.mutex.Lock() defer b.mutex.Unlock() + if _, ok := b.jobs[name]; ok { + return fmt.Errorf("job '%s' already exists", name) + } + logger.Debug("background", name, "adding job") b.jobs[name] = job{cancel, done} + + return nil } func (b Background) get(name string) (job, bool) { -- cgit v1.2.3