【Go・sqlx】where文に配列を使ってレコードを取得したい

Question

お聞きしたいこと

• sqlxライブラリとGoを使って、IDを取得したい。
  • where文で絞り込む時に、配列の値を使いたい。

環境

• Go 1.19.3
• sqlx
• PostgreSQL

具体的な環境

DB

userテーブル

order_id	user_id	name
1	111	田中
2	222	佐藤
3	333	鈴木
4	444	中島

コード

func main() {
	var useIDs []string
	useIDs = append(useIDs, "111", "333")

	orderIDs, err := listOrderIDByUseID(conn, useIDs)
	if err != nil {
		fmt.Println(err)
	}

	fmt.Println(orderIDs)
}

func listOrderIDByUseID(conn repository.DBconn, useIDs []string) ([]string, error) {
	var rows []string
	// WHERE文で useIDs の配列を渡して、該当する order_id の配列を返したい (下記は上手く行かなかったコード)
	err := conn.Select(&rows,
		`
		SELECT
			u.order_id
		FROM user AS u
		WHERE
			u.user_id = ANY($1)
		`,
		pq.Array(useIDs),
	)

	if err != nil {
		return nil, err
	}

	return rows, nil
}

希望の出力結果

❯ go run main.go                
[1 3]

試したこと

• conn.Select内のWHEREで指定するuseIDをrangeで回して渡しましたが、上手くいきませんでした
  • 参考記事: golangでpostgresを操作する。

さいごに

詳しい方ご教授いただけますと嬉しいです！
よろしくお願いいたします！！

Comments

[melian]

もしかして、user テーブルの user_id は integer 型ではないでしょうか？

Answer 1

sqlxのIn関数で実現できました！