124 lines
4.7 KiB
Go
124 lines
4.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log"
|
|
"net/http"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
_ "time/tzdata"
|
|
|
|
"chatapp3-golang/internal/bootstrap"
|
|
"chatapp3-golang/internal/router"
|
|
"chatapp3-golang/internal/service/baishun"
|
|
"chatapp3-golang/internal/service/gameopen"
|
|
"chatapp3-golang/internal/service/gameprovider"
|
|
"chatapp3-golang/internal/service/hostcenter"
|
|
"chatapp3-golang/internal/service/invite"
|
|
"chatapp3-golang/internal/service/lingxian"
|
|
"chatapp3-golang/internal/service/luckygift"
|
|
"chatapp3-golang/internal/service/rechargeagency"
|
|
"chatapp3-golang/internal/service/rechargereward"
|
|
"chatapp3-golang/internal/service/registerreward"
|
|
"chatapp3-golang/internal/service/roomturnoverreward"
|
|
"chatapp3-golang/internal/service/signinreward"
|
|
"chatapp3-golang/internal/service/vip"
|
|
"chatapp3-golang/internal/service/weekstar"
|
|
)
|
|
|
|
func main() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
app, err := bootstrap.New(ctx)
|
|
if err != nil {
|
|
log.Fatalf("bootstrap api failed: %v", err)
|
|
}
|
|
|
|
inviteService := invite.NewInviteService(app.Config, app.Repository.DB, app.Repository.Redis, &app.Gateways)
|
|
baishunService := baishun.NewBaishunService(app.Config, app.Repository.DB, app.Repository.Redis, &app.Gateways)
|
|
lingxianService := lingxian.NewService(app.Config, app.Repository.DB, app.Repository.Redis)
|
|
gameOpenService := gameopen.NewGameOpenService(app.Config, app.Repository.DB, &app.Gateways)
|
|
luckyGiftService := luckygift.NewLuckyGiftService(app.Config, app.Repository.DB, app.Repository.Redis, app.Gateways.Wallet)
|
|
rechargeAgencyService := rechargeagency.NewService(app.Config, app.Repository.DB, &app.Gateways)
|
|
registerRewardService := registerreward.NewRegisterRewardService(app.Config, app.Repository.DB, app.Repository.Redis, &app.Gateways)
|
|
rechargeRewardService := rechargereward.NewService(app.Config, app.Repository.DB, &app.Gateways)
|
|
signInRewardService := signinreward.NewSignInRewardService(app.Config, app.Repository.DB, app.Repository.Redis, &app.Gateways)
|
|
vipService := vip.NewService(app.Config, app.Repository.DB, &app.Gateways)
|
|
weekStarService := weekstar.NewWeekStarService(app.Config, app.Repository.DB, app.Repository.Redis, &app.Gateways)
|
|
roomTurnoverRewardService := roomturnoverreward.NewService(app.Config, app.Repository.DB, &app.Gateways)
|
|
hostCenterService := hostcenter.NewService(app.Config, &app.Gateways)
|
|
|
|
workerCtx, workerCancel := context.WithCancel(context.Background())
|
|
defer workerCancel()
|
|
if err := registerRewardService.Start(workerCtx); err != nil {
|
|
log.Fatalf("start register reward consumer failed: %v", err)
|
|
}
|
|
if err := weekStarService.StartMessageConsumer(workerCtx); err != nil {
|
|
log.Fatalf("start week star message consumer failed: %v", err)
|
|
}
|
|
|
|
gameProviders := gameprovider.NewRegistry(
|
|
baishun.NewAppProvider(baishunService),
|
|
lingxian.NewAppProvider(lingxianService),
|
|
)
|
|
engine := router.NewRouter(app.Config, app.Repository, &app.Gateways, router.Services{
|
|
Invite: inviteService,
|
|
Baishun: baishunService,
|
|
Lingxian: lingxianService,
|
|
GameOpen: gameOpenService,
|
|
GameProviders: gameProviders,
|
|
LuckyGift: luckyGiftService,
|
|
HostCenter: hostCenterService,
|
|
RechargeAgency: rechargeAgencyService,
|
|
RechargeReward: rechargeRewardService,
|
|
RegisterReward: registerRewardService,
|
|
RoomTurnoverReward: roomTurnoverRewardService,
|
|
SignInReward: signInRewardService,
|
|
VIP: vipService,
|
|
WeekStar: weekStarService,
|
|
})
|
|
server := &http.Server{
|
|
Addr: app.Config.HTTP.ListenAddr,
|
|
Handler: engine,
|
|
ReadHeaderTimeout: 10 * time.Second,
|
|
}
|
|
serverErrors := make(chan error, 1)
|
|
go func() {
|
|
log.Printf("chatapp3-golang api listening on %s", app.Config.HTTP.ListenAddr)
|
|
if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
serverErrors <- err
|
|
return
|
|
}
|
|
serverErrors <- nil
|
|
}()
|
|
|
|
shutdownSignal, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
select {
|
|
case err := <-serverErrors:
|
|
if err != nil {
|
|
log.Fatalf("run server failed: %v", err)
|
|
}
|
|
return
|
|
case <-shutdownSignal.Done():
|
|
log.Printf("shutdown signal received, draining http traffic")
|
|
workerCancel()
|
|
}
|
|
|
|
shutdownCtx, cancelShutdown := context.WithTimeout(context.Background(), app.Config.HTTP.ShutdownTimeout)
|
|
defer cancelShutdown()
|
|
if err := server.Shutdown(shutdownCtx); err != nil {
|
|
log.Printf("graceful shutdown failed: %v", err)
|
|
if closeErr := server.Close(); closeErr != nil {
|
|
log.Printf("force close failed: %v", closeErr)
|
|
}
|
|
}
|
|
if err := <-serverErrors; err != nil {
|
|
log.Fatalf("server exited with error during shutdown: %v", err)
|
|
}
|
|
}
|