| | | 1 | | using System; |
| | | 2 | | using System.Collections.Generic; |
| | | 3 | | using Akka.Actor; |
| | | 4 | | using SeungYongShim.Kafka; |
| | | 5 | | |
| | | 6 | | namespace SeungYongShim.Akka.OpenTelemetry.Kafka |
| | | 7 | | { |
| | | 8 | | public class KafkaConsumerActor : ReceiveActor |
| | | 9 | | { |
| | 1 | 10 | | public KafkaConsumerActor(KafkaConsumer kafkaConsumer, |
| | 1 | 11 | | IList<string> topics, |
| | 1 | 12 | | string groupId, |
| | 1 | 13 | | IActorRef parserActor) |
| | 1 | 14 | | { |
| | 1 | 15 | | var self = Context.Self; |
| | 1 | 16 | | KafkaConsumer = kafkaConsumer; |
| | | 17 | | |
| | 2 | 18 | | KafkaConsumer.Run(groupId, topics, m => self.Tell(m)); |
| | | 19 | | |
| | 1 | 20 | | ReceiveAsync<Commitable>(async msg => |
| | 2 | 21 | | { |
| | 2 | 22 | | await parserActor.Ask<Commit>(msg.Body); |
| | 1 | 23 | | msg.Commit(); |
| | 1 | 24 | | }); |
| | 1 | 25 | | } |
| | | 26 | | |
| | | 27 | | private KafkaConsumer KafkaConsumer { get; } |
| | | 28 | | |
| | | 29 | | protected override void PostStop() |
| | 1 | 30 | | { |
| | 1 | 31 | | KafkaConsumer.Stop(); |
| | 1 | 32 | | KafkaConsumer.Dispose(); |
| | 1 | 33 | | base.PostStop(); |
| | 1 | 34 | | } |
| | | 35 | | |
| | | 36 | | protected override void PreRestart(Exception reason, object message) |
| | 0 | 37 | | { |
| | 0 | 38 | | KafkaConsumer.Stop(); |
| | 0 | 39 | | KafkaConsumer.Dispose(); |
| | 0 | 40 | | base.PreRestart(reason, message); |
| | 0 | 41 | | } |
| | | 42 | | |
| | 0 | 43 | | public record Commit(); |
| | | 44 | | } |
| | | 45 | | } |