fix mailhog/MailHog-UI#10 - delete for in-memory storage

FIXME fix is nasty, especially with lots of messages, or deleting using concurrent API requests
mailhog · Oct 6, 2015 · 33a9b88 · 33a9b88
1 parent 153c814
commit 33a9b88
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/memory.go b/memory.go
@@ -158,6 +158,11 @@ func (memory *InMemory) List(start int, limit int) (*data.Messages, error) {
 func (memory *InMemory) DeleteOne(id string) error {
 	index := memory.MessageIDIndex[id]
 	delete(memory.MessageIDIndex, id)
+	for k, v := range memory.MessageIDIndex {
+		if v > index {
+			memory.MessageIDIndex[k] = v - 1
+		}
+	}
 	memory.Messages = append(memory.Messages[:index], memory.Messages[index+1:]...)
 	return nil
 }