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

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

@ -34,17 +34,24 @@ class ConvertRunner(val referenceId: String, val listener: IConvertListener) {
val syncedDialogs = Syncro().sync(dialogs) val syncedDialogs = Syncro().sync(dialogs)
val converted = Export(inFile, syncedDialogs).write() try {
converted.forEach { val converted = Export(inFile, syncedDialogs).write()
val item = ConvertWork( converted.forEach {
inFile = inFile.absolutePath, val item = ConvertWork(
collection = subtitleInfo.collection, inFile = inFile.absolutePath,
language = subtitleInfo.language, collection = subtitleInfo.collection,
outFile = it.absolutePath language = subtitleInfo.language,
) outFile = it.absolutePath
withContext(Dispatchers.Default) { )
listener.onEnded(referenceId, subtitleInfo, work = item) withContext(Dispatchers.Default) {
} listener.onEnded(referenceId, subtitleInfo, work = item)
}
}
} catch (e: Exception) {
e.printStackTrace()
withContext(Dispatchers.Default) {
listener.onError(referenceId, subtitleInfo, "See log")
}
} }
} }

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

15
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.SimpleMessageListener
import no.iktdev.streamit.library.kafka.listener.deserializer.IMessageDataDeserialization import no.iktdev.streamit.library.kafka.listener.deserializer.IMessageDataDeserialization
import org.apache.kafka.clients.consumer.ConsumerRecord import org.apache.kafka.clients.consumer.ConsumerRecord
import org.jetbrains.exposed.sql.transactions.transaction
import java.io.File import java.io.File
class ExtractedSubtitleConsumer : DefaultKafkaReader("collectorConsumerExtractedSubtitle") { class ExtractedSubtitleConsumer : DefaultKafkaReader("collectorConsumerExtractedSubtitle") {
@ -26,12 +27,14 @@ class ExtractedSubtitleConsumer : DefaultKafkaReader("collectorConsumerExtracted
} }
val of = File(workResult.outFile) val of = File(workResult.outFile)
SubtitleQuery( transaction {
title = of.nameWithoutExtension, SubtitleQuery(
language = workResult.language, title = of.nameWithoutExtension,
collection = workResult.collection, language = workResult.language,
format = of.extension.uppercase() collection = workResult.collection,
).insertAndGetStatus() format = of.extension.uppercase()
).insertAndGetStatus()
}
} }
} }