File size: 1,356 Bytes
fea99b3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | package entdriver
import (
"database/sql"
"entgo.io/ent/dialect"
entDialectSQL "entgo.io/ent/dialect/sql"
entdb "github.com/openmeterio/openmeter/openmeter/ent/db"
)
type EntPostgresDriver struct {
driver *entDialectSQL.Driver
client *entdb.Client
}
// Close releases all the underlying resources.
func (d *EntPostgresDriver) Close() error {
if err := d.client.Close(); err != nil {
return err
}
if err := d.driver.Close(); err != nil {
return err
}
return nil
}
// Driver returns the underlying Driver.
func (d *EntPostgresDriver) Driver() *entDialectSQL.Driver {
return d.driver
}
// Client returns the underlying Client.
func (d *EntPostgresDriver) Client() *entdb.Client {
return d.client
}
// Clone returns a new EntPostgresDriver initialized by using the underlying *sql.DB.
func (d *EntPostgresDriver) Clone() *EntPostgresDriver {
driver := entDialectSQL.OpenDB(dialect.Postgres, d.driver.DB())
client := entdb.NewClient(entdb.Driver(driver))
return &EntPostgresDriver{
driver: driver,
client: client,
}
}
// NewEntPostgresDriver returns a EntPostgresDriver created from *sql.DB.
func NewEntPostgresDriver(db *sql.DB) *EntPostgresDriver {
driver := entDialectSQL.OpenDB(dialect.Postgres, db)
client := entdb.NewClient(entdb.Driver(driver))
return &EntPostgresDriver{
driver: driver,
client: client,
}
}
|