|
| 1 | +import { Effect, Layer } from "effect" |
| 2 | +import { Fff } from "#fff" |
| 3 | +import { AppRuntime } from "@/effect/app-runtime" |
| 4 | +import { Search } from "@/file/search" |
| 5 | +import { bootstrap } from "@/cli/bootstrap" |
| 6 | + |
| 7 | +const dir = process.cwd() |
| 8 | + |
| 9 | +const FILE_QUERIES = ["fff", "package.json", "tools/ experiment"] |
| 10 | +const GREP_QUERIES = ["FileFinder", "import", "grep", "autocomplete"] |
| 11 | +const GLOB_QUERIES = ["**/*.test.ts"] |
| 12 | + |
| 13 | +const FILE_LIMIT = 100 |
| 14 | +const GREP_LIMIT = 50 |
| 15 | +const GLOB_LIMIT = 50 |
| 16 | + |
| 17 | +const search = <A>(effect: Effect.Effect<A, unknown, Search.Service>) => AppRuntime.runPromise(effect as never) as Promise<A> |
| 18 | + |
| 19 | +await bootstrap(dir, async () => { |
| 20 | + // --- raw Fff picker --- |
| 21 | + const t0 = performance.now() |
| 22 | + const made = Fff.create({ basePath: dir, aiMode: true }) |
| 23 | + if (!made.ok) { |
| 24 | + console.error("Fff.create failed:", made.error) |
| 25 | + process.exit(1) |
| 26 | + } |
| 27 | + const picker = made.value |
| 28 | + console.log(`picker create: ${(performance.now() - t0).toFixed(1)}ms`) |
| 29 | + |
| 30 | + const tw = performance.now() |
| 31 | + const deadline = tw + 2500 |
| 32 | + while (picker.isScanning() && performance.now() < deadline) { |
| 33 | + await new Promise((resolve) => setTimeout(resolve, 25)) |
| 34 | + } |
| 35 | + console.log(`wait for scan (poll): ${(performance.now() - tw).toFixed(1)}ms`) |
| 36 | + |
| 37 | + // warmup grep to let the content index build |
| 38 | + const tWarmup = performance.now() |
| 39 | + picker.grep("_warmup_", { mode: "regex", maxMatchesPerFile: 1, timeBudgetMs: 1_500 }) |
| 40 | + console.log(`grep warmup: ${(performance.now() - tWarmup).toFixed(1)}ms`) |
| 41 | + |
| 42 | + console.log() |
| 43 | + console.log("--- raw picker (warm) ---") |
| 44 | + |
| 45 | + for (const q of FILE_QUERIES) { |
| 46 | + const t = performance.now() |
| 47 | + const r = picker.fileSearch(q, { pageSize: Math.max(FILE_LIMIT, 100) }) |
| 48 | + const count = r.ok ? r.value.items.length : "err" |
| 49 | + console.log(`[picker] fileSearch "${q}": ${(performance.now() - t).toFixed(1)}ms (${count} results)`) |
| 50 | + } |
| 51 | + |
| 52 | + for (const q of GREP_QUERIES) { |
| 53 | + const t = performance.now() |
| 54 | + const r = picker.grep(q, { mode: "regex", pageSize: GREP_LIMIT, timeBudgetMs: 1_500 }) |
| 55 | + const count = r.ok ? r.value.items.length : "err" |
| 56 | + console.log(`[picker] grep "${q}": ${(performance.now() - t).toFixed(1)}ms (${count} matches)`) |
| 57 | + } |
| 58 | + |
| 59 | + picker.destroy() |
| 60 | + |
| 61 | + // --- Ripgrep service (via Search with file:["."] to force rg path) --- |
| 62 | + console.log() |
| 63 | + console.log("--- Ripgrep (via Search service) ---") |
| 64 | + |
| 65 | + // warmup |
| 66 | + await search(Search.Service.use((svc) => svc.search({ cwd: dir, pattern: "_warmup_rg_", limit: 1, file: ["."] }))) |
| 67 | + |
| 68 | + for (const q of GREP_QUERIES) { |
| 69 | + const t = performance.now() |
| 70 | + const r = await search(Search.Service.use((svc) => svc.search({ cwd: dir, pattern: q, limit: GREP_LIMIT, file: ["."] }))) |
| 71 | + console.log( |
| 72 | + `[ripgrep] grep "${q}": ${(performance.now() - t).toFixed(1)}ms (${r.items.length} total, limit is per-file not total)`, |
| 73 | + ) |
| 74 | + } |
| 75 | + |
| 76 | + // --- Search service: init breakdown --- |
| 77 | + console.log() |
| 78 | + |
| 79 | + // 1) runtime + InstanceState + picker create + scan poll |
| 80 | + const tRuntime = performance.now() |
| 81 | + await search(Search.Service.use((svc) => svc.file({ cwd: dir, query: "_warmup_file_", limit: 1 }))) |
| 82 | + console.log(`[Search] init file (runtime + picker + scan): ${(performance.now() - tRuntime).toFixed(1)}ms`) |
| 83 | + |
| 84 | + // 2) grep warmup (content index cold-start inside the Search service picker) |
| 85 | + const tGrepWarmup = performance.now() |
| 86 | + await search(Search.Service.use((svc) => svc.search({ cwd: dir, pattern: "_warmup_grep_", limit: 1 }))) |
| 87 | + console.log(`[Search] init grep (content index warmup): ${(performance.now() - tGrepWarmup).toFixed(1)}ms`) |
| 88 | + |
| 89 | + console.log() |
| 90 | + console.log("--- Search service (warm) ---") |
| 91 | + |
| 92 | + for (const q of FILE_QUERIES) { |
| 93 | + const t = performance.now() |
| 94 | + const r = await search(Search.Service.use((svc) => svc.file({ cwd: dir, query: q, limit: FILE_LIMIT }))) |
| 95 | + console.log( |
| 96 | + `[Search.file] "${q}": ${(performance.now() - t).toFixed(1)}ms (${r?.length ?? "undefined (cache fallback)"} results)`, |
| 97 | + ) |
| 98 | + } |
| 99 | + |
| 100 | + for (const q of GREP_QUERIES) { |
| 101 | + const t = performance.now() |
| 102 | + const r = await search(Search.Service.use((svc) => svc.search({ cwd: dir, pattern: q, limit: GREP_LIMIT }))) |
| 103 | + console.log( |
| 104 | + `[Search.search] "${q}": ${(performance.now() - t).toFixed(1)}ms (${r.items.length} matches, engine=${r.engine})`, |
| 105 | + ) |
| 106 | + } |
| 107 | + |
| 108 | + for (const q of GLOB_QUERIES) { |
| 109 | + const t = performance.now() |
| 110 | + const r = await search(Search.Service.use((svc) => svc.glob({ cwd: dir, pattern: q, limit: GLOB_LIMIT }))) |
| 111 | + console.log( |
| 112 | + `[Search.glob] "${q}": ${(performance.now() - t).toFixed(1)}ms (${r.files.length} files, truncated=${r.truncated})`, |
| 113 | + ) |
| 114 | + } |
| 115 | +}) |
| 116 | + |
| 117 | +process.exit(0) |
| 118 | + |
0 commit comments