48 lines
984 B
Go
48 lines
984 B
Go
|
package orm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"gitlab.com/wpetit/goweb/service"
|
||
|
|
)
|
||
|
|
|
||
|
|
const ServiceName service.Name = "orm"
|
||
|
|
|
||
|
|
type Service struct {
|
||
|
|
db *gorm.DB
|
||
|
|
migration *MigrationManager
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) DB() *gorm.DB {
|
||
|
|
return s.db
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) Migration() *MigrationManager {
|
||
|
|
return s.migration
|
||
|
|
}
|
||
|
|
|
||
|
|
// From retrieves the orm service in the given container.
|
||
|
|
func From(container *service.Container) (*Service, error) {
|
||
|
|
service, err := container.Service(ServiceName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrapf(err, "error while retrieving '%s' service", ServiceName)
|
||
|
|
}
|
||
|
|
|
||
|
|
srv, ok := service.(*Service)
|
||
|
|
if !ok {
|
||
|
|
return nil, errors.Errorf("retrieved service is not a valid '%s' service", ServiceName)
|
||
|
|
}
|
||
|
|
|
||
|
|
return srv, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Must retrieves the orm pool service in the given container or panic otherwise.
|
||
|
|
func Must(container *service.Container) *Service {
|
||
|
|
srv, err := From(container)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return srv
|
||
|
|
}
|