57 lines
2.0 KiB
MySQL
57 lines
2.0 KiB
MySQL
|
-- 1) 补齐缺失列（幂等）
|
|||
|
|
ALTER TABLE documents
|
|||
|
|
ADD COLUMN IF NOT EXISTS content_sha TEXT;
|
|||
|
|
|
|||
|
|
ALTER TABLE documents
|
|||
|
|
ADD COLUMN IF NOT EXISTS doc_key TEXT
|
|||
|
|
GENERATED ALWAYS AS (repo || ':' || path || ':' || chunk_id) STORED;
|
|||
|
|
|
|||
|
|
-- 2) 先删除旧的向量索引（若存在，避免定义不一致）
|
|||
|
|
DROP INDEX IF EXISTS documents_embedding_idx;
|
|||
|
|
|
|||
|
|
-- 3) 重建 HNSW 索引（正确顺序：WITH -> WHERE）
|
|||
|
|
CREATE INDEX IF NOT EXISTS documents_embedding_idx
|
|||
|
|
ON documents
|
|||
|
|
USING hnsw (embedding vector_cosine_ops)
|
|||
|
|
WITH (m = 16, ef_construction = 64)
|
|||
|
|
WHERE embedding IS NOT NULL;
|
|||
|
|
|
|||
|
|
-- 4) 元数据/常用索引已存在就跳过（这里仅确保存在）
|
|||
|
|
CREATE INDEX IF NOT EXISTS idx_documents_metadata ON documents USING gin (metadata);
|
|||
|
|
CREATE INDEX IF NOT EXISTS idx_documents_repo ON documents (repo);
|
|||
|
|
CREATE INDEX IF NOT EXISTS idx_documents_path ON documents (path);
|
|||
|
|
CREATE INDEX IF NOT EXISTS idx_documents_repo_path ON documents (repo, path);
|
|||
|
|
|
|||
|
|
-- 5) 全文检索列与索引（若之前没建会自动补齐）
|
|||
|
|
DO $$
|
|||
|
|
BEGIN
|
|||
|
|
IF NOT EXISTS (
|
|||
|
|
SELECT 1 FROM information_schema.columns
|
|||
|
|
WHERE table_name='documents' AND column_name='content_tsv'
|
|||
|
|
) THEN
|
|||
|
|
ALTER TABLE documents ADD COLUMN content_tsv tsvector;
|
|||
|
|
END IF;
|
|||
|
|
END$$;
|
|||
|
|
|
|||
|
|
CREATE OR REPLACE FUNCTION documents_tsv_trigger()
|
|||
|
|
RETURNS trigger AS $$
|
|||
|
|
BEGIN
|
|||
|
|
NEW.content_tsv := setweight(to_tsvector('english', coalesce(NEW.content,'')), 'A');
|
|||
|
|
RETURN NEW;
|
|||
|
|
END$$ LANGUAGE plpgsql;
|
|||
|
|
|
|||
|
|
DO $$
|
|||
|
|
BEGIN
|
|||
|
|
IF NOT EXISTS (SELECT 1 FROM pg_trigger WHERE tgname = 'documents_tsv_update') THEN
|
|||
|
|
CREATE TRIGGER documents_tsv_update
|
|||
|
|
BEFORE INSERT OR UPDATE OF content ON documents
|
|||
|
|
FOR EACH ROW EXECUTE FUNCTION documents_tsv_trigger();
|
|||
|
|
END IF;
|
|||
|
|
END$$;
|
|||
|
|
|
|||
|
|
CREATE INDEX IF NOT EXISTS idx_documents_tsv ON documents USING gin (content_tsv);
|
|||
|
|
|
|||
|
|
-- 6) 为 ON CONFLICT 提供唯一约束/索引（两种都可以，有其一即可）
|
|||
|
|
-- 简化做法：建唯一索引（即可满足 ON CONFLICT）
|
|||
|
|
CREATE UNIQUE INDEX IF NOT EXISTS documents_doc_key_uk ON documents (doc_key);
|