mirror/api
1
0
Fork 0
mirror of https://github.com/unraid/api.git synced 2025-12-31 21:49:57 -06:00
Code Issues Packages Projects Releases Wiki Activity

fix: retry VMs init for up to 2 min (#1612)

Browse Source
This commit is contained in:
Pujit Mehrotra
2025-08-22 15:29:44 -04:00
committed by GitHub
parent fd895cacf0
commit b2e7801238
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/src/unraid-api/graph/resolvers/vms/vms.service.spec.ts
View File

@@ -196,7 +196,7 @@ describe('VmsService', () => {
service = module.get<VmsService>(VmsService); service = module.get<VmsService>(VmsService);
// Initialize the service // Initialize the service
await service.onModuleInit(); await service.onApplicationBootstrap();
}); });
afterAll(async () => { afterAll(async () => {

22
api/src/unraid-api/graph/resolvers/vms/vms.service.ts
View File

@@ -1,4 +1,5 @@
import { Injectable, Logger, OnModuleDestroy, OnModuleInit } from '@nestjs/common'; import { Injectable, Logger, OnApplicationBootstrap, OnModuleDestroy } from '@nestjs/common';
import { Timeout } from '@nestjs/schedule';
import { constants } from 'fs'; import { constants } from 'fs';
import { access } from 'fs/promises'; import { access } from 'fs/promises';
@@ -11,7 +12,7 @@ import { getters } from '@app/store/index.js';
import { VmDomain, VmState } from '@app/unraid-api/graph/resolvers/vms/vms.model.js'; import { VmDomain, VmState } from '@app/unraid-api/graph/resolvers/vms/vms.model.js';
@Injectable() @Injectable()
export class VmsService implements OnModuleInit, OnModuleDestroy { export class VmsService implements OnApplicationBootstrap, OnModuleDestroy {
private readonly logger = new Logger(VmsService.name); private readonly logger = new Logger(VmsService.name);
private hypervisor: InstanceType<typeof HypervisorClass> | null = null; private hypervisor: InstanceType<typeof HypervisorClass> | null = null;
private isVmsAvailable: boolean = false; private isVmsAvailable: boolean = false;
@@ -38,11 +39,26 @@ export class VmsService implements OnModuleInit, OnModuleDestroy {
} }
} }
async onModuleInit() { async onApplicationBootstrap() {
this.logger.debug(`Initializing VMs service with URI: ${this.uri}`); this.logger.debug(`Initializing VMs service with URI: ${this.uri}`);
await this.attemptHypervisorInitializationAndWatch(); await this.attemptHypervisorInitializationAndWatch();
} }
@Timeout(10_000)
async healInitialization(maxRetries = 12, delay = 10_000) {
let retries = 1;
while (!this.isVmsAvailable && retries <= maxRetries) {
this.logger.log(`Attempting to initialize VMs service...attempt ${retries}/${maxRetries}`);
await this.attemptHypervisorInitializationAndWatch();
if (this.isVmsAvailable) {
this.logger.log('VMs service initialized successfully');
break;
}
await new Promise((resolve) => setTimeout(resolve, delay));
retries++;
}
}
async onModuleDestroy() { async onModuleDestroy() {
this.logger.debug('Closing file watcher...'); this.logger.debug('Closing file watcher...');
await this.watcher?.close(); await this.watcher?.close();