Fix segment too large to send issue

Author: juffalow

src/config.ts

@@ -260,6 +260,6 @@ export default {
        * @requires SERVICES_TRACE_TYPE=AWS_XRAY
        */
       plugins: process.env.SERVICES_TRACE_PLUGINS,
-    }
+    },
   },
 }

src/logger/index.ts

@@ -1,5 +1,5 @@
 import winston, { format } from 'winston';
-import services from '../services'
+import services from '../services';
 import config from '../config';
 
 /**

src/worker.ts

@@ -8,36 +8,45 @@ import repositories from './repositories';
 const app = Consumer.create({
   queueUrl: config.services.queue.url,
   handleMessage: async (message) => {
-    logger.debug('Received message from queue!', { ...message });
+    const namespace: any = services.Trace.getNamespace();
+    const segment: any = services.Trace.createSegment('upload-service-worker');
 
-    const body = JSON.parse(message.Body);
+    await namespace.runPromise(async () => {
+      services.Trace.setSegment(segment);
 
-    if ('traceId' in body) {
-      namespace.set('traceId', body.traceId);
-    } else {
-      logger.warn('Trace ID is not present in message body!');
-    }
+      logger.debug('Received message from queue!', { ...message });
 
-    if ('fileId' in body) {
-      try {
-        const file = await repositories.File.get(body.fileId, null);
-        const values = await services.Job.process(body.mimetype, file) as Record<string, string | number | boolean>;
-        await repositories.File.update({ ...values, cacheable: !body.mimetype.startsWith('video') }, { id: file.id });
-      } catch (err) {
-        logger.error('Could not process file!', { error: err.message, stack: err.stack });
-      }
-    }
+      const body = JSON.parse(message.Body);
 
-    if ('detail' in body && 'jobId' in body.detail) {
-      if (body.detail.status === 'COMPLETE') {
-        const updatedFile = await video.AWSVideo.complete(body) as Record<string, string | number | boolean>;
-        await repositories.File.update({ ...updatedFile, cacheable: true }, { id: updatedFile['id'] as string });
+      if ('traceId' in body) {
+        namespace.set('traceId', body.traceId);
+      } else {
+        logger.warn('Trace ID is not present in message body!');
       }
-
-      if (body.detail.status === 'ERROR') {
-        logger.error('Media convert job ended with error!', body);
+  
+      if ('fileId' in body) {
+        try {
+          const file = await repositories.File.get(body.fileId, null);
+          const values = await services.Job.process(body.mimetype, file) as Record<string, string | number | boolean>;
+          await repositories.File.update({ ...values, cacheable: !body.mimetype.startsWith('video') }, { id: file.id });
+        } catch (err) {
+          logger.error('Could not process file!', { error: err.message, stack: err.stack });
+        }
+      }
+  
+      if ('detail' in body && 'jobId' in body.detail) {
+        if (body.detail.status === 'COMPLETE') {
+          const updatedFile = await video.AWSVideo.complete(body) as Record<string, string | number | boolean>;
+          await repositories.File.update({ ...updatedFile, cacheable: true }, { id: updatedFile['id'] as string });
+        }
+  
+        if (body.detail.status === 'ERROR') {
+          logger.error('Media convert job ended with error!', body);
+        }
       }
-    }
+
+      segment.close();
+    });
   },
   sqs: services.AWS.sqs,
 });
@@ -52,21 +61,14 @@ app.on('error', (err) => {
 
 app.on('processing_error', (err) => {
   logger.error('Error while processing message from queue!', { message: err.message, stack: err.stack });
+
+  services.Trace.closeSegment();
 });
 
 app.on('timeout_error', (err) => {
   logger.error('Timeout error!', { message: err.message, stack: err.stack });
 });
 
-const namespace: any = services.Trace.getNamespace();
-const segment: any = services.Trace.createSegment('upload-service-worker');
+logger.info('Upload worker is running');
 
-namespace.run(() => {
-  services.Trace.setSegment(segment);
-
-  logger.info('Upload worker is running');
-
-  app.start();
-
-  segment.close();
-});
+app.start();