122 lines
3.8 KiB
Go
122 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/longpeng/configcenter/internal/api/httpapi"
|
|
"github.com/longpeng/configcenter/internal/auth"
|
|
"github.com/longpeng/configcenter/internal/config"
|
|
"github.com/longpeng/configcenter/internal/metrics"
|
|
"github.com/longpeng/configcenter/internal/outbox"
|
|
runtimepkg "github.com/longpeng/configcenter/internal/runtime"
|
|
etcdstore "github.com/longpeng/configcenter/internal/runtime/etcd"
|
|
memoryruntime "github.com/longpeng/configcenter/internal/runtime/memory"
|
|
storepkg "github.com/longpeng/configcenter/internal/store"
|
|
memory_store "github.com/longpeng/configcenter/internal/store/memory"
|
|
postgres_store "github.com/longpeng/configcenter/internal/store/postgres"
|
|
"github.com/longpeng/configcenter/internal/watch"
|
|
)
|
|
|
|
func main() {
|
|
migrateOnly := flag.Bool("migrate-only", false, "apply database migrations and exit")
|
|
flag.Parse()
|
|
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelInfo}))
|
|
cfg := config.Load()
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
repository, err := openStore(ctx, cfg, logger)
|
|
if err != nil {
|
|
logger.Error("open control-plane store", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
defer repository.Close()
|
|
if *migrateOnly {
|
|
logger.Info("database migrations applied")
|
|
return
|
|
}
|
|
authorizer, err := auth.New(repository, cfg.AuthEnabled, cfg.JWTSecret, cfg.JWTTTL)
|
|
if err != nil {
|
|
logger.Error("initialize authentication", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
if err := authorizer.Bootstrap(ctx, cfg.BootstrapAdminUsername, cfg.BootstrapAdminPassword, cfg.BootstrapAdminDisplayName); err != nil {
|
|
logger.Error("bootstrap administrator", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
runtimeStore, err := openRuntime(cfg, logger)
|
|
if err != nil {
|
|
logger.Error("open runtime store", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
defer runtimeStore.Close() //nolint:errcheck
|
|
|
|
hub := watch.New(ctx, runtimeStore)
|
|
collector := metrics.New()
|
|
worker := outbox.New(repository, runtimeStore, cfg.OutboxInterval, cfg.OutboxBatchSize, cfg.OutboxMaxRetry, collector, logger)
|
|
go worker.Run(ctx)
|
|
|
|
api := httpapi.New(repository, runtimeStore, hub, authorizer, collector, cfg.AllowedOrigins, logger)
|
|
server := &http.Server{
|
|
Addr: cfg.HTTPAddr,
|
|
Handler: api.Handler(),
|
|
ReadHeaderTimeout: 5 * time.Second,
|
|
ReadTimeout: 15 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
serverErrors := make(chan error, 1)
|
|
go func() {
|
|
logger.Info("config center listening", "address", cfg.HTTPAddr)
|
|
serverErrors <- server.ListenAndServe()
|
|
}()
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
logger.Info("shutting down config center")
|
|
case err := <-serverErrors:
|
|
if err != nil && err != http.ErrServerClosed {
|
|
logger.Error("http server stopped", "error", err)
|
|
}
|
|
}
|
|
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), cfg.ShutdownTimeout)
|
|
defer cancel()
|
|
if err := server.Shutdown(shutdownCtx); err != nil {
|
|
logger.Error("graceful shutdown", "error", err)
|
|
}
|
|
}
|
|
|
|
func openStore(ctx context.Context, cfg config.Config, logger *slog.Logger) (storepkg.Store, error) {
|
|
if cfg.DatabaseURL == "" {
|
|
logger.Warn("DATABASE_URL is empty; using the in-memory control-plane store")
|
|
return memory_store.New(true), nil
|
|
}
|
|
repository, err := postgres_store.Open(ctx, cfg.DatabaseURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := repository.Migrate(ctx); err != nil {
|
|
repository.Close()
|
|
return nil, err
|
|
}
|
|
return repository, nil
|
|
}
|
|
|
|
func openRuntime(cfg config.Config, logger *slog.Logger) (runtimepkg.Store, error) {
|
|
if len(cfg.EtcdEndpoints) == 0 {
|
|
logger.Warn("ETCD_ENDPOINTS is empty; using the in-memory runtime store")
|
|
return memoryruntime.New(), nil
|
|
}
|
|
return etcdstore.Open(cfg.EtcdEndpoints, cfg.EtcdDialTimeout)
|
|
}
|