如何在postgres中使用sqlx获取最后插入行的uuid

golang学习网今天将给大家带来《如何在postgres中使用sqlx获取最后插入行的uuid》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

我正在使用 sqlx 在表中插入一些数据，如下所示

func (*riderepositoryimpl) insert(entity interface{}, tx persistence.transaction) (sql.result, error) {
    ride := entity.(*model.ride)

    placeholders := repository.insertplaceholders(len(ridecolumns))
    sql := fmt.sprintf("insert into %s(%s) values(%s)", tablename, strings.join(columns, ","), placeholders)
    return tx.exec(sql, ride.id.string(), ride.deviceiotid, ride.vehicleid.string(), ride.userid.string(),ride.additioncomments)
}

并像这样调用这个函数

func (p *RideRepositoryImpl) Save(ride *model.Ride, tx persistence.Transaction) error {
    return repository.Save(ride, p.insert, tx)

现在我想在保存该记录后立即获取已保存记录的uuid。有什么干净的方法可以立即执行此操作吗？

正确答案

postgresql 为此提供了 RETURNING 子句。

有时，在修改行时获取数据很有用 正在被操纵。 insert、update 和 delete 命令 它们都有一个可选的 returning 子句来支持这一点。用于 returning 避免执行额外的数据库查询来收集 数据，并且在其他情况下很难实现的情况下尤其有价值 可靠地识别修改的行。

// add the returning clause to your insert query
sql := fmt.sprintf("insert into %s(%s) values(%s) returning <name_of_uuid_column>", tablename, strings.join(columns, ","), placeholders)

// use queryrow instead of exec
row := tx.queryrow(sql, ride.id.string(), ride.deviceiotid, ride.vehicleid.string(), ride.userid.string(),ride.additioncomments)

// scan the result of the query
var uuid string
if err := row.scan(&uuid); err != nil {
    panic(err)
}

// ...

有关 returning 的其他 insert 特定信息，您可以转到 INSERT 文档并使用 ctrl/cmd+f 搜索页面“返回”。

此外，如果您需要函数仍然返回 sql.result 值来满足某些要求，那么您可以返回自己的实现。

var _ sql.Result = sqlresult{} // compiler check

type sqlresult struct { lastid, nrows int64 }

func (r sqlresult) LastInsertId() (int64, error) { return r.lastid, nil }
func (r sqlresult) RowsAffected() (int64, error) { return r.nrows, nil }

func (*RideRepositoryImpl) insert(entity interface{}, tx persistence.Transaction) (sql.Result, error) {
    ride := entity.(*model.Ride)

    placeHolders := repository.InsertPlaceholders(len(rideColumns))
    sql := fmt.Sprintf("INSERT INTO %s(%s) VALUES(%s) RETURNING <name_of_uuid_column>", TableName, strings.Join(Columns, ","), placeHolders)
    row := tx.QueryRow(sql, ride.ID.String(), ride.DeviceIotID, ride.VehicleID.String(), ride.UserID.String(),ride.AdditionComments)
    if err := row.Scan(&ride.<NameOfUUIDField>); err != nil {
        return nil, err
    }
    return sqlresult{0, 1}, nil
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何在postgres中使用sqlx获取最后插入行的uuid》文章吧，也可关注golang学习网公众号了解相关技术文章。