Skip to content

Commit

Permalink
improve log messages and remove println
Browse files Browse the repository at this point in the history
  • Loading branch information
marjisound committed Oct 4, 2024
1 parent 9e20507 commit 610c558
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ class ExternalTranscriptionExtractor(index: Index, transcribeConfig: TranscribeC
transcriptionJob.flatMap {
job => {
try {
logger.info(s"sending message to ${transcribeConfig.transcriptionServiceQueueUrl}")
logger.info(s"sending message to Transcription Service Queue")

val sendMessageCommand = new SendMessageRequest()
.withQueueUrl(transcribeConfig.transcriptionServiceQueueUrl)
Expand Down
5 changes: 4 additions & 1 deletion backend/app/extraction/ExternalTranscriptionWorker.scala
Original file line number Diff line number Diff line change
Expand Up @@ -19,11 +19,14 @@ import scala.jdk.CollectionConverters.CollectionHasAsScala
class ExternalTranscriptionWorker(manifest: WorkerManifest, amazonSQSClient: AmazonSQS, transcribeConfig: TranscribeConfig, blobStorage: ObjectStorage, index: Index)(implicit executionContext: ExecutionContext) extends Logging{

def pollForResults(): Int = {
logger.info("Fetching messages from external transcription output queue")
val messages = amazonSQSClient.receiveMessage(
new ReceiveMessageRequest(transcribeConfig.transcriptionOutputQueueUrl).withMaxNumberOfMessages(10)
).getMessages

if (messages.size() > 0) logger.info(s"retrieved ${messages.size()} messages from queue ${transcribeConfig.transcriptionOutputQueueUrl}")
if (messages.size() > 0)
logger.info(s"retrieved ${messages.size()} messages from queue Transcription Output Queue")
else logger.info("No message found")

messages.asScala.toList.foldLeft(0) { (completed, message) =>
val result = for {
Expand Down
2 changes: 0 additions & 2 deletions backend/app/services/ExternalWorkerScheduler.scala
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,10 @@ class ExternalWorkerScheduler(actorSystem: ActorSystem, worker: ExternalTranscri

def go(): Unit = {
try {
println("running ExternalWorkerScheduler")
val completed = worker.pollForResults()
if (completed > 0) {
go()
} else {
println(s"try again ExternalWorkerScheduler in ${interval}")
tryAgainIn(interval)
}
} catch {
Expand Down

0 comments on commit 610c558

Please sign in to comment.