重构项目

cmd/server/main.go

@@ -0,0 +1,102 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"go.uber.org/zap"
+	"gopkg.d7z.net/gitea-pages/pkg"
+	"gopkg.d7z.net/gitea-pages/pkg/providers"
+	"gopkg.d7z.net/middleware/cache"
+	"gopkg.d7z.net/middleware/kv"
+)
+
+var (
+	configPath = "config-local.yaml"
+	debug      = false
+)
+
+func init() {
+	flag.StringVar(&configPath, "conf", configPath, "config file path")
+	flag.BoolVar(&debug, "debug", debug, "debug mode")
+	flag.Parse()
+}
+
+func main() {
+	call := logInject()
+	defer call()
+	config, err := LoadConfig(configPath)
+	if err != nil {
+		log.Fatalf("fail to load config file: %v", err)
+	}
+
+	gitea, err := providers.NewGitea(config.Auth.Server, config.Auth.Token)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	cacheMeta, err := kv.NewKVFromURL(config.Cache.Meta)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	defer cacheMeta.Close()
+	cacheBlob, err := cache.NewCacheFromURL(config.Cache.Blob)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	defer cacheBlob.Close()
+	backend := providers.NewProviderCache(gitea, cacheMeta, config.Cache.MetaTTL,
+		cacheBlob, uint64(config.Cache.BlobLimit),
+	)
+	defer backend.Close()
+	db, err := kv.NewKVFromURL(config.Database.URL)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	defer db.Close()
+	pageServer := pkg.NewPageServer(
+		http.DefaultClient,
+		backend,
+		config.Domain,
+		config.Page.DefaultBranch,
+		db,
+		cacheMeta,
+		config.Cache.MetaTTL,
+		config.ErrorHandler,
+	)
+	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM, syscall.SIGQUIT)
+	defer stop()
+
+	svc := http.Server{Addr: config.Bind, Handler: pageServer}
+	go func() {
+		<-ctx.Done()
+		zap.L().Debug("shutdown gracefully")
+		_ = svc.Close()
+	}()
+	_ = svc.ListenAndServe()
+}
+
+func logInject() func() {
+	atom := zap.NewAtomicLevel()
+	if debug {
+		atom.SetLevel(zap.DebugLevel)
+	} else {
+		atom.SetLevel(zap.InfoLevel)
+	}
+	cfg := zap.NewProductionConfig()
+	cfg.Level = atom
+
+	logger, _ := cfg.Build()
+	zap.ReplaceGlobals(logger)
+	zap.L().Debug("debug enabled")
+	return func() {
+		if err := logger.Sync(); err != nil {
+			fmt.Println(err)
+		}
+	}
+}