beego orm 批量插入并更新

CHUYUBO · · 1126 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

 需要用到根据唯一索引来进行批量更新和插入,这里利用ON DUPLICATE KEY UPDATE 来进行更新,遇到的问题:

1)因为刚接触go 不太清楚如何将切片转换为字符串,所以自己写了一个 方法 MapTranslateString

2)因为刚接触go 不太清楚orm的 sql的占位符除了‘?’之外还能是什么,导致打印出来的sql语句一堆问号,如果有错误也不好排查

// o orm 对象, dataList 需要传入参数 fieldList dataList 的 map的字符串键值 (更新的字段),tableName 表名称 ,size 一次插入长度
func (oh *OrmHelper) OnDuplicateKeyUpdate(O orm.Ormer, dataList []map[string]interface{}, fieldList []string, tableName string, size int) {
    if size < 100 {
        size = 100
    }
 
    tempList := dataList
 
    for dataList != nil {
 
        if len(dataList) > size {
            tempList = dataList[:size]
            temp := append(dataList[:0], dataList[size:]...)
            dataList = temp
        } else {
            tempList = dataList
            dataList = nil
        }
 
        sql := "INSERT INTO " + tableName + "(" + MapTranslateString(fieldList, ",") + ") VALUES"
        rawInterface := make([]interface{}, 0)
        for _, data := range tempList {
            var rawInterfaceTemp = make([]interface{}, len(fieldList))
            if data != nil {
                sql += "("
                var i = 0
                for _, field := range fieldList {
                    if data[field] == nil {
                        panic("批量插入失败!传入的数据中不存在字段" + field)
                    }
                    sql += "?,"
                    rawInterfaceTemp[i] = data[field]
                    i++
                }
                rawInterface = append(rawInterface, rawInterfaceTemp)
                sql = string([]byte(sql[:len(sql)-1])) + "),"
            }
        }
        sql = string([]byte(sql[:len(sql)-1]))
        sql += " ON DUPLICATE KEY UPDATE "
        for _, field := range fieldList {
            sql += "`" + field + "`=values(`" + field + "`),"
        }
        sql = string([]byte(sql[:len(sql)-1]))
        _, err := O.Raw(sql, rawInterface).Exec()
        if err != nil {
            panic(err)
        }
    }
}
 
func MapTranslateString(mapString []string, op string) string {
    str := ""
    length := len(mapString)
    if length > 0 {
        for i, value := range mapString {
            if i == 0 {
                str += value
            } else if i < length {
                str += op + value
            }
        }
    }
    return str
}

 


有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1126 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传