Browse Source

Fix DeleteWorker behavior to match tests

jiegec 5 years ago
parent
commit
90b4e5debb
1 changed files with 4 additions and 0 deletions
  1. 4 0
      manager/db.go

+ 4 - 0
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)
 }