You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
152 lines
3.8 KiB
152 lines
3.8 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/mediocregopher/blog.mediocregopher.com/srv/api"
|
|
cfgpkg "github.com/mediocregopher/blog.mediocregopher.com/srv/cfg"
|
|
"github.com/mediocregopher/blog.mediocregopher.com/srv/chat"
|
|
"github.com/mediocregopher/blog.mediocregopher.com/srv/mailinglist"
|
|
"github.com/mediocregopher/blog.mediocregopher.com/srv/post"
|
|
"github.com/mediocregopher/blog.mediocregopher.com/srv/pow"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
|
|
"github.com/tilinna/clock"
|
|
)
|
|
|
|
func main() {
|
|
|
|
ctx := context.Background()
|
|
|
|
cfg := cfgpkg.NewBlogCfg(cfgpkg.Params{})
|
|
|
|
var dataDir cfgpkg.DataDir
|
|
dataDir.SetupCfg(cfg)
|
|
defer dataDir.Close()
|
|
ctx = mctx.WithAnnotator(ctx, &dataDir)
|
|
|
|
var powMgrParams pow.ManagerParams
|
|
powMgrParams.SetupCfg(cfg)
|
|
ctx = mctx.WithAnnotator(ctx, &powMgrParams)
|
|
|
|
var mailerParams mailinglist.MailerParams
|
|
mailerParams.SetupCfg(cfg)
|
|
ctx = mctx.WithAnnotator(ctx, &mailerParams)
|
|
|
|
var mlParams mailinglist.Params
|
|
mlParams.SetupCfg(cfg)
|
|
ctx = mctx.WithAnnotator(ctx, &mlParams)
|
|
|
|
var apiParams api.Params
|
|
apiParams.SetupCfg(cfg)
|
|
ctx = mctx.WithAnnotator(ctx, &apiParams)
|
|
|
|
var radixClient cfgpkg.RadixClient
|
|
radixClient.SetupCfg(cfg)
|
|
defer radixClient.Close()
|
|
ctx = mctx.WithAnnotator(ctx, &radixClient)
|
|
|
|
chatGlobalRoomMaxMsgs := cfg.Int("chat-global-room-max-messages", 1000, "Maximum number of messages the global chat room can retain")
|
|
chatUserIDCalcSecret := cfg.String("chat-user-id-calc-secret", "", "Secret to use when calculating user ids")
|
|
|
|
// initialization
|
|
err := cfg.Init(ctx)
|
|
|
|
logger := mlog.NewLogger(nil)
|
|
defer logger.Close()
|
|
|
|
logger.Info(ctx, "process started")
|
|
defer logger.Info(ctx, "process exiting")
|
|
|
|
if err != nil {
|
|
logger.Fatal(ctx, "initializing", err)
|
|
}
|
|
|
|
ctx = mctx.Annotate(ctx,
|
|
"chatGlobalRoomMaxMsgs", *chatGlobalRoomMaxMsgs,
|
|
)
|
|
|
|
clock := clock.Realtime()
|
|
|
|
powStore := pow.NewMemoryStore(clock)
|
|
defer powStore.Close()
|
|
|
|
powMgrParams.Store = powStore
|
|
powMgrParams.Clock = clock
|
|
|
|
powMgr := pow.NewManager(powMgrParams)
|
|
|
|
var mailer mailinglist.Mailer
|
|
if mailerParams.SMTPAddr == "" {
|
|
logger.Info(ctx, "-smtp-addr not given, using a fake Mailer")
|
|
mailer = mailinglist.NewLogMailer(logger.WithNamespace("fake-mailer"))
|
|
} else {
|
|
mailer = mailinglist.NewMailer(mailerParams)
|
|
}
|
|
|
|
mlStore, err := mailinglist.NewStore(dataDir)
|
|
if err != nil {
|
|
logger.Fatal(ctx, "initializing mailing list storage", err)
|
|
}
|
|
defer mlStore.Close()
|
|
|
|
mlParams.Store = mlStore
|
|
mlParams.Mailer = mailer
|
|
mlParams.Clock = clock
|
|
|
|
ml := mailinglist.New(mlParams)
|
|
|
|
chatGlobalRoom, err := chat.NewRoom(ctx, chat.RoomParams{
|
|
Logger: logger.WithNamespace("global-chat-room"),
|
|
Redis: radixClient.Client,
|
|
ID: "global",
|
|
MaxMessages: *chatGlobalRoomMaxMsgs,
|
|
})
|
|
if err != nil {
|
|
logger.Fatal(ctx, "initializing global chat room", err)
|
|
}
|
|
defer chatGlobalRoom.Close()
|
|
|
|
chatUserIDCalc := chat.NewUserIDCalculator([]byte(*chatUserIDCalcSecret))
|
|
|
|
postSQLDB, err := post.NewSQLDB(dataDir)
|
|
if err != nil {
|
|
logger.Fatal(ctx, "initializing sql db for post data", err)
|
|
}
|
|
defer postSQLDB.Close()
|
|
|
|
postStore := post.NewStore(postSQLDB)
|
|
|
|
apiParams.Logger = logger.WithNamespace("api")
|
|
apiParams.PowManager = powMgr
|
|
apiParams.PostStore = postStore
|
|
apiParams.MailingList = ml
|
|
apiParams.GlobalRoom = chatGlobalRoom
|
|
apiParams.UserIDCalculator = chatUserIDCalc
|
|
|
|
logger.Info(ctx, "listening")
|
|
a, err := api.New(apiParams)
|
|
if err != nil {
|
|
logger.Fatal(ctx, "initializing api", err)
|
|
}
|
|
defer func() {
|
|
shutdownCtx, cancel := context.WithTimeout(ctx, 5*time.Second)
|
|
defer cancel()
|
|
|
|
if err := a.Shutdown(shutdownCtx); err != nil {
|
|
logger.Fatal(ctx, "shutting down api", err)
|
|
}
|
|
}()
|
|
|
|
// wait
|
|
|
|
sigCh := make(chan os.Signal, 1)
|
|
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
|
<-sigCh
|
|
|
|
// let the defers begin
|
|
}
|
|
|