mirror of
https://github.com/samanhappy/mcphub.git
synced 2026-01-01 04:08:52 -05:00
Refactor service registration and revert lazy loading implementation (#234)
This commit is contained in:
@@ -17,14 +17,7 @@ const defaultConfig = {
|
||||
mcpHubVersion: getPackageVersion(),
|
||||
};
|
||||
|
||||
let dataService: DataService | null = null;
|
||||
|
||||
const getDataServiceInstance = (): DataService => {
|
||||
if (!dataService) {
|
||||
dataService = getDataService();
|
||||
}
|
||||
return dataService;
|
||||
};
|
||||
const dataService: DataService = getDataService();
|
||||
|
||||
// Settings cache
|
||||
let settingsCache: McpSettings | null = null;
|
||||
@@ -61,16 +54,13 @@ export const loadOriginalSettings = (): McpSettings => {
|
||||
};
|
||||
|
||||
export const loadSettings = (): McpSettings => {
|
||||
return getDataServiceInstance().filterSettings!(loadOriginalSettings());
|
||||
return dataService.filterSettings!(loadOriginalSettings());
|
||||
};
|
||||
|
||||
export const saveSettings = (settings: McpSettings): boolean => {
|
||||
const settingsPath = getSettingsPath();
|
||||
try {
|
||||
const mergedSettings = getDataServiceInstance().mergeSettings!(
|
||||
loadOriginalSettings(),
|
||||
settings,
|
||||
);
|
||||
const mergedSettings = dataService.mergeSettings!(loadOriginalSettings(), settings);
|
||||
fs.writeFileSync(settingsPath, JSON.stringify(mergedSettings, null, 2), 'utf8');
|
||||
|
||||
// Update cache after successful save
|
||||
@@ -88,7 +78,6 @@ export const saveSettings = (settings: McpSettings): boolean => {
|
||||
*/
|
||||
export const clearSettingsCache = (): void => {
|
||||
settingsCache = null;
|
||||
dataService = null; // Also clear the data service cache
|
||||
};
|
||||
|
||||
/**
|
||||
|
||||
@@ -13,7 +13,6 @@ import {
|
||||
} from './services/sseService.js';
|
||||
import { initializeDefaultUser } from './models/User.js';
|
||||
import { sseUserContextMiddleware } from './middlewares/userContext.js';
|
||||
import { initializeServices } from './services/services.js';
|
||||
|
||||
// Get the current working directory (will be project root in most cases)
|
||||
const currentFileDir = process.cwd() + '/src';
|
||||
@@ -32,9 +31,6 @@ export class AppServer {
|
||||
|
||||
async initialize(): Promise<void> {
|
||||
try {
|
||||
// Initialize services
|
||||
await initializeServices();
|
||||
|
||||
// Initialize default admin user if no users exist
|
||||
await initializeDefaultUser();
|
||||
|
||||
|
||||
@@ -1,3 +1,6 @@
|
||||
import { createRequire } from 'module';
|
||||
import { join } from 'path';
|
||||
|
||||
type Class<T> = new (...args: any[]) => T;
|
||||
|
||||
interface Service<T> {
|
||||
@@ -8,17 +11,19 @@ interface Service<T> {
|
||||
const registry = new Map<string, Service<any>>();
|
||||
const instances = new Map<string, unknown>();
|
||||
|
||||
export async function registerService<T>(key: string, entry: Service<T>) {
|
||||
export function registerService<T>(key: string, entry: Service<T>) {
|
||||
// Try to load override immediately during registration
|
||||
const overridePath = './' + key + 'x.js';
|
||||
await import(overridePath)
|
||||
.then((mod) => {
|
||||
const override = mod[key.charAt(0).toUpperCase() + key.slice(1) + 'x'];
|
||||
if (typeof override === 'function') {
|
||||
entry.override = override;
|
||||
}
|
||||
})
|
||||
.catch(() => {}); // Silently ignore if override doesn't exist
|
||||
const overridePath = join(process.cwd(), 'src', 'services', key + 'x.ts');
|
||||
try {
|
||||
const require = createRequire(process.cwd());
|
||||
const mod = require(overridePath);
|
||||
const override = mod[key.charAt(0).toUpperCase() + key.slice(1) + 'x'];
|
||||
if (typeof override === 'function') {
|
||||
entry.override = override;
|
||||
}
|
||||
} catch (error) {
|
||||
// Silently ignore if override doesn't exist
|
||||
}
|
||||
|
||||
registry.set(key, entry);
|
||||
}
|
||||
|
||||
@@ -1,31 +1,10 @@
|
||||
import { registerService, getService } from './registry.js';
|
||||
import { DataService, DataServiceImpl } from './dataService.js';
|
||||
|
||||
let initialized = false;
|
||||
let tempDataService: DataService | null = null;
|
||||
|
||||
async function initializeServices() {
|
||||
if (initialized) return;
|
||||
|
||||
await registerService('dataService', {
|
||||
defaultImpl: DataServiceImpl,
|
||||
});
|
||||
|
||||
initialized = true;
|
||||
tempDataService = null; // Clear temp service once real one is ready
|
||||
}
|
||||
registerService('dataService', {
|
||||
defaultImpl: DataServiceImpl,
|
||||
});
|
||||
|
||||
export function getDataService(): DataService {
|
||||
if (initialized) {
|
||||
return getService<DataService>('dataService');
|
||||
}
|
||||
|
||||
// Return temporary service for cases where services haven't been initialized yet
|
||||
// This allows module loading to work even before server initialization
|
||||
if (!tempDataService) {
|
||||
tempDataService = new DataServiceImpl();
|
||||
}
|
||||
return tempDataService;
|
||||
return getService<DataService>('dataService');
|
||||
}
|
||||
|
||||
export { initializeServices };
|
||||
|
||||
Reference in New Issue
Block a user