Go语言 使用sqlx返回Postgres中新插入的行

k2fxgqgv  于 11个月前  发布在  Go
关注(0)|答案(4)|浏览(95)

我使用https://github.com/jmoiron/sqlx来查询Postgres。
插入新行时是否可以取回整行数据?
下面是我运行的查询:

result, err := Db.Exec("INSERT INTO users (name) VALUES ($1)", user.Name)

字符串
或者我应该使用现有的user结构体作为数据库中新条目的真实来源?

w46czmvw

w46czmvw1#

以下是sqlx的交易文档:
结果有两个可能的数据:LastInsertId()或RowsAffected(),其可用性取决于驱动程序。例如,在MySQL中,LastInsertId()将在具有自动递增键的插入中可用,但在PostgreSQL中,只能通过使用RETURNING子句从普通行游标中检索此信息。
所以我做了一个完整的demo,演示了如何使用sqlx执行transaction,demo将在addresses表中创建一个地址行,然后在users表中使用新的address_id PK作为用户的user_address_id FK创建一个用户。

package transaction

import (
    "database/sql"
    "github.com/jmoiron/sqlx"
    "log"
    "github.com/pkg/errors"
)
import (
    "github.com/icrowley/fake"
)

type User struct {
    UserID int `db:"user_id"`
    UserNme string `db:"user_nme"`
    UserEmail string `db:"user_email"`
    UserAddressId sql.NullInt64 `db:"user_address_id"`
}

type ITransactionSamples interface {
    CreateUserTransaction() (*User, error)
}

type TransactionSamples struct {
    Db *sqlx.DB
}

func NewTransactionSamples(Db *sqlx.DB) ITransactionSamples {
    return &TransactionSamples{Db}
}

func (ts *TransactionSamples) CreateUserTransaction() (*User, error) {
    tx := ts.Db.MustBegin()
    var lastInsertId int
    err := tx.QueryRowx(`INSERT INTO addresses (address_id, address_city, address_country, address_state) VALUES ($1, $2, $3, $4) RETURNING address_id`, 3, fake.City(), fake.Country(), fake.State()).Scan(&lastInsertId)
    if err != nil {
        tx.Rollback()
        return nil, errors.Wrap(err, "insert address error")
    }
    log.Println("lastInsertId: ", lastInsertId)

    var user User
    err = tx.QueryRowx(`INSERT INTO users (user_id, user_nme, user_email, user_address_id) VALUES ($1, $2, $3, $4) RETURNING *;`, 6, fake.UserName(), fake.EmailAddress(), lastInsertId).StructScan(&user)
    if err != nil {
        tx.Rollback()
        return nil, errors.Wrap(err, "insert user error")
    }
    err = tx.Commit()
    if err != nil {
        return nil, errors.Wrap(err, "tx.Commit()")
    }
    return &user, nil
}

字符串
下面是测试结果:

☁  transaction [master] ⚡  go test -v -count 1 ./...
=== RUN   TestCreateUserTransaction
2019/06/27 16:38:50 lastInsertId:  3
--- PASS: TestCreateUserTransaction (0.01s)
    transaction_test.go:28: &transaction.User{UserID:6, UserNme:"corrupti", UserEmail:"[email protected]", UserAddressId:sql.NullInt64{Int64:3, Valid:true}}
PASS
ok      sqlx-samples/transaction        3.254s

tct7dpnv

tct7dpnv2#

这是一个示例代码,用于命名查询和插入数据和ID的强类型结构。
包括查询和结构以涵盖所使用的语法。

const query = `INSERT INTO checks (
        start, status) VALUES (
        :start, :status)
        returning id;`

type Row struct {
    Status string `db:"status"`
    Start time.Time `db:"start"`
}

func InsertCheck(ctx context.Context, row Row, tx *sqlx.Tx) (int64, error) {
    return insert(ctx, row, insertCheck, "checks", tx)
}

// insert inserts row into table using query SQL command
// table used only for loging, actual table name defined in query
// should not be used from services directly - implement strong type wrappers
// function expects query with named parameters
func insert(ctx context.Context, row interface{}, query string, table string, tx *sqlx.Tx) (int64, error) {
    // convert named query to native parameters format
    query, args, err := tx.BindNamed(query, row)
    if err != nil {
        return 0, fmt.Errorf("cannot bind parameters for insert into %q: %w", table, err)
    }

    var id struct {
        Val int64 `db:"id"`
    }

    err = sqlx.GetContext(ctx, tx, &id, query, args...)
    if err != nil {
        return 0, fmt.Errorf("cannot insert into %q: %w", table, err)
    }

    return id.Val, nil
}

字符串

0pizxfdo

0pizxfdo3#

PostgreSQL支持INSERT语句的RETURNING语法。
范例:

INSERT INTO users(...) VALUES(...) RETURNING id, name, foo, bar

字符串
文档:https://www.postgresql.org/docs/9.6/static/sql-insert.html
可选的RETURNING子句可使DDL根据实际插入的每一行计算并返回值(如果使用ON CONFLICT DO UPDATE子句,则更新)。这主要用于获取由默认值提供的值,例如序列号。但是,允许使用表的列的任何表达式。RETURNING列表的语法与SELECT的输出列表的语法相同。只有成功插入或更新将被返回。

zynd9foi

zynd9foi4#

你可以使用Get函数:

db.Get(user, "INSERT INTO users (name, email) VALUES ($1, $2) RETURNING *", user.Name, user.Email)

字符串

相关问题