Files
goodbuddy/src/main/assistant-storage-startup.ts
T
mesalogo d80d42b032 feat(storage): move desktop SQLite ownership into a storage utility process
Main no longer opens desktop business SQLite. A single utility-process
storage owner holds the Assistant, Knowledge, attachment, document-result,
remote-binding and native model-ledger databases and runs whole domain
transactions; Main and Runtime callers use explicit async domain operations.

- Remove the Main readWithFallback path and the supervision worker's
  separate business write connection; manifests are scanned read-only and
  committed by the sole writer.
- Remote events are ACKed only after commit; process loss reports
  unconfirmed writes instead of replaying them.
- Use native utilityProcess messaging with one in-flight bound (32) and
  ordered waiting instead of capacity rejection and retries.
- Start the storage process before the window so startup overlaps.
- Keep short tool output in memory and store full output once in owned
  storage; paging, branching, restart and deletion reuse that copy.
- Put required runtime launch files under goodbuddy-prefixed owned temp
  directories and reclaim confirmed stale ones; stop writing duplicate
  parsed.md.
- Add lint guards for SQLite ownership and architecture, concurrency
  acceptance fixtures, and maintenance/architecture documentation.

Validation: typecheck and lint pass. Alternating baseline (edf019aa) vs
candidate runs show no regression on single-session, C01, C02, C04 and a
5-minute C06; Main p99 drops from ~21 ms to ~16.7 ms under concurrency.
Fresh portable copy (1.75M events) upgrade content unchanged; real-model
output paging passes with 4 external requests. Full npm test runs had
1-9 different timing-sensitive failures per run that pass when rerun.
Not covered: C03 review publication, C07 source body, app-level C05.
2026-10-06 15:40:18 +08:00

74 lines
2.7 KiB
TypeScript

import { app, ipcMain, type BrowserWindow } from 'electron'
import {
assistantStorageActionSchema,
assistantStorageProgressSchema,
type AssistantStorageProgress
} from '../shared/assistant-storage-contracts'
import { ipcChannels } from '../shared/ipc-channels'
import type { DesktopStorageClient } from './desktop-storage-client'
import { assertTrustedSender } from './trusted-ipc-sender'
import { loadMainWindow } from './window'
export async function prepareAssistantStorage(
window: BrowserWindow,
createStorage: (publish: (progress: AssistantStorageProgress) => void) => Pick<DesktopStorageClient, 'ready' | 'retry' | 'close'>,
signal: AbortSignal
): Promise<void> {
if (signal.aborted) return
let progress: AssistantStorageProgress = {
stage: 'upgrading', processed: 0, total: 0, bytesBefore: 0
}
let shown = false
let retry: (() => void) | undefined
let abort!: () => void
const aborted = new Promise<void>(resolve => { abort = resolve })
signal.addEventListener('abort', abort, { once: true })
ipcMain.handle(ipcChannels.storageUpgradeProgress, (event) => {
assertTrustedSender(event, window)
return progress
})
ipcMain.handle(ipcChannels.storageUpgradeAction, (event, input: unknown) => {
assertTrustedSender(event, window)
const action = assistantStorageActionSchema.parse(input)
if (action === 'quit') app.quit()
else if (progress.stage === 'failed') retry?.()
})
const publish = (next: AssistantStorageProgress): void => {
progress = assistantStorageProgressSchema.parse(next)
if (window.isDestroyed()) return
if (!shown) {
shown = true
loadMainWindow(window, true)
}
window.setProgressBar(next.stage === 'failed' ? -1 :
next.total > 0 && next.stage === 'converting'
? next.processed / next.total : 2)
}
try {
const storage = createStorage(publish)
let readiness = storage.ready
while (!signal.aborted) {
try {
await Promise.race([readiness, aborted])
if (signal.aborted) break
return
} catch (error) {
if (signal.aborted) break
publish({
...progress, stage: 'failed',
error: error instanceof Error ? error.message : '本地数据升级失败'
})
await Promise.race([new Promise<void>(resolve => { retry = resolve }), aborted])
retry = undefined
if (!signal.aborted) readiness = storage.retry()
}
}
await storage.close()
} finally {
signal.removeEventListener('abort', abort)
ipcMain.removeHandler(ipcChannels.storageUpgradeProgress)
ipcMain.removeHandler(ipcChannels.storageUpgradeAction)
if (!window.isDestroyed()) window.setProgressBar(-1)
}
}