camzalewski/personal-site
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Added shutdown method to container.

Browse source
This commit is contained in:
mikestefanello 2021-12-19 13:22:44 -05:00
parent c6c9ed7fd2
commit 85981e75a7
5 changed files with 35 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

35
services/container.go
View file

@ -21,13 +21,14 @@ import (
)
type Container struct {
Web *echo.Echo
Config *config.Config
Cache *cache.Cache
Database *sql.DB
ORM *ent.Client
Mail *mail.Client
Auth *AuthClient
Web *echo.Echo
Config *config.Config
Cache *cache.Cache
cacheClient *redis.Client
Database *sql.DB
ORM *ent.Client
Mail *mail.Client
Auth *AuthClient
}
func NewContainer() *Container {
@ -42,6 +43,20 @@ func NewContainer() *Container {
return c
}
func (c *Container) Shutdown() error {
if err := c.cacheClient.Close(); err != nil {
return err
}
if err := c.ORM.Close(); err != nil {
return err
}
if err := c.Database.Close(); err != nil {
return err
}
return nil
}
func (c *Container) initConfig() {
cfg, err := config.GetConfig()
if err != nil {
@ -63,14 +78,14 @@ func (c *Container) initWeb() {
}
func (c *Container) initCache() {
cacheClient := redis.NewClient(&redis.Options{
c.cacheClient = redis.NewClient(&redis.Options{
Addr: fmt.Sprintf("%s:%d", c.Config.Cache.Hostname, c.Config.Cache.Port),
Password: c.Config.Cache.Password,
})
if _, err := cacheClient.Ping(context.Background()).Result(); err != nil {
if _, err := c.cacheClient.Ping(context.Background()).Result(); err != nil {
panic(fmt.Sprintf("failed to connect to cache server: %v", err))
}
cacheStore := store.NewRedis(cacheClient, nil)
cacheStore := store.NewRedis(c.cacheClient, nil)
c.Cache = cache.New(cacheStore)
}