Cleanup container init.
This commit is contained in:
parent
8657380530
commit
e5038f7bbf
3 changed files with 27 additions and 16 deletions
|
|
@ -22,34 +22,36 @@ type Container struct {
|
||||||
Config *config.Config
|
Config *config.Config
|
||||||
Cache *cache.Cache
|
Cache *cache.Cache
|
||||||
Database *sql.DB
|
Database *sql.DB
|
||||||
Ent *ent.Client
|
ORM *ent.Client
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewContainer() *Container {
|
func (c *Container) initWeb() {
|
||||||
var c Container
|
|
||||||
|
|
||||||
// Web
|
|
||||||
c.Web = echo.New()
|
c.Web = echo.New()
|
||||||
|
}
|
||||||
|
|
||||||
// Configuration
|
func (c *Container) initConfig() {
|
||||||
cfg, err := config.GetConfig()
|
cfg, err := config.GetConfig()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Web.Logger.Fatalf("failed to load configuration: %v", err)
|
c.Web.Logger.Fatalf("failed to load configuration: %v", err)
|
||||||
}
|
}
|
||||||
c.Config = &cfg
|
c.Config = &cfg
|
||||||
|
}
|
||||||
|
|
||||||
// Cache
|
func (c *Container) initCache() {
|
||||||
cacheClient := redis.NewClient(&redis.Options{
|
cacheClient := redis.NewClient(&redis.Options{
|
||||||
Addr: fmt.Sprintf("%s:%d", c.Config.Cache.Hostname, c.Config.Cache.Port),
|
Addr: fmt.Sprintf("%s:%d", c.Config.Cache.Hostname, c.Config.Cache.Port),
|
||||||
Password: c.Config.Cache.Password,
|
Password: c.Config.Cache.Password,
|
||||||
})
|
})
|
||||||
if _, err = cacheClient.Ping(context.Background()).Result(); err != nil {
|
if _, err := cacheClient.Ping(context.Background()).Result(); err != nil {
|
||||||
c.Web.Logger.Fatalf("failed to connect to cache server: %v", err)
|
c.Web.Logger.Fatalf("failed to connect to cache server: %v", err)
|
||||||
}
|
}
|
||||||
cacheStore := store.NewRedis(cacheClient, nil)
|
cacheStore := store.NewRedis(cacheClient, nil)
|
||||||
c.Cache = cache.New(cacheStore)
|
c.Cache = cache.New(cacheStore)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) initDatabase() {
|
||||||
|
var err error
|
||||||
|
|
||||||
// Database
|
|
||||||
addr := fmt.Sprintf("postgresql://%s:%s@%s/%s",
|
addr := fmt.Sprintf("postgresql://%s:%s@%s/%s",
|
||||||
c.Config.Database.User,
|
c.Config.Database.User,
|
||||||
c.Config.Database.Password,
|
c.Config.Database.Password,
|
||||||
|
|
@ -60,13 +62,22 @@ func NewContainer() *Container {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Web.Logger.Fatalf("failed to connect to database: %v", err)
|
c.Web.Logger.Fatalf("failed to connect to database: %v", err)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Ent
|
func (c *Container) initORM() {
|
||||||
drv := entsql.OpenDB(dialect.Postgres, c.Database)
|
drv := entsql.OpenDB(dialect.Postgres, c.Database)
|
||||||
c.Ent = ent.NewClient(ent.Driver(drv))
|
c.ORM = ent.NewClient(ent.Driver(drv))
|
||||||
if err := c.Ent.Schema.Create(context.Background()); err != nil {
|
if err := c.ORM.Schema.Create(context.Background()); err != nil {
|
||||||
c.Web.Logger.Fatalf("failed to create database schema: %v", err)
|
c.Web.Logger.Fatalf("failed to create database schema: %v", err)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return &c
|
|
||||||
|
func NewContainer() *Container {
|
||||||
|
c := new(Container)
|
||||||
|
c.initWeb()
|
||||||
|
c.initConfig()
|
||||||
|
c.initCache()
|
||||||
|
c.initDatabase()
|
||||||
|
c.initORM()
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -31,7 +31,7 @@ func (l *Login) Post(c echo.Context) error {
|
||||||
return l.Get(c)
|
return l.Get(c)
|
||||||
}
|
}
|
||||||
|
|
||||||
u, err := l.Container.Ent.User.
|
u, err := l.Container.ORM.User.
|
||||||
Query().
|
Query().
|
||||||
Where(user.Username(name)).
|
Where(user.Username(name)).
|
||||||
First(c.Request().Context())
|
First(c.Request().Context())
|
||||||
|
|
|
||||||
|
|
@ -56,7 +56,7 @@ func (r *Register) Post(c echo.Context) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Attempt creating the user
|
// Attempt creating the user
|
||||||
u, err := r.Container.Ent.User.
|
u, err := r.Container.ORM.User.
|
||||||
Create().
|
Create().
|
||||||
SetUsername(form.Username).
|
SetUsername(form.Username).
|
||||||
SetPassword(string(pwHash)).
|
SetPassword(string(pwHash)).
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue