This commit is contained in:
Brage 2023-07-25 00:55:43 +02:00
parent 412e2ef0b7
commit 8723b73c6e
3 changed files with 17 additions and 2 deletions

View File

@ -0,0 +1,13 @@
package no.iktdev.streamit.content.common.deserializers
import no.iktdev.streamit.content.common.dto.reader.work.ConvertWork
import no.iktdev.streamit.content.common.dto.reader.work.EncodeWork
import no.iktdev.streamit.content.common.dto.reader.work.ExtractWork
import no.iktdev.streamit.library.kafka.dto.Message
import no.iktdev.streamit.library.kafka.listener.deserializer.IMessageDataDeserialization
class ConvertWorkDeserializer: IMessageDataDeserialization<ConvertWork> {
override fun deserialize(incomingMessage: Message): ConvertWork? {
return incomingMessage.dataAs(ConvertWork::class.java)
}
}

View File

@ -16,6 +16,7 @@ class DeserializerRegistry {
KafkaEvents.EVENT_ENCODER_ENDED_VIDEO_FILE to EncodeWorkDeserializer(),
KafkaEvents.EVENT_READER_ENCODE_GENERATED_SUBTITLE to ExtractWorkDeserializer(),
KafkaEvents.EVENT_ENCODER_ENDED_SUBTITLE_FILE to ExtractWorkDeserializer(),
KafkaEvents.EVENT_CONVERTER_ENDED_SUBTITLE_FILE to ConvertWorkDeserializer()
)
fun getRegistry(): Map<KafkaEvents, IMessageDataDeserialization<*>> = _registry.toMap()

View File

@ -3,6 +3,7 @@ package no.iktdev.streamit.content.reader.collector
import no.iktdev.streamit.content.common.CommonConfig
import no.iktdev.streamit.content.common.DefaultKafkaReader
import no.iktdev.streamit.content.common.deserializers.DeserializerRegistry
import no.iktdev.streamit.content.common.dto.reader.work.ConvertWork
import no.iktdev.streamit.content.common.dto.reader.work.ExtractWork
import no.iktdev.streamit.library.db.query.SubtitleQuery
import no.iktdev.streamit.library.kafka.KafkaEvents
@ -22,7 +23,7 @@ class ConvertedSubtitleConsumer : DefaultKafkaReader("collectorConsumerConverted
accepts = listOf(KafkaEvents.EVENT_CONVERTER_ENDED_SUBTITLE_FILE.event)
) {
override fun onMessageReceived(data: ConsumerRecord<String, Message>) {
val workResult = data.value().dataAs(ExtractWork::class.java)
val workResult = data.value().dataAs(ConvertWork::class.java)
if (!data.value().isSuccessful() || workResult == null) {
return
}
@ -42,6 +43,6 @@ class ConvertedSubtitleConsumer : DefaultKafkaReader("collectorConsumerConverted
}
override fun loadDeserializers(): Map<String, IMessageDataDeserialization<*>> {
return DeserializerRegistry.getEventToDeserializer(KafkaEvents.EVENT_ENCODER_ENDED_SUBTITLE_FILE)
return DeserializerRegistry.getEventToDeserializer(KafkaEvents.EVENT_CONVERTER_ENDED_SUBTITLE_FILE)
}
}