refactor: improve logging messages in MCPService and clean up useMCPServers hook
- Updated logging messages in MCPService for clarity and consistency. - Removed unused dispatch and initial server loading logic from useMCPServers hook to streamline the code.
This commit is contained in:
parent
555c5baafa
commit
b441d76991
@ -75,11 +75,11 @@ export default class MCPService extends EventEmitter {
|
|||||||
|
|
||||||
this.initPromise = (async () => {
|
this.initPromise = (async () => {
|
||||||
try {
|
try {
|
||||||
|
log.info('[MCP] Starting initialization')
|
||||||
|
|
||||||
// Wait for servers to be loaded from Redux
|
// Wait for servers to be loaded from Redux
|
||||||
await this.waitForServers()
|
await this.waitForServers()
|
||||||
|
|
||||||
log.info('[MCP] Starting initialization')
|
|
||||||
|
|
||||||
// Load SDK components in parallel for better performance
|
// Load SDK components in parallel for better performance
|
||||||
const [Client, StdioTransport, SSETransport] = await Promise.all([
|
const [Client, StdioTransport, SSETransport] = await Promise.all([
|
||||||
this.importClient(),
|
this.importClient(),
|
||||||
@ -96,7 +96,7 @@ export default class MCPService extends EventEmitter {
|
|||||||
|
|
||||||
// Load active servers
|
// Load active servers
|
||||||
await this.loadActiveServers()
|
await this.loadActiveServers()
|
||||||
log.info('[MCP] Initialization completed successfully')
|
log.info('[MCP] Initialization successfully')
|
||||||
|
|
||||||
return
|
return
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
@ -333,7 +333,7 @@ export default class MCPService extends EventEmitter {
|
|||||||
this.clients[name] = client
|
this.clients[name] = client
|
||||||
this.activeServers.set(name, { client, server })
|
this.activeServers.set(name, { client, server })
|
||||||
|
|
||||||
log.info(`[MCP] Server ${name} started successfully`)
|
log.info(`[MCP] Activated server: ${server.name}`)
|
||||||
this.emit('server-started', { name })
|
this.emit('server-started', { name })
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
log.error(`[MCP] Error activating server ${name}:`, error)
|
log.error(`[MCP] Error activating server ${name}:`, error)
|
||||||
@ -476,14 +476,13 @@ export default class MCPService extends EventEmitter {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
log.info(`[MCP] Loading ${activeServers.length} active servers`)
|
log.info(`[MCP] Start loading ${activeServers.length} active servers`)
|
||||||
|
|
||||||
// Activate servers in parallel for better performance
|
// Activate servers in parallel for better performance
|
||||||
await Promise.allSettled(
|
await Promise.allSettled(
|
||||||
activeServers.map(async (server) => {
|
activeServers.map(async (server) => {
|
||||||
try {
|
try {
|
||||||
await this.activate(server)
|
await this.activate(server)
|
||||||
log.info(`[MCP] Successfully activated server: ${server.name}`)
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
this.logError(`Failed to activate server ${server.name}`)
|
this.logError(`Failed to activate server ${server.name}`)
|
||||||
this.emit('server-error', { name: server.name, error })
|
this.emit('server-error', { name: server.name, error })
|
||||||
@ -491,7 +490,7 @@ export default class MCPService extends EventEmitter {
|
|||||||
})
|
})
|
||||||
)
|
)
|
||||||
|
|
||||||
log.info(`[MCP] Loaded and activated ${Object.keys(this.clients).length} servers`)
|
log.info(`[MCP] End loading ${Object.keys(this.clients).length} active servers`)
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
@ -1,4 +1,4 @@
|
|||||||
import store, { useAppDispatch, useAppSelector } from '@renderer/store'
|
import store, { useAppSelector } from '@renderer/store'
|
||||||
import { setMCPServers } from '@renderer/store/mcp'
|
import { setMCPServers } from '@renderer/store/mcp'
|
||||||
import { MCPServer } from '@renderer/types'
|
import { MCPServer } from '@renderer/types'
|
||||||
import { useEffect } from 'react'
|
import { useEffect } from 'react'
|
||||||
@ -12,21 +12,6 @@ ipcRenderer.on('mcp:servers-changed', (_event, servers) => {
|
|||||||
|
|
||||||
export const useMCPServers = () => {
|
export const useMCPServers = () => {
|
||||||
const mcpServers = useAppSelector((state) => state.mcp.servers)
|
const mcpServers = useAppSelector((state) => state.mcp.servers)
|
||||||
const dispatch = useAppDispatch()
|
|
||||||
|
|
||||||
// Initial load of MCP servers from main process
|
|
||||||
useEffect(() => {
|
|
||||||
const loadServers = async () => {
|
|
||||||
try {
|
|
||||||
const servers = await window.api.mcp.listServers()
|
|
||||||
dispatch(setMCPServers(servers))
|
|
||||||
} catch (error) {
|
|
||||||
console.error('Failed to load MCP servers:', error)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
loadServers()
|
|
||||||
}, [dispatch])
|
|
||||||
|
|
||||||
const addMCPServer = async (server: MCPServer) => {
|
const addMCPServer = async (server: MCPServer) => {
|
||||||
try {
|
try {
|
||||||
@ -84,9 +69,24 @@ export const useMCPServers = () => {
|
|||||||
|
|
||||||
export const useInitMCPServers = () => {
|
export const useInitMCPServers = () => {
|
||||||
const mcpServers = useAppSelector((state) => state.mcp.servers)
|
const mcpServers = useAppSelector((state) => state.mcp.servers)
|
||||||
|
// const dispatch = useAppDispatch()
|
||||||
|
|
||||||
// Send servers to main process when they change in Redux
|
// Send servers to main process when they change in Redux
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
ipcRenderer.send('mcp:servers-from-renderer', mcpServers)
|
ipcRenderer.send('mcp:servers-from-renderer', mcpServers)
|
||||||
}, [mcpServers])
|
}, [mcpServers])
|
||||||
|
|
||||||
|
// Initial load of MCP servers from main process
|
||||||
|
// useEffect(() => {
|
||||||
|
// const loadServers = async () => {
|
||||||
|
// try {
|
||||||
|
// const servers = await window.api.mcp.listServers()
|
||||||
|
// dispatch(setMCPServers(servers))
|
||||||
|
// } catch (error) {
|
||||||
|
// console.error('Failed to load MCP servers:', error)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// loadServers()
|
||||||
|
// }, [dispatch])
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user