67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.118im.com/im_team/im_workbench/internal/accounts"
|
|
"git.118im.com/im_team/im_workbench/internal/auth"
|
|
"git.118im.com/im_team/im_workbench/internal/config"
|
|
"git.118im.com/im_team/im_workbench/internal/finance"
|
|
"git.118im.com/im_team/im_workbench/internal/httpapi"
|
|
"git.118im.com/im_team/im_workbench/internal/mall"
|
|
"git.118im.com/im_team/im_workbench/internal/portal"
|
|
"git.118im.com/im_team/im_workbench/internal/store"
|
|
"git.118im.com/im_team/im_workbench/internal/tasks"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type App struct {
|
|
store *store.Store
|
|
redis *redis.Client
|
|
api *httpapi.API
|
|
}
|
|
|
|
func New(cfg *config.Config) (*App, error) {
|
|
st, err := store.Open(cfg.MySQL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
redisClient := redis.NewClient(&redis.Options{
|
|
Addr: cfg.Redis.Addr, Username: cfg.Redis.Username, Password: cfg.Redis.Password, DB: cfg.Redis.DB,
|
|
})
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
if err := redisClient.Ping(ctx).Err(); err != nil {
|
|
_ = redisClient.Close()
|
|
_ = st.Close()
|
|
return nil, err
|
|
}
|
|
authenticator := auth.New(cfg.Auth, redisClient)
|
|
accountService := accounts.New(st, redisClient, cfg.Auth)
|
|
portalService := portal.New(st)
|
|
taskService := tasks.New(st)
|
|
mallService := mall.New(st)
|
|
financeService := finance.New(st, redisClient, cfg.Auth)
|
|
return &App{store: st, redis: redisClient, api: httpapi.New(authenticator, accountService, portalService, taskService, mallService, financeService)}, nil
|
|
}
|
|
|
|
func (a *App) Handler() http.Handler {
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/healthz", a.api.Healthz)
|
|
mux.Handle("/metrics", promhttp.Handler())
|
|
a.api.Register(mux)
|
|
return mux
|
|
}
|
|
|
|
func (a *App) Close() {
|
|
if a.redis != nil {
|
|
_ = a.redis.Close()
|
|
}
|
|
if a.store != nil {
|
|
_ = a.store.Close()
|
|
}
|
|
}
|