gorm postgres全文搜索

在 GORM 中使用 PostgreSQL 全文搜索核心思路是利用 GORM 的 Raw() 或 Where() 配合原生 SQL 表达式调用 PostgreSQL 的 tsvector、tsquery 和 操作符。以下是完整的实现方案---1. 表结构与迁移在模型中定义一个 tsvector 类型的搜索列并通过 GORM 迁移创建type Article struct {ID uint gorm:primaryKeyTitle string gorm:type:text;not nullContent string gorm:type:text// tsvector 列用于全文搜索SearchVec string gorm:type:tsvector;index:idx_articles_search,type:ginCreatedAt time.Time}// AutoMigrate 会自动创建表和 GIN 索引db.AutoMigrate(Article{}) 注意GORM 的 index tag 支持指定索引类型为 gin这对全文搜索性能至关重要。---2. 填充 tsvector 数据方式一应用层手动更新适合数据量小或初始化// 将 Title 和 Content 合并转换为 tsvectordb.Exec(UPDATE articlesSET search_vec to_tsvector(chinese, COALESCE(title, ) || || COALESCE(content, )))方式二数据库触发器自动维护推荐通过 GORM 的 Migrator().RunWithValue() 或 Exec() 创建触发器确保数据变更时自动更新 tsvector// 创建触发器函数db.Exec(CREATE OR REPLACE FUNCTION articles_search_update() RETURNS trigger AS $$BEGINNEW.search_vec : to_tsvector(chinese,COALESCE(NEW.title, ) || || COALESCE(NEW.content, ));RETURN NEW;END;$$ LANGUAGE plpgsql;)// 绑定触发器db.Exec(CREATE TRIGGER trigger_articles_search_updateBEFORE INSERT OR UPDATE ON articlesFOR EACH ROW EXECUTE FUNCTION articles_search_update();) 触发器方案能确保搜索索引始终与源数据同步是生产环境的标准做法。---3. 执行全文搜索查询基础搜索简单匹配keyword : gorm 教程var articles []Articledb.Where(search_vec plainto_tsquery(chinese, ?), keyword).Find(articles)plainto_tsquery 会将用户输入的纯文本自动转换为 tsquery无需手动处理布尔运算符。带相关性排序的搜索keyword : gorm 教程var results []struct {ArticleRank float64 gorm:column:rank}db.Raw(SELECT *, ts_rank(search_vec, plainto_tsquery(chinese, ?)) as rankFROM articlesWHERE search_vec plainto_tsquery(chinese, ?)ORDER BY rank DESC, keyword, keyword).Scan(results)ts_rank() 会根据词频和位置计算相关性得分实现类似搜索引擎的排序效果。多字段加权搜索标题权重更高如果需要让标题匹配的结果排在前面可以使用 setweight// 首先修改触发器给不同字段设置权重db.Exec(CREATE OR REPLACE FUNCTION articles_search_update() RETURNS trigger AS $$BEGINNEW.search_vec :setweight(to_tsvector(chinese, COALESCE(NEW.title, )), A) ||setweight(to_tsvector(chinese, COALESCE(NEW.content, )), B);RETURN NEW;END;$$ LANGUAGE plpgsql;)查询时使用 ts_rank() 会自动考虑权重db.Raw(SELECT *, ts_rank(search_vec, plainto_tsquery(chinese, ?)) as rankFROM articlesWHERE search_vec plainto_tsquery(chinese, ?)ORDER BY rank DESCLIMIT 20, keyword, keyword).Scan(results)权重等级为 A(1.0) B(0.4) C(0.2) D(0.1)标题命中时会获得更高排名。支持网页搜索语法推荐用于用户输入websearch_to_tsquery 支持类似 Google 的搜索语法OR、-排除、短语// 用户输入: gorm OR postgres -mysqlkeyword : gorm OR postgres -mysqldb.Where(search_vec websearch_to_tsquery(chinese, ?), keyword).Find(articles)这比 plainto_tsquery 更灵活适合暴露给终端用户的搜索框。---4. 搜索结果高亮使用 ts_headline() 生成带高亮标记的摘要db.Raw(SELECTid,title,ts_headline(chinese, content,plainto_tsquery(chinese, ?),StartSelmark, StopSel/mark, MaxWords50, MinWords10) as snippetFROM articlesWHERE search_vec plainto_tsquery(chinese, ?), keyword, keyword).Scan(results)这会返回匹配关键词被 mark 标签包裹的文本片段方便前端直接展示。---5. 前缀搜索自动补全// 搜索以 post 开头的词prefix : postdb.Where(search_vec to_tsquery(chinese, ? || :*), prefix).Find(articles):* 是前缀匹配运算符适合实现搜索建议功能。---6. 中文支持注意事项PostgreSQL 默认的全文搜索配置对中文支持有限建议1. 使用 simple 配置对中文不做词干提取按空格分词to_tsvector(simple, title) // 中文内容推荐用 simple2. 安装额外扩展如 pg_jieba 或 zhparser 实现中文分词CREATE EXTENSION pg_jieba;-- 然后使用 jiebacfg 作为配置名3. 混合语言内容可以存储多语言 tsvector 或统一使用 simple 配置。---完整示例代码package mainimport (fmttimegorm.io/driver/postgresgorm.io/gorm)type Article struct {ID uint gorm:primaryKeyTitle string gorm:type:text;not nullContent string gorm:type:textSearchVec string gorm:type:tsvector;index:idx_articles_search,type:ginCreatedAt time.Time}func main() {dsn : hostlocalhost userpostgres passwordpostgres dbnametest port5432 sslmodedisabledb, err : gorm.Open(postgres.Open(dsn), gorm.Config{})if err ! nil {panic(err)}// 自动迁移db.AutoMigrate(Article{})// 创建触发器只需执行一次db.Exec(CREATE OR REPLACE FUNCTION articles_search_update() RETURNS trigger AS $$BEGINNEW.search_vec :setweight(to_tsvector(simple, COALESCE(NEW.title, )), A) ||setweight(to_tsvector(simple, COALESCE(NEW.content, )), B);RETURN NEW;END;$$ LANGUAGE plpgsql;)db.Exec(DROP TRIGGER IF EXISTS trigger_articles_search_update ON articles;CREATE TRIGGER trigger_articles_search_updateBEFORE INSERT OR UPDATE ON articlesFOR EACH ROW EXECUTE FUNCTION articles_search_update();)// 插入测试数据db.Create(Article{Title: GORM 入门教程, Content: GORM 是 Go 语言优秀的 ORM 框架})db.Create(Article{Title: PostgreSQL 全文搜索, Content: 使用 tsvector 实现高效搜索})// 执行搜索keyword : GORM 教程var articles []Articleresult : db.Where(search_vec websearch_to_tsquery(simple, ?),keyword,).Find(articles)if result.Error ! nil {panic(result.Error)}for _, a : range articles {fmt.Printf(ID: %d, Title: %s\n, a.ID, a.Title)}}---关键点总结功能 GORM 实现方式 PostgreSQL 函数文本转索引 tsvector 列 GIN 索引 to_tsvector()自动同步 数据库触发器 BEFORE INSERT OR UPDATE用户搜索 Where(... plainto_tsquery(...)) plainto_tsquery / websearch_to_tsquery相关性排序 Raw() ts_rank() ts_rank() / ts_rank_cd()结果高亮 Raw() ts_headline() ts_headline()前缀匹配 to_tsquery(prefix:*) :* 运算符这套方案充分利用了 PostgreSQL 原生全文搜索能力通过 GORM 的 SQL 注入安全查询接口实现性能上 GIN 索引可以支持百万级数据毫秒级响应。