Просмотр исходного кода

Return an error if the topic/partition is already consumed.

Willem van Bergen 11 лет назад
Родитель
Сommit
29c4ebd656
1 измененных файлов с 4 добавлено и 0 удалено
  1. 4 0
      mocks/consumer.go

+ 4 - 0
mocks/consumer.go

@@ -47,6 +47,10 @@ func (c *Consumer) ConsumePartition(topic string, partition int32, offset int64)
 	}
 
 	pc := c.partitionConsumers[topic][partition]
+	if pc.consumed {
+		return nil, sarama.ConfigurationError("The topic/partition is already being consumed")
+	}
+
 	pc.consumed = true
 	go pc.handleExpectations()
 	return pc, nil