refactor: 后端从 Django 换成 Bun + SQLite
Django 多个 worker 内存占用 1-2 GB，换成 Bun 单进程后约 25 MB， 同时去掉 Postgres 和 Redis，只剩 Caddy + Bun 两个容器。 - server/：Hono + bun:sqlite + Bun 原生 WebSocket，接口与原 django-ninja 保持兼容；表结构、时间/UUID 存储格式、密码哈希与 Django 一致，旧数据可直接使用 - 评分加权、贝叶斯平滑、成绩册的计算结果与原实现逐位一致（复刻 Python round/sum） - scripts/：Postgres → SQLite 迁移、新建超管、提示词层级标注、重算分数 - 前端：Django Admin 去掉后，用户新建/修改改为页面内弹窗； 非超管的挑战公开按钮置灰 - 权限不足返回 JSON 403（原来是重定向到不存在的登录页） - docker-compose / Caddyfile / dev.sh / CLAUDE.md 更新，删除 AGENTS.md - 服务器切换步骤见 docs/migrate-to-bun.md Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
xuyueandClaude Opus 5.5 committed
1 parent
47d4679984
commit
26f05d1d76
297 files changed
+4659
-42956
No files matched your search
@@ -0,0 +1,225 @@
|
||||
import type { Server, ServerWebSocket, WebSocketHandler } from "bun"
|
||||
import { config } from "./config"
|
||||
import { all, run, iso, uuidOut } from "./db"
|
||||
import { parseCookie, userFromSessionKey, type User } from "./auth"
|
||||
import { canAccessTask, getOrCreateActiveConversation, insertMessage } from "./services/prompt"
|
||||
import {
|
||||
extractCode,
|
||||
parseGuidanceResponse,
|
||||
streamChat,
|
||||
streamGuidance,
|
||||
type ChatMessage,
|
||||
} from "./services/llm"
|
||||
import { isAllowedOrigin } from "./origin"
|
||||
|
||||
type Kind = "prompt" | "guidance"
|
||||
|
||||
export interface WsData {
|
||||
kind: Kind
|
||||
user: User
|
||||
taskId: number
|
||||
conversationId: string
|
||||
// 正在等待 AI 回复的用户消息；对话中断时删掉，避免留下没有回复的提问
|
||||
pendingUserMessageId: number | null
|
||||
abort: AbortController
|
||||
closed: boolean
|
||||
// 串行处理消息（与 Django Channels 的行为一致）
|
||||
queue: Promise<void>
|
||||
// 引导对话只在本次连接内保留上下文
|
||||
guidanceHistory: ChatMessage[]
|
||||
}
|
||||
|
||||
type WS = ServerWebSocket<WsData>
|
||||
|
||||
const PATH_RE = /^\/ws\/(prompt|guidance)\/(\d+)\/?$/
|
||||
|
||||
// 处理 WebSocket 升级请求；返回 undefined 表示已升级，返回 Response 表示拒绝或不是 ws 路径
|
||||
export function upgrade(req: Request, server: Server<WsData>): Response | undefined | null {
|
||||
const url = new URL(req.url)
|
||||
const match = PATH_RE.exec(url.pathname)
|
||||
if (!match) return null
|
||||
if (!isAllowedOrigin(req.headers.get("origin"), req.headers.get("host"))) {
|
||||
return new Response("Forbidden", { status: 403 })
|
||||
}
|
||||
const user = userFromSessionKey(parseCookie(req.headers.get("cookie"), config.sessionCookie))
|
||||
const taskId = Number(match[2])
|
||||
if (!user || !canAccessTask(user, taskId)) return new Response("Forbidden", { status: 403 })
|
||||
|
||||
const ok = server.upgrade(req, {
|
||||
data: {
|
||||
kind: match[1] as Kind,
|
||||
user,
|
||||
taskId,
|
||||
conversationId: "",
|
||||
pendingUserMessageId: null,
|
||||
abort: new AbortController(),
|
||||
closed: false,
|
||||
queue: Promise.resolve(),
|
||||
guidanceHistory: [],
|
||||
},
|
||||
})
|
||||
return ok ? undefined : new Response("Upgrade failed", { status: 400 })
|
||||
}
|
||||
|
||||
function send(ws: WS, data: unknown) {
|
||||
if (!ws.data.closed) ws.send(JSON.stringify(data))
|
||||
}
|
||||
|
||||
function dropPendingUserMessage(ws: WS) {
|
||||
if (ws.data.pendingUserMessageId !== null) {
|
||||
run(`DELETE FROM prompt_message WHERE id = $id`, { id: ws.data.pendingUserMessageId })
|
||||
ws.data.pendingUserMessageId = null
|
||||
}
|
||||
}
|
||||
|
||||
function conversationHistory(conversationId: string) {
|
||||
return all<{
|
||||
id: number
|
||||
role: string
|
||||
content: string
|
||||
code_html: string | null
|
||||
code_css: string | null
|
||||
code_js: string | null
|
||||
submission_id: string | null
|
||||
created: string
|
||||
}>(
|
||||
`SELECT id, role, content, code_html, code_css, code_js, submission_id, created
|
||||
FROM prompt_message WHERE conversation_id = $id AND source = 'conversation'
|
||||
ORDER BY created, id`,
|
||||
{ id: conversationId },
|
||||
)
|
||||
}
|
||||
|
||||
async function streamToClient(ws: WS, chunks: AsyncIterable<string>) {
|
||||
let full = ""
|
||||
for await (const chunk of chunks) {
|
||||
full += chunk
|
||||
send(ws, { type: "stream", content: chunk })
|
||||
}
|
||||
return full
|
||||
}
|
||||
|
||||
async function handlePrompt(ws: WS, data: any) {
|
||||
if ((data.type ?? "message") === "new_conversation") return
|
||||
const prompt = String(data.content ?? "").trim()
|
||||
const model = String(data.model ?? "")
|
||||
if (!prompt) return
|
||||
|
||||
ws.data.pendingUserMessageId = insertMessage({
|
||||
conversation_id: ws.data.conversationId,
|
||||
role: "user",
|
||||
content: prompt,
|
||||
})
|
||||
try {
|
||||
const history: ChatMessage[] = conversationHistory(ws.data.conversationId).map((m) => ({
|
||||
role: m.role as ChatMessage["role"],
|
||||
content: m.content,
|
||||
}))
|
||||
let full: string
|
||||
try {
|
||||
full = await streamToClient(ws, streamChat(history, model, ws.data.abort.signal))
|
||||
} catch (e) {
|
||||
send(ws, { type: "error", content: `AI 服务出错：${e instanceof Error ? e.message : e}` })
|
||||
return
|
||||
}
|
||||
if (ws.data.closed) return
|
||||
|
||||
const code = extractCode(full)
|
||||
const messageId = insertMessage({
|
||||
conversation_id: ws.data.conversationId,
|
||||
role: "assistant",
|
||||
content: full,
|
||||
code_html: code.html,
|
||||
code_css: code.css,
|
||||
code_js: code.js,
|
||||
})
|
||||
ws.data.pendingUserMessageId = null
|
||||
send(ws, { type: "complete", code, message_id: messageId })
|
||||
} finally {
|
||||
dropPendingUserMessage(ws)
|
||||
}
|
||||
}
|
||||
|
||||
async function handleGuidance(ws: WS, data: any) {
|
||||
const prompt = String(data.content ?? "").trim()
|
||||
if (!prompt) return
|
||||
|
||||
ws.data.pendingUserMessageId = insertMessage({
|
||||
conversation_id: ws.data.conversationId,
|
||||
role: "user",
|
||||
content: prompt,
|
||||
source: "guidance",
|
||||
})
|
||||
ws.data.guidanceHistory.push({ role: "user", content: prompt })
|
||||
try {
|
||||
let full: string
|
||||
try {
|
||||
full = await streamToClient(ws, streamGuidance(ws.data.guidanceHistory, ws.data.abort.signal))
|
||||
} catch (e) {
|
||||
send(ws, { type: "error", content: `AI 服务出错：${e instanceof Error ? e.message : e}` })
|
||||
return
|
||||
}
|
||||
if (ws.data.closed) return
|
||||
|
||||
const [clean, isReady] = parseGuidanceResponse(full)
|
||||
ws.data.guidanceHistory.push({ role: "assistant", content: clean })
|
||||
const messageId = insertMessage({
|
||||
conversation_id: ws.data.conversationId,
|
||||
role: "assistant",
|
||||
content: clean,
|
||||
source: "guidance",
|
||||
})
|
||||
ws.data.pendingUserMessageId = null
|
||||
send(ws, { type: "complete", message_id: messageId, is_ready: isReady })
|
||||
} finally {
|
||||
dropPendingUserMessage(ws)
|
||||
}
|
||||
}
|
||||
|
||||
export const websocket: WebSocketHandler<WsData> = {
|
||||
// AI 生成可能要一两分钟，期间没有客户端消息
|
||||
idleTimeout: 960,
|
||||
sendPings: true,
|
||||
|
||||
open(ws) {
|
||||
const d = ws.data
|
||||
d.conversationId = getOrCreateActiveConversation(d.user.id, d.taskId)
|
||||
if (d.kind === "guidance") {
|
||||
send(ws, { type: "init", messages: [] })
|
||||
return
|
||||
}
|
||||
send(ws, {
|
||||
type: "init",
|
||||
conversation_id: uuidOut(d.conversationId),
|
||||
messages: conversationHistory(d.conversationId).map((m) => ({
|
||||
id: m.id,
|
||||
role: m.role,
|
||||
content: m.content,
|
||||
code:
|
||||
m.role === "assistant" ? { html: m.code_html, css: m.code_css, js: m.code_js } : null,
|
||||
submitted: m.role === "assistant" ? m.submission_id !== null : null,
|
||||
created: iso(m.created),
|
||||
})),
|
||||
})
|
||||
},
|
||||
|
||||
message(ws, raw) {
|
||||
let data: any
|
||||
try {
|
||||
data = JSON.parse(String(raw))
|
||||
} catch {
|
||||
return
|
||||
}
|
||||
const handler = ws.data.kind === "prompt" ? handlePrompt : handleGuidance
|
||||
ws.data.queue = ws.data.queue
|
||||
.then(() => (ws.data.closed ? undefined : handler(ws, data)))
|
||||
.catch((e) => console.error("websocket handler failed:", e))
|
||||
},
|
||||
|
||||
close(ws) {
|
||||
ws.data.closed = true
|
||||
// 客户端断开时停止向模型请求，未完成的提问会在 finally 中删除
|
||||
ws.data.abort.abort()
|
||||
dropPendingUserMessage(ws)
|
||||
},
|
||||
}
|
||||
Reference in new issue
Block a user