basic.go

package postgres

import (
	"github.com/go-pg/pg/v9"
	"gitlab.com/xdevs23/goqlorm/database"
)

type TxMutationAdapter struct {
	tx *pg.Tx
}

func (TxMA TxMutationAdapter) Insert(obj interface{}) error {
	return TxMA.tx.Insert(obj)
}

func (TxMA TxMutationAdapter) Update(obj interface{}) error {
	return TxMA.tx.Update(obj)
}

func (TxMA TxMutationAdapter) Delete(obj interface{}) error {
	return TxMA.tx.Delete(obj)
}

func (db Database) Close() error {
	return db.PgDB.Close()
}

func (db Database) RunInTransaction(fn database.InnerTransactionFunc) error {
	return db.PgDB.RunInTransaction(func(tx *pg.Tx) error {
		return fn(TxMutationAdapter{tx})
	})
}