iktdev/MediaProcessing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Wrapped in transactions

Browse Source
This commit is contained in:
Brage 2023-07-25 01:22:58 +02:00
parent dc8d5c44e4
commit 0b9e08c22a
3 changed files with 47 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Convert/src/main/kotlin/no/iktdev/streamit/content/convert/ConvertRunner.kt
View File

@ -34,6 +34,7 @@ class ConvertRunner(val referenceId: String, val listener: IConvertListener) {
val syncedDialogs = Syncro().sync(dialogs)
try {
val converted = Export(inFile, syncedDialogs).write()
converted.forEach {
val item = ConvertWork(
@ -46,6 +47,12 @@ class ConvertRunner(val referenceId: String, val listener: IConvertListener) {
listener.onEnded(referenceId, subtitleInfo, work = item)
}
}
} catch (e: Exception) {
e.printStackTrace()
withContext(Dispatchers.Default) {
listener.onError(referenceId, subtitleInfo, "See log")
}
}
}

18
Reader/src/main/kotlin/no/iktdev/streamit/content/reader/collector/EncodedVideoConsumer.kt
View File

@ -14,6 +14,7 @@ import no.iktdev.streamit.library.kafka.dto.StatusType
import no.iktdev.streamit.library.kafka.listener.collector.CollectorMessageListener
import no.iktdev.streamit.library.kafka.listener.collector.ICollectedMessagesEvent
import no.iktdev.streamit.library.kafka.listener.deserializer.IMessageDataDeserialization
import org.jetbrains.exposed.sql.transactions.transaction
import org.springframework.stereotype.Service
import java.io.File
@ -66,10 +67,15 @@ class EncodedVideoConsumer: DefaultKafkaReader("collectorConsumerEncodedVideo"),
val contentType = metadata?.type ?: return
val iid = if (contentType == "movie") MovieQuery(videoFileNameWithExtension).insertAndGetId() else null
val iid = if (contentType == "movie") transaction {
MovieQuery(videoFileNameWithExtension).insertAndGetId()
} else null
if (serieData != null) {
val success = SerieQuery(serieData.title, serieData.episode, serieData.season, fileData.title, videoFileNameWithExtension).insertAndGetStatus()
val success = transaction {
SerieQuery(serieData.title, serieData.episode, serieData.season, fileData.title, videoFileNameWithExtension)
.insertAndGetStatus()
}
if (!success)
return
}
@ -86,8 +92,10 @@ class EncodedVideoConsumer: DefaultKafkaReader("collectorConsumerEncodedVideo"),
}
val metaGenre = metadata.genres
val gq = GenreQuery(*metaGenre.toTypedArray())
transaction {
gq.insertAndGetIds()
val gids = gq.getIds().joinToString(",")
}
val gids = transaction { gq.getIds().joinToString(",") }
val cq = CatalogQuery(
title = fileData.title,
@ -97,15 +105,17 @@ class EncodedVideoConsumer: DefaultKafkaReader("collectorConsumerEncodedVideo"),
iid = iid,
genres = gids
)
val cid = cq.insertAndGetId() ?: cq.getId() ?: return
val cid = transaction { cq.insertAndGetId() ?: cq.getId() } ?: return
if (!metadata.summary.isNullOrBlank()) {
val summary = metadata.summary ?: return
transaction {
SummaryQuery(
cid = cid,
language = "eng", // TODO: Fix later,
description = summary
)
}
}
val message = Message(referenceId = collection.getReferenceId() ?: "M.I.A", status = Status(StatusType.SUCCESS))
produceMessage(KafkaEvents.EVENT_COLLECTOR_VIDEO_STORED, message, null)

3
Reader/src/main/kotlin/no/iktdev/streamit/content/reader/collector/ExtractedSubtitleConsumer.kt
View File

@ -10,6 +10,7 @@ import no.iktdev.streamit.library.kafka.dto.Message
import no.iktdev.streamit.library.kafka.listener.SimpleMessageListener
import no.iktdev.streamit.library.kafka.listener.deserializer.IMessageDataDeserialization
import org.apache.kafka.clients.consumer.ConsumerRecord
import org.jetbrains.exposed.sql.transactions.transaction
import java.io.File
class ExtractedSubtitleConsumer : DefaultKafkaReader("collectorConsumerExtractedSubtitle") {
@ -26,6 +27,7 @@ class ExtractedSubtitleConsumer : DefaultKafkaReader("collectorConsumerExtracted
}
val of = File(workResult.outFile)
transaction {
SubtitleQuery(
title = of.nameWithoutExtension,
language = workResult.language,
@ -34,6 +36,7 @@ class ExtractedSubtitleConsumer : DefaultKafkaReader("collectorConsumerExtracted
).insertAndGetStatus()
}
}
}
init {
listener.listen()