最近在用golang搭建个人博客的后台，数据库选用了gorm + mySQL的组合，其中文章(articles)和标签(tags)两个表是多对多的关系，由于对数据库不是很熟悉，外加gorm的文档较为简略，在这里踩了很多的坑。现在在这篇文章中总结下实现的方法。
第一步：定义gorm表的struct
type Article struct {
gorm.Model
Title string `gorm:"not null"`
Content string `gorm:"not null"`
Tags []Tag `gorm:"many2many:tag_articles"`
}
type Tag struct {
TagId string `gorm:"primary_key"`
TagName string `gorm:"unique;not null"`
Articles []Article `gorm:"many2many:tag_articles"`
}
第二步：连接数据库，创建表
func InitDB() *gorm.DB {
//创建一个数据库的连接
var err error
db, err := gorm.Open("mysql","root:071918@tcp(127.0.0.1:3306)/blog?charset=utf8")
if err != nil {
panic(err)
}
// 调用该语句会自动生成联结表tag_articles
db.AutoMigrate(&Article{}, &Tag{})
return db
}
第三步：存数据
tag1 := Tag{ TagId: "001", TagName: "React" }
db.Save(&tag1)
article := Article{
Title: "Redux进阶教程",
Content: "正文内容",
// 子项中的数据必须是保存过或者查找出的数据（即数据库已有的数据）
Tags: []Tag {
tag1,
},
}
或者
tagIdList := ["001", "002"]
var tags []db.Tag
// 分别查询tagIdList，并将结果插入tags列表
for i := range tagIdList {
tag := db.Tag{}
DB.Where("tag_id = ?", tagIdList[i]).First(&tag)
tags = append(tags, tag)
}
articleModel := db.Article{
Title: "这是标题",
Content: "这是正文",
Tags: tags,
}
// 保存
err := DB.Save(&articleModel)
在上述步骤中有需要注意的是，Tags对应的数组项必须事先存入数据库，或者事先查找过，否则仅会生成article的记录，不会生成tag_articles联结表的记录。
第四步：查询
因为不熟悉gorm的API，有一部分查询是使用基本的sql语句查询的，性能方面可能不太好，但是作为个人博客后端使用问题不大。
查询带标签(Tag)子项的文章(Article)列表
var articleList []Article
err := DB.Preload("Tags").Find(&articleList)
fmt.Println(articleList)
查询带文章数(Count)的标签(Tag)列表
// 定义所需的struct
type tagResult struct {
Count int
TagName string
TagId string
}
var tagList []tagResult
err := DB.Raw("SELECT COUNT(*) count, t.tag_name, t.tag_id " +
// 下面这句使用笛卡尔积连接两个表，复杂度变成了m * n，性能较差
"FROM tag_articles ta, tags t " +
"WHERE t.tag_id=ta.tag_tag_id " +
"GROUP BY ta.tag_tag_id ").Scan(&tagList)
fmt.Println(tagList)
根据标签(TagId)查询全部文章列表
// 定义所需的struct
type articleResult struct {
Title string
ArticleId string
}
var articleList []articleResult
tagId := "001"
// 查询语句
err := DB.Raw("SELECT a.title, a.article_id "+
// 下面这句使用笛卡尔积连接两个表，复杂度变成了m * n，性能较差
"FROM tag_articles ta, articles a "+
"WHERE a.article_id=ta.article_article_id "+
"AND ta.tag_tag_id= ? ", tagId).Scan(&articleList)
fmt.Println(articleList)
网友评论