57 lines
2.0 KiB
PL/PgSQL
57 lines
2.0 KiB
PL/PgSQL
-- 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);
|