2021-12-18 07:07:12 -08:00
|
|
|
package services
|
2021-12-03 03:11:01 -08:00
|
|
|
|
|
|
|
import (
|
2021-12-05 17:56:38 -08:00
|
|
|
"context"
|
2021-12-10 17:44:23 -08:00
|
|
|
"database/sql"
|
2021-12-05 17:56:38 -08:00
|
|
|
"fmt"
|
|
|
|
|
2021-12-10 17:44:23 -08:00
|
|
|
"entgo.io/ent/dialect"
|
|
|
|
entsql "entgo.io/ent/dialect/sql"
|
2021-12-06 11:53:28 -08:00
|
|
|
"github.com/eko/gocache/v2/cache"
|
|
|
|
"github.com/eko/gocache/v2/store"
|
2021-12-05 17:56:38 -08:00
|
|
|
"github.com/go-redis/redis/v8"
|
2021-12-10 17:44:23 -08:00
|
|
|
_ "github.com/jackc/pgx/v4/stdlib"
|
2021-12-03 03:11:01 -08:00
|
|
|
"github.com/labstack/echo/v4"
|
2021-12-13 09:51:00 -08:00
|
|
|
"github.com/labstack/gommon/log"
|
2021-12-03 03:11:01 -08:00
|
|
|
|
|
|
|
"goweb/config"
|
2021-12-10 17:44:23 -08:00
|
|
|
"goweb/ent"
|
2021-12-03 03:11:01 -08:00
|
|
|
)
|
|
|
|
|
|
|
|
type Container struct {
|
2021-12-21 08:00:32 -08:00
|
|
|
Web *echo.Echo
|
|
|
|
Config *config.Config
|
|
|
|
Cache *cache.Cache
|
|
|
|
cacheClient *redis.Client
|
|
|
|
Database *sql.DB
|
|
|
|
ORM *ent.Client
|
|
|
|
Mail *MailClient
|
|
|
|
Auth *AuthClient
|
|
|
|
TemplateRenderer *TemplateRenderer
|
2021-12-03 03:11:01 -08:00
|
|
|
}
|
|
|
|
|
2021-12-12 18:28:53 -08:00
|
|
|
func NewContainer() *Container {
|
|
|
|
c := new(Container)
|
|
|
|
c.initConfig()
|
2021-12-13 09:51:00 -08:00
|
|
|
c.initWeb()
|
2021-12-12 18:28:53 -08:00
|
|
|
c.initCache()
|
|
|
|
c.initDatabase()
|
|
|
|
c.initORM()
|
2021-12-15 06:29:43 -08:00
|
|
|
c.initAuth()
|
2021-12-19 17:09:01 -08:00
|
|
|
c.initTemplateRenderer()
|
2021-12-19 17:37:51 -08:00
|
|
|
c.initMail()
|
2021-12-12 18:28:53 -08:00
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
2021-12-19 10:22:44 -08:00
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2021-12-11 10:18:32 -08:00
|
|
|
func (c *Container) initConfig() {
|
2021-12-03 03:11:01 -08:00
|
|
|
cfg, err := config.GetConfig()
|
|
|
|
if err != nil {
|
2021-12-13 09:51:00 -08:00
|
|
|
panic(fmt.Sprintf("failed to load config: %v", err))
|
2021-12-03 03:11:01 -08:00
|
|
|
}
|
|
|
|
c.Config = &cfg
|
2021-12-11 10:18:32 -08:00
|
|
|
}
|
2021-12-03 03:11:01 -08:00
|
|
|
|
2021-12-13 09:51:00 -08:00
|
|
|
func (c *Container) initWeb() {
|
|
|
|
c.Web = echo.New()
|
|
|
|
|
|
|
|
// Configure logging
|
|
|
|
switch c.Config.App.Environment {
|
|
|
|
case config.EnvProduction:
|
|
|
|
c.Web.Logger.SetLevel(log.WARN)
|
|
|
|
default:
|
|
|
|
c.Web.Logger.SetLevel(log.DEBUG)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-11 10:18:32 -08:00
|
|
|
func (c *Container) initCache() {
|
2021-12-19 10:22:44 -08:00
|
|
|
c.cacheClient = redis.NewClient(&redis.Options{
|
2021-12-05 17:56:38 -08:00
|
|
|
Addr: fmt.Sprintf("%s:%d", c.Config.Cache.Hostname, c.Config.Cache.Port),
|
|
|
|
Password: c.Config.Cache.Password,
|
|
|
|
})
|
2021-12-19 10:22:44 -08:00
|
|
|
if _, err := c.cacheClient.Ping(context.Background()).Result(); err != nil {
|
2021-12-13 09:51:00 -08:00
|
|
|
panic(fmt.Sprintf("failed to connect to cache server: %v", err))
|
2021-12-05 17:56:38 -08:00
|
|
|
}
|
2021-12-19 10:22:44 -08:00
|
|
|
cacheStore := store.NewRedis(c.cacheClient, nil)
|
2021-12-06 11:53:28 -08:00
|
|
|
c.Cache = cache.New(cacheStore)
|
2021-12-11 10:18:32 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (c *Container) initDatabase() {
|
|
|
|
var err error
|
2021-12-05 17:56:38 -08:00
|
|
|
|
2021-12-13 09:51:00 -08:00
|
|
|
getAddr := func(dbName string) string {
|
|
|
|
return fmt.Sprintf("postgresql://%s:%s@%s/%s",
|
|
|
|
c.Config.Database.User,
|
|
|
|
c.Config.Database.Password,
|
|
|
|
c.Config.Database.Hostname,
|
|
|
|
dbName,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
c.Database, err = sql.Open("pgx", getAddr(c.Config.Database.Database))
|
2021-12-10 17:44:23 -08:00
|
|
|
if err != nil {
|
2021-12-13 09:51:00 -08:00
|
|
|
panic(fmt.Sprintf("failed to connect to database: %v", err))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check if this is a test environment
|
|
|
|
if c.Config.App.Environment == config.EnvTest {
|
|
|
|
// Drop the test database, ignoring errors in case it doesn't yet exist
|
|
|
|
_, _ = c.Database.Exec("DROP DATABASE " + c.Config.Database.TestDatabase)
|
|
|
|
|
|
|
|
// Create the test database
|
|
|
|
if _, err = c.Database.Exec("CREATE DATABASE " + c.Config.Database.TestDatabase); err != nil {
|
|
|
|
panic(fmt.Sprintf("failed to create test database: %v", err))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Connect to the test database
|
|
|
|
if err = c.Database.Close(); err != nil {
|
|
|
|
panic(fmt.Sprintf("failed to close database connection: %v", err))
|
|
|
|
}
|
|
|
|
c.Database, err = sql.Open("pgx", getAddr(c.Config.Database.TestDatabase))
|
|
|
|
if err != nil {
|
|
|
|
panic(fmt.Sprintf("failed to connect to database: %v", err))
|
|
|
|
}
|
2021-12-10 17:44:23 -08:00
|
|
|
}
|
2021-12-11 10:18:32 -08:00
|
|
|
}
|
2021-12-10 17:44:23 -08:00
|
|
|
|
2021-12-11 10:18:32 -08:00
|
|
|
func (c *Container) initORM() {
|
2021-12-10 17:44:23 -08:00
|
|
|
drv := entsql.OpenDB(dialect.Postgres, c.Database)
|
2021-12-11 10:18:32 -08:00
|
|
|
c.ORM = ent.NewClient(ent.Driver(drv))
|
|
|
|
if err := c.ORM.Schema.Create(context.Background()); err != nil {
|
2021-12-13 09:51:00 -08:00
|
|
|
panic(fmt.Sprintf("failed to create database schema: %v", err))
|
2021-12-10 17:44:23 -08:00
|
|
|
}
|
2021-12-11 10:18:32 -08:00
|
|
|
}
|
2021-12-14 08:29:45 -08:00
|
|
|
|
2021-12-15 06:29:43 -08:00
|
|
|
func (c *Container) initAuth() {
|
2021-12-17 17:58:51 -08:00
|
|
|
c.Auth = NewAuthClient(c.Config, c.ORM)
|
2021-12-15 06:29:43 -08:00
|
|
|
}
|
2021-12-19 17:09:01 -08:00
|
|
|
|
|
|
|
func (c *Container) initTemplateRenderer() {
|
2021-12-21 08:00:32 -08:00
|
|
|
c.TemplateRenderer = NewTemplateRenderer(c.Config)
|
2021-12-19 17:09:01 -08:00
|
|
|
}
|
2021-12-19 17:37:51 -08:00
|
|
|
|
|
|
|
func (c *Container) initMail() {
|
|
|
|
var err error
|
2021-12-21 08:00:32 -08:00
|
|
|
c.Mail, err = NewMailClient(c.Config, c.TemplateRenderer)
|
2021-12-19 17:37:51 -08:00
|
|
|
if err != nil {
|
|
|
|
panic(fmt.Sprintf("failed to create mail client: %v", err))
|
|
|
|
}
|
|
|
|
}
|