31 lines
485 B
Go
31 lines
485 B
Go
package pgsql
|
|
|
|
import (
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
"opensavecloudserver/data/datasource"
|
|
)
|
|
|
|
type (
|
|
DatabaseDatasource struct {
|
|
conn *gorm.DB
|
|
}
|
|
)
|
|
|
|
func NewDatabaseDatasource() datasource.Datasource {
|
|
return new(DatabaseDatasource)
|
|
}
|
|
|
|
func (dd *DatabaseDatasource) Connect(dsn string) error {
|
|
conn, err := gorm.Open(postgres.Open(dsn))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dd.conn = conn
|
|
return nil
|
|
}
|
|
|
|
func (dd *DatabaseDatasource) DB() *gorm.DB {
|
|
return dd.conn
|
|
}
|