go/squirrel で最後にinsertされたidを取得したい

Question

わからないこと

squirrelで、insertされたID(primary key、オートインクリメント値)を取得する方法を教えていただきたいです。

database/sqlではLastInsertIdというものがあるようでしたが、同様の機能のものをsquirrelで見つけることができませんでした。
なにかありましたら教えていただけると助かります。よろしくお願いいたします。

stmt, err := db.Prepare("INSERT table SET unique_id=? ON DUPLICATE KEY UPDATE id=LAST_INSERT_ID(id)")
res, err := stmt.Exec(unique_id)
lid, err := res.LastInsertId()

追記

現状下記のようなinsertメソッドにしていて、ここからinsertした場合のidを取得したいです。

func (p *Postgres) InsertHoge(tx *sqlx.Tx, hoge models.Hoge) (models.Hoge, error) {

	query, args, err := sq.StatementBuilder.PlaceholderFormat(sq.Dollar).
		Insert(`"hoge"`).
		Columns(
			`"title"`,
		).
		Values(
			hoge.Title,
		).
		ToSql()

	if err != nil {
		return hoge, err
	}

	if err := tx.Get(&hoge, query, args...); err != nil {
		return hoge, err
	}

	return hoge, nil

}

Comments

[d_tutuz]

squirrel といっているのは https://github.com/Masterminds/squirrel のことであっていますか？？

[hello_whats_up]

はい、そうです！
不足があり申し訳ありません

Answer 1

追記いただいたコード(InsertHoge メソッド)をベースに全体的に書き換えました。ローカルで試す際は Docker などを用いて PostgreSQL に接続できる環境でお試しください。

LastInsertId は PostgreSQL のドライバーでは使えない (参照: no LastInsertId available in Postgresql) ため RETURNING 句を用いるのはどうでしょうか？RETURNING は squirrel のクエリビルダで付与するイメージです。

コード

package main

import (
	"database/sql"
	"fmt"

	"github.com/jmoiron/sqlx"
	_ "github.com/lib/pq"

	sq "github.com/Masterminds/squirrel"
)

func (p *Postgres) InsertHoge(hoge *Hoge) (*Hoge, error) {
	query, args, err := sq.StatementBuilder.PlaceholderFormat(sq.Dollar).
		Insert("hoge").Columns("title").Values(hoge.Title).Suffix("RETURNING *").ToSql()

	if err != nil {
		return nil, err
	}

	tx, err := p.Beginx()
	if err != nil {
		return nil, err
	}

	var h Hoge
	if err := tx.QueryRowx(query, args[0]).StructScan(&h); err != nil {
		return nil, err
	}

	tx.Commit()

	// エラーのときのロールバックの処理は割愛
	return &h, nil
}

// ------------------------------------------------------------------------------------------------
// ローカルの Docker の起動
// $ docker run --name tmp-postgres -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres -d -p 5432:5432 postgres
// ------------------------------------------------------------------------------------------------

func main() {
	// Docker上のデータベースへの接続
	db, err := sql.Open("postgres", fmt.Sprintf(
		"user=%s password=%s host=%s port=%s dbname=%s sslmode=%s",
		"postgres", "postgres", "localhost", "5432", "postgres", "disable",
	))
	if err != nil {
		panic(err)
	}

	// sqlxライブラリの使用
	x := sqlx.NewDb(db, "postgres")

	// Postgres構造体の値生成
	p := &Postgres{x}

	// クエリの実行
	m, err := p.InsertHoge(&Hoge{Title: "gopher"})
	if err != nil {
		panic(err)
	}

	// 結果の表示
	fmt.Printf("%+v\n", m)
}

/*
-- DDLは以下のようなものを想定

CREATE TABLE hoge (
    id SERIAL NOT NULL,
    title VARCHAR(255) NOT NULL,
    PRIMARY KEY (id)
);
*/
type Hoge struct {
	Id    int    `db:"id"`
	Title string `db:"title"`
}

type Postgres struct{ *sqlx.DB }

取得できる結果例

以下のように ID と Title の組み合わせでオートインクリメントされた ID の値も取得できます。

&{Id:3 Title:gopher}

Comments

[hello_whats_up]

回答ありがとうございます。
本文に追記したようなコード中で使用してみたのですが、`unreferenced` などのエラーが出てしまうので、上記のようなコード中での使用法を教えていただけないでしょうか？

[d_tutuz]

追記いただいたコードで全面的に回答を書き換えました！お試しくださいっ

[hello_whats_up]

ありがとうございます！
試してみてまた連絡させていただきます！

[hello_whats_up]

ありがとうございます！
教えて頂いたコードで取得することができました！
あと一つ教えていただきたいのですが、取得したIDの値をInsertHogeメソッド内で取得することも可能なのでしょうか？？

[d_tutuz]

tx.QueryRowx(query, args[0]).StructScan(&h) で変数 h の中に Id と Title がセットされるので、変数 h を用いれば InsertHoge メソッド内で Id を使うことができます。

[hello_whats_up]

おかげさまで解決することができました。
ご丁寧にありがとうございました！

[d_tutuz]

良かったです！