From 90b4e5debb1f59fd23f7307875dd0d25170d95b6 Mon Sep 17 00:00:00 2001 From: jiegec Date: Tue, 13 Oct 2020 22:49:49 +0800 Subject: [PATCH] Fix DeleteWorker behavior to match tests --- manager/db.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/manager/db.go b/manager/db.go index d254513..1a6472d 100644 --- a/manager/db.go +++ b/manager/db.go @@ -121,6 +121,10 @@ func (b *kvDBAdapter) GetWorker(workerID string) (w WorkerStatus, err error) { } func (b *kvDBAdapter) DeleteWorker(workerID string) error { + v, _ := b.db.Get(_workerBucketKey, workerID) + if v == nil { + return fmt.Errorf("invalid workerID %s", workerID) + } return b.db.Delete(_workerBucketKey, workerID) }