go语言beego框架orm模型中添加批量插入数据方法及使用

Go  /  管理员 发布于 3年前   2088

beego中实现批量添加信息到数据库中，我这里以发布文章同时添加标签为案例（标签可以为多个）

可以接上一篇go的文章，自行翻阅

开始

1.标签模型 (省略一下不相干代码)

package models

import (
	"github.com/astaxie/beego/orm"
)

//tag表需显示的字段
type Tag struct {
	Tag_id  int `orm:"column(tag_id);pk"` // 设置主键
	Art_id  int
	Tagname string
	Addtime string
}

//实例
func init() {
	orm.RegisterModel(new(Tag))
}

...

//方法 添加
func AddTag(tag *Tag) (int64, error) {
	o := orm.NewOrm()
	id, err := o.Insert(tag)
	return id, err
}

//批量插入
func AddsTag(tags []Tag) (int64, error) {
	o := orm.NewOrm()
	nums, err := o.InsertMulti(len(tags), tags)
	return nums, err
}

2.控制器中调用 （两种方式）

a.调用模型中单条插入方法

//循环插入标签

s := time.Now().Unix()
timestamp := strconv.FormatInt(int64(s), 10)

tags := "1,2,3"
ss := strings.Split(tags, ",")
for i := 0; i < len(ss); i++ {
	tag := models.Tag{Art_id: int(id), Tagname: ss[i], Addtime: timestamp}
	models.AddTag(&tag)
}

b.数据组成二维数组，一次性插入

//一次性批量插入标签
s1 := time.Now().Unix()
timestamp := strconv.FormatInt(int64(s1), 10)

tags := "1,2,3"
ss := strings.Split(tags, ",")

tag := models.Tag{}
s := make([]models.Tag, 0)

for _, v := range ss {
	tag.Art_id = int(id)
	tag.Tagname = v
	tag.Addtime = timestamp

	s = append(s, tag)
}

//调用批量插入模型
models.AddsTag(s)

有兴趣的同学可以自行测试