浏览代码

Fixed incorrect microservices URLs after updating dockerfiles (#159)

Alex 3 年之前
父节点
当前提交
c5adbea6e1

+ 1 - 1
microservices/src/image-classifier/image-classifier.controller.ts

@@ -5,7 +5,7 @@ import { ImageClassifierService } from './image-classifier.service';
 export class ImageClassifierController {
   constructor(
     private readonly imageClassifierService: ImageClassifierService,
-  ) {}
+  ) { }
 
   @Post('/tagImage')
   async tagImage(@Body('thumbnailPath') thumbnailPath: string) {

+ 2 - 1
microservices/src/object-detection/object-detection.controller.ts

@@ -1,11 +1,12 @@
 import { Body, Controller, Post } from '@nestjs/common';
 import { ObjectDetectionService } from './object-detection.service';
+import { Logger } from '@nestjs/common';
 
 @Controller('object-detection')
 export class ObjectDetectionController {
   constructor(
     private readonly objectDetectionService: ObjectDetectionService,
-  ) {}
+  ) { }
 
   @Post('/detectObject')
   async detectObject(@Body('thumbnailPath') thumbnailPath: string) {

+ 17 - 12
server/src/modules/background-task/background-task.processor.ts

@@ -115,7 +115,7 @@ export class BackgroundTaskProcessor {
   async tagImage(job) {
     const { thumbnailPath, asset }: { thumbnailPath: string; asset: AssetEntity } = job.data;
 
-    const res = await axios.post('http://immich_microservices:3001/image-classifier/tagImage', {
+    const res = await axios.post('http://immich-microservices:3001/image-classifier/tagImage', {
       thumbnailPath: thumbnailPath,
     });
 
@@ -132,19 +132,24 @@ export class BackgroundTaskProcessor {
 
   @Process('detect-object')
   async detectObject(job) {
-    const { thumbnailPath, asset }: { thumbnailPath: string; asset: AssetEntity } = job.data;
-
-    const res = await axios.post('http://immich_microservices:3001/object-detection/detectObject', {
-      thumbnailPath: thumbnailPath,
-    });
+    try {
+      const { thumbnailPath, asset }: { thumbnailPath: string; asset: AssetEntity } = job.data;
 
-    if (res.status == 201 && res.data.length > 0) {
-      const smartInfo = new SmartInfoEntity();
-      smartInfo.assetId = asset.id;
-      smartInfo.objects = [...res.data];
-      await this.smartInfoRepository.upsert(smartInfo, {
-        conflictPaths: ['assetId'],
+      const res = await axios.post('http://immich-microservices:3001/object-detection/detectObject', {
+        thumbnailPath: thumbnailPath,
       });
+
+      if (res.status == 201 && res.data.length > 0) {
+        const smartInfo = new SmartInfoEntity();
+        smartInfo.assetId = asset.id;
+        smartInfo.objects = [...res.data];
+        await this.smartInfoRepository.upsert(smartInfo, {
+          conflictPaths: ['assetId'],
+        });
+      }
+    } catch (error) {
+      Logger.error(`Failed to trigger object detection pipe line ${error.toString()}`)
     }
+
   }
 }

+ 1 - 1
server/src/modules/background-task/background-task.service.ts

@@ -9,7 +9,7 @@ export class BackgroundTaskService {
   constructor(
     @InjectQueue('background-task')
     private backgroundTaskQueue: Queue,
-  ) {}
+  ) { }
 
   async extractExif(savedAsset: AssetEntity, fileName: string, fileSize: number) {
     await this.backgroundTaskQueue.add(