86 lines
1.6 KiB
Go
86 lines
1.6 KiB
Go
|
package models
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
type migrationFunc func() error
|
||
|
|
|
||
|
|
type service struct {
|
||
|
|
name string
|
||
|
|
automigrate migrationFunc
|
||
|
|
destructivereset migrationFunc
|
||
|
|
}
|
||
|
|
|
||
|
|
type Services struct {
|
||
|
|
AccountS AccountService
|
||
|
|
Database *sql.DB
|
||
|
|
services []service
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Services) AutoMigrate() error {
|
||
|
|
for _, service := range s.services {
|
||
|
|
err := service.automigrate()
|
||
|
|
if err != nil {
|
||
|
|
return pkgErr(fmt.Sprintf("error auto-migrating %s service", service.name), err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Services) Close() error {
|
||
|
|
err := s.Database.Close()
|
||
|
|
if err != nil {
|
||
|
|
return pkgErr("error closing database", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Services) DestructiveReset() error {
|
||
|
|
for _, service := range s.services {
|
||
|
|
err := service.destructivereset()
|
||
|
|
if err != nil {
|
||
|
|
return pkgErr(fmt.Sprintf("error destructive-resetting %s service", service.name), err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type ServicesInit func(*Services) error
|
||
|
|
|
||
|
|
func NewServices(inits ...ServicesInit) (*Services, error) {
|
||
|
|
var s Services
|
||
|
|
for _, init := range inits {
|
||
|
|
err := init(&s)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return &s, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithDatabase(file string) ServicesInit {
|
||
|
|
return func(s *Services) error {
|
||
|
|
db, err := sql.Open("sqlite3", "file:"+file+"?_foreign_keys=ON")
|
||
|
|
if err != nil {
|
||
|
|
return pkgErr("error opening database", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
s.Database = db
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithAccountService() ServicesInit {
|
||
|
|
return func(s *Services) error {
|
||
|
|
s.AccountS = NewAccountService(s.Database)
|
||
|
|
s.services = append(s.services, service{accountTable, s.AccountS.AutoMigrate, s.AccountS.DestructiveReset})
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|