refactor: move version service scheduling to scheduler (#5120)

Unleash · Oct 23, 2023 · 828e463 · 828e463
1 parent 8c97847
commit 828e463
Show file tree

Hide file tree

Showing 14 changed files with 7 additions and 24 deletions.
diff --git a/src/lib/features/client-feature-toggles/tests/client-feature-toggle.e2e.test.ts b/src/lib/features/client-feature-toggles/tests/client-feature-toggle.e2e.test.ts
@@ -25,7 +25,6 @@ async function getSetup() {
         clientFeatureToggleStore: stores.clientFeatureToggleStore,
         request: supertest(app),
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/admin-api/config.test.ts b/src/lib/routes/admin-api/config.test.ts
@@ -29,7 +29,6 @@ async function getSetup() {
         base,
         request: supertest(app),
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/admin-api/context.test.ts b/src/lib/routes/admin-api/context.test.ts
@@ -21,7 +21,6 @@ async function getSetup() {
         base,
         request: supertest(app),
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/admin-api/metrics.test.ts b/src/lib/routes/admin-api/metrics.test.ts
@@ -20,7 +20,6 @@ async function getSetup() {
         perms,
         config,
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/admin-api/public-signup.test.ts b/src/lib/routes/admin-api/public-signup.test.ts
@@ -34,7 +34,6 @@ describe('Public Signup API', () => {
             stores,
             perms,
             destroy: () => {
-                services.versionService.destroy();
                 services.clientInstanceService.destroy();
                 services.publicSignupTokenService.destroy();
             },

diff --git a/src/lib/routes/admin-api/strategy.test.ts b/src/lib/routes/admin-api/strategy.test.ts
@@ -19,7 +19,6 @@ async function getSetup() {
     const app = await getApp(config, stores, services);
 
     destroy = () => {
-        services.versionService.destroy();
         services.clientInstanceService.destroy();
     };
 

diff --git a/src/lib/routes/admin-api/tag.test.ts b/src/lib/routes/admin-api/tag.test.ts
@@ -22,7 +22,6 @@ async function getSetup() {
         tagStore: stores.tagStore,
         request: supertest(app),
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/backstage.test.ts b/src/lib/routes/backstage.test.ts
@@ -19,6 +19,5 @@ test('should enable prometheus', async () => {
         .get('/internal-backstage/prometheus')
         .expect('Content-Type', /text/)
         .expect(200);
-    services.versionService.destroy();
     services.clientInstanceService.destroy();
 });
diff --git a/src/lib/routes/client-api/metrics.test.ts b/src/lib/routes/client-api/metrics.test.ts
@@ -1,5 +1,4 @@
 import supertest from 'supertest';
-import createStores from '../../../test/fixtures/store';
 import getApp from '../../app';
 import { createTestConfig } from '../../../test/config/test-config';
 import { clientMetricsSchema } from '../../services/client-metrics/schema';
@@ -21,7 +20,6 @@ async function getSetup(opts?: IUnleashOptions) {
         stores: db.stores,
         services,
         destroy: async () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
             await db.destroy();
         },

diff --git a/src/lib/routes/client-api/register.test.ts b/src/lib/routes/client-api/register.test.ts
@@ -15,7 +15,6 @@ async function getSetup() {
         request: supertest(app),
         stores,
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/health-check.test.ts b/src/lib/routes/health-check.test.ts
@@ -16,7 +16,6 @@ async function getSetup() {
         request: supertest(app),
         stores,
         destroy: () => {
-            services.versionService.destroy();
             services.clientInstanceService.destroy();
         },
     };

diff --git a/src/lib/routes/public-invite.test.ts b/src/lib/routes/public-invite.test.ts
@@ -40,7 +40,6 @@ describe('Public Signup API', () => {
             stores,
             perms,
             destroy: () => {
-                services.versionService.destroy();
                 services.clientInstanceService.destroy();
                 services.publicSignupTokenService.destroy();
             },

diff --git a/src/lib/services/index.ts b/src/lib/services/index.ts
@@ -116,6 +116,7 @@ export const scheduleServices = async (
         maintenanceService,
         eventAnnouncerService,
         featureToggleService,
+        versionService,
         lastSeenService,
     } = services;
 
@@ -198,6 +199,12 @@ export const scheduleServices = async (
         minutesToMilliseconds(1),
         'updatePotentiallyStaleFeatures',
     );
+
+    schedulerService.schedule(
+        versionService.checkLatestVersion.bind(versionService),
+        hoursToMilliseconds(48),
+        'checkLatestVersion',
+    );
 };
 
 export const createServices = (

diff --git a/src/lib/services/version-service.ts b/src/lib/services/version-service.ts
@@ -16,7 +16,6 @@ import { IUnleashConfig } from '../types/option';
 import version from '../util/version';
 import { Logger } from '../logger';
 import { ISettingStore } from '../types/stores/settings-store';
-import { hoursToMilliseconds } from 'date-fns';
 import { IStrategyStore } from 'lib/types';
 import { FEATURES_EXPORTED, FEATURES_IMPORTED } from '../types';
 import { CUSTOM_ROOT_ROLE_TYPE } from '../util';
@@ -191,11 +190,6 @@ export default class VersionService {
     async setup(): Promise<void> {
         await this.setInstanceId();
         await this.checkLatestVersion();
-        this.timer = setInterval(
-            async () => this.checkLatestVersion(),
-            hoursToMilliseconds(48),
-        );
-        this.timer.unref();
     }
 
     async setInstanceId(): Promise<void> {
@@ -365,11 +359,6 @@ export default class VersionService {
             instanceId: this.instanceId,
         };
     }
-
-    destroy(): void {
-        clearInterval(this.timer);
-        this.timer = null;
-    }
 }
 
 module.exports = VersionService;