2022-10-22 17:38:39 +08:00
|
|
|
package server
|
2022-09-07 23:34:37 +08:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2023-12-18 20:55:37 +08:00
|
|
|
"errors"
|
2022-09-07 23:34:37 +08:00
|
|
|
"fmt"
|
2023-07-14 21:47:11 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/api/consumer"
|
2023-07-15 16:19:49 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/misc/config"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/misc/log"
|
2023-07-14 21:47:11 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/model"
|
2023-07-15 16:19:49 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/repo/db"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/web/router"
|
2023-07-14 21:47:11 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/pkg/utils"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/pkg/zapasynq"
|
2022-10-22 17:38:39 +08:00
|
|
|
"github.com/hibiken/asynq"
|
2023-07-15 16:19:49 +08:00
|
|
|
"github.com/samber/do"
|
2022-09-07 23:34:37 +08:00
|
|
|
"go.uber.org/zap"
|
|
|
|
"net/http"
|
|
|
|
"os"
|
|
|
|
"os/signal"
|
2022-10-22 17:38:39 +08:00
|
|
|
"runtime"
|
2022-09-07 23:34:37 +08:00
|
|
|
"syscall"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
2023-12-19 16:30:58 +08:00
|
|
|
func RunServerMigrate(i *do.Injector) error {
|
2023-12-19 01:09:02 +08:00
|
|
|
slog := do.MustInvoke[log.Service](i).GetLogger("app.server")
|
|
|
|
|
|
|
|
// Migrate and shutdown database
|
|
|
|
err := do.MustInvoke[db.Service](i).Close()
|
|
|
|
if err != nil {
|
|
|
|
slog.Warn("Database Close Failed", zap.Error(err))
|
|
|
|
}
|
|
|
|
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-12-18 20:55:37 +08:00
|
|
|
func RunServer(i *do.Injector) error {
|
2023-07-15 16:19:49 +08:00
|
|
|
conf := do.MustInvoke[config.Service](i).GetConfig()
|
|
|
|
slog := do.MustInvoke[log.Service](i).GetLogger("app.server")
|
2022-09-07 23:34:37 +08:00
|
|
|
|
2023-12-18 20:55:37 +08:00
|
|
|
// Prepare Router
|
2023-07-15 16:19:49 +08:00
|
|
|
routers := do.MustInvoke[router.Service](i).GetRouter()
|
2022-09-07 23:34:37 +08:00
|
|
|
|
2022-09-08 22:00:25 +08:00
|
|
|
// Create Server
|
2023-07-15 16:19:49 +08:00
|
|
|
addr := fmt.Sprintf("%s:%d", conf.WebServer.Address, conf.WebServer.Port)
|
2022-09-07 23:34:37 +08:00
|
|
|
server := &http.Server{
|
|
|
|
Addr: addr,
|
2022-10-22 17:38:39 +08:00
|
|
|
Handler: routers,
|
2022-09-07 23:34:37 +08:00
|
|
|
}
|
|
|
|
|
2022-09-08 22:00:25 +08:00
|
|
|
// Run Server
|
2022-09-07 23:34:37 +08:00
|
|
|
go func() {
|
2023-12-18 20:55:37 +08:00
|
|
|
if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
2023-07-15 16:19:49 +08:00
|
|
|
slog.Fatal("ListenAndServe Failed", zap.Error(err))
|
2022-09-07 23:34:37 +08:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2022-10-22 17:38:39 +08:00
|
|
|
// Create Queue
|
|
|
|
queueMux := asynq.NewServeMux()
|
2022-10-23 17:29:35 +08:00
|
|
|
{
|
2023-07-15 16:19:49 +08:00
|
|
|
handler := consumer.NewConsumer(i)
|
2022-10-23 17:29:35 +08:00
|
|
|
queueMux.HandleFunc(model.TypeProblemUpdate, handler.ProblemUpdate)
|
|
|
|
queueMux.HandleFunc(model.TypeSubmitUpdate, handler.SubmitUpdate)
|
|
|
|
}
|
2022-10-22 17:38:39 +08:00
|
|
|
queueSrv := asynq.NewServer(
|
|
|
|
asynq.RedisClientOpt{
|
2023-12-18 23:43:27 +08:00
|
|
|
Addr: fmt.Sprintf("%s:%d", conf.Redis.Address, conf.Redis.Port),
|
2023-07-15 16:19:49 +08:00
|
|
|
Password: conf.Redis.Password,
|
|
|
|
DB: conf.Redis.QueueDb,
|
2022-10-22 17:38:39 +08:00
|
|
|
},
|
|
|
|
asynq.Config{
|
2022-10-30 22:40:59 +08:00
|
|
|
Concurrency: utils.If(runtime.NumCPU() > 1, runtime.NumCPU()-1, 1),
|
2023-07-15 16:19:49 +08:00
|
|
|
Logger: zapasynq.New(slog),
|
2022-10-22 17:38:39 +08:00
|
|
|
Queues: map[string]int{model.QueueServer: 1},
|
|
|
|
},
|
|
|
|
)
|
|
|
|
|
|
|
|
// Run Queue
|
|
|
|
if err := queueSrv.Start(queueMux); err != nil {
|
2023-07-15 16:19:49 +08:00
|
|
|
slog.Fatal("queueSrv.Start Failed", zap.Error(err))
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|
|
|
|
|
2022-09-07 23:34:37 +08:00
|
|
|
// Handle SIGINT and SIGTERM.
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
<-quit
|
2023-07-15 16:19:49 +08:00
|
|
|
slog.Info("Shutting down server ...")
|
2022-09-07 23:34:37 +08:00
|
|
|
|
2022-10-22 17:38:39 +08:00
|
|
|
// Graceful Shutdown Server
|
2022-09-07 23:34:37 +08:00
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
|
|
defer cancel()
|
|
|
|
err := server.Shutdown(ctx)
|
|
|
|
if err != nil {
|
2023-07-15 16:19:49 +08:00
|
|
|
slog.Warn("Server Shutdown Failed", zap.Error(err))
|
2022-09-08 22:00:25 +08:00
|
|
|
}
|
2022-10-22 17:38:39 +08:00
|
|
|
|
|
|
|
// Graceful Shutdown Queue
|
|
|
|
queueSrv.Shutdown()
|
|
|
|
|
|
|
|
// Graceful Shutdown Database
|
2023-07-15 16:19:49 +08:00
|
|
|
err = do.MustInvoke[db.Service](i).Close()
|
2022-09-08 22:00:25 +08:00
|
|
|
if err != nil {
|
2023-07-15 16:19:49 +08:00
|
|
|
slog.Warn("Database Close Failed", zap.Error(err))
|
2022-09-07 23:34:37 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
return err
|
|
|
|
}
|