40 lines
639 B
Go
40 lines
639 B
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Store interface {
|
||
|
|
Querier
|
||
|
|
}
|
||
|
|
|
||
|
|
type SQLStore struct {
|
||
|
|
*Queries
|
||
|
|
db *sql.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewStore(db *sql.DB) Store {
|
||
|
|
return &SQLStore{
|
||
|
|
db: db,
|
||
|
|
Queries: New(db),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TRANSACTION QUERY FUNCTION
|
||
|
|
// func (store *SQLStore) execTx(ctx context.Context, fn func(*Queries) error) error {
|
||
|
|
// tx, err := store.db.BeginTx(ctx, nil)
|
||
|
|
// if err != nil {
|
||
|
|
// return err
|
||
|
|
// }
|
||
|
|
|
||
|
|
// q := New(tx)
|
||
|
|
// err = fn(q)
|
||
|
|
// if err != nil {
|
||
|
|
// if rbErr := tx.Rollback(); rbErr != nil {
|
||
|
|
// return fmt.Errorf("tx err: %v, rb err : %v", err, rbErr)
|
||
|
|
// }
|
||
|
|
// return err
|
||
|
|
// }
|
||
|
|
// return tx.Commit()
|
||
|
|
// }
|