数据源和 oss

2026-08-13 23:17:55 +08:00
parent f1a0b17bb0
commit c4bf7ee03b
40 changed files with 1975 additions and 922 deletions
@@ -0,0 +1,99 @@
+package main
+
+import (
+	"flag"
+	"log/slog"
+	"os"
+	"path/filepath"
+
+	"kra/internal/conf"
+	"kra/internal/server"
+
+	"github.com/go-kratos/kratos/contrib/otel/v3/tracing"
+	"github.com/go-kratos/kratos/v3"
+	"github.com/go-kratos/kratos/v3/config"
+	"github.com/go-kratos/kratos/v3/config/file"
+	"github.com/go-kratos/kratos/v3/log"
+	"github.com/go-kratos/kratos/v3/transport/grpc"
+
+	_ "go.uber.org/automaxprocs"
+)
+
+// go build -ldflags "-X main.Version=x.y.z"
+var (
+	// Name is the name of the compiled software.
+	Name string
+	// Version is the version of the compiled software.
+	Version string
+	// flagconf is the config flag.
+	flagconf string
+
+	id, _ = os.Hostname()
+)
+
+func init() {
+	flag.StringVar(&flagconf, "conf", "../../configs", "config path, eg: -conf config.yaml")
+}
+
+func newApp(logger *slog.Logger, gs *grpc.Server, hs *server.GinServer, scheduler *server.TaskScheduler) *kratos.App {
+	return kratos.New(
+		kratos.ID(id),
+		kratos.Name(Name),
+		kratos.Version(Version),
+		kratos.Metadata(map[string]string{}),
+		kratos.Logger(logger),
+		kratos.Server(
+			gs,
+			hs,
+			scheduler,
+		),
+	)
+}
+
+func main() {
+	flag.Parse()
+	logger := log.NewLogger(
+		slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
+			AddSource: true,
+			Level:     slog.LevelInfo,
+		}),
+		log.WithExtractor(tracing.TraceAttrs),
+	).With(
+		slog.String("service.id", id),
+		slog.String("service.name", Name),
+		slog.String("service.version", Version),
+	)
+	log.SetDefault(logger)
+	c := config.New(
+		config.WithSource(
+			file.NewSource(flagconf),
+		),
+	)
+	defer c.Close()
+
+	if err := c.Load(); err != nil {
+		panic(err)
+	}
+
+	var bc conf.Bootstrap
+	if err := c.Scan(&bc); err != nil {
+		panic(err)
+	}
+	if bc.Admin != nil {
+		bc.Admin.ConfigPath = flagconf
+		if info, err := os.Stat(flagconf); err == nil && info.IsDir() {
+			bc.Admin.ConfigPath = filepath.Join(flagconf, "config.yaml")
+		}
+	}
+
+	app, cleanup, err := wireApp(bc.Server, bc.Data, bc.Admin, logger)
+	if err != nil {
+		panic(err)
+	}
+	defer cleanup()
+
+	// start and wait for stop signal
+	if err := app.Run(); err != nil {
+		panic(err)
+	}
+}
@@ -0,0 +1,24 @@
+//go:build wireinject
+// +build wireinject
+
+// The build tag makes sure the stub is not built in the final build.
+
+package main
+
+import (
+	"log/slog"
+
+	"kra/internal/biz"
+	"kra/internal/conf"
+	"kra/internal/data"
+	"kra/internal/server"
+	"kra/internal/service"
+
+	"github.com/go-kratos/kratos/v3"
+	"github.com/google/wire"
+)
+
+// wireApp init kratos application.
+func wireApp(*conf.Server, *conf.Data, *conf.AdminBackend, *slog.Logger) (*kratos.App, func(), error) {
+	panic(wire.Build(server.ProviderSet, data.ProviderSet, biz.ProviderSet, service.ProviderSet, newApp))
+}
@@ -0,0 +1,77 @@
+// Code generated by Wire. DO NOT EDIT.
+
+//go:generate go run -mod=mod github.com/google/wire/cmd/wire
+//go:build !wireinject
+// +build !wireinject
+
+package main
+
+import (
+	"github.com/go-kratos/kratos/v3"
+	"kra/internal/biz"
+	"kra/internal/conf"
+	"kra/internal/data"
+	"kra/internal/server"
+	"kra/internal/service"
+	"log/slog"
+)
+
+import (
+	_ "go.uber.org/automaxprocs"
+)
+
+// Injectors from wire.go:
+
+// wireApp init kratos application.
+func wireApp(confServer *conf.Server, confData *conf.Data, adminBackend *conf.AdminBackend, logger *slog.Logger) (*kratos.App, func(), error) {
+	dataData, cleanup, err := data.NewData(confData, adminBackend)
+	if err != nil {
+		return nil, nil, err
+	}
+	adminRepo := data.NewAdminRepo(dataData)
+	adminUsecase := biz.NewAdminUsecase(adminRepo)
+	adminService := service.NewAdminService(adminUsecase)
+	grpcServer := server.NewGRPCServer(confServer, adminService)
+	systemRepo := data.NewSystemRepo(dataData)
+	cache := data.NewCache(dataData)
+	fileStorage, err := data.NewFileStorage(adminBackend)
+	if err != nil {
+		cleanup()
+		return nil, nil, err
+	}
+	systemUsecase := biz.NewSystemUsecase(systemRepo, cache, fileStorage)
+	settingsRepo := data.NewSettingsRepo(dataData)
+	settingsUsecase := biz.NewSettingsUsecase(settingsRepo)
+	settingsService := service.NewSettingsService(settingsUsecase, adminBackend)
+	systemService := service.NewSystemService(systemUsecase, adminBackend, settingsService)
+	accessRepo := data.NewAccessRepo(dataData)
+	accessUsecase := biz.NewAccessUsecase(accessRepo)
+	accessService := service.NewAccessService(accessUsecase)
+	versionRepo := data.NewVersionRepo(dataData)
+	versionUsecase := biz.NewVersionUsecase(versionRepo)
+	versionService := service.NewVersionService(versionUsecase)
+	exportRepo := data.NewExportRepo(dataData)
+	exportUsecase := biz.NewExportUsecase(exportRepo)
+	exportService := service.NewExportService(exportUsecase)
+	auditRepo := data.NewAuditRepo(dataData)
+	auditUsecase := biz.NewAuditUsecase(auditRepo)
+	auditService := service.NewAuditService(auditUsecase)
+	taskRepo := data.NewTaskRepo(dataData)
+	taskUsecase := biz.NewTaskUsecase(taskRepo)
+	taskService := service.NewTaskService(taskUsecase)
+	mediaRepo := data.NewMediaRepo(dataData)
+	mediaUsecase := biz.NewMediaUsecase(mediaRepo, fileStorage)
+	mediaService := service.NewMediaService(mediaUsecase)
+	announcementRepo := data.NewAnnouncementRepo(dataData)
+	announcementUsecase := biz.NewAnnouncementUsecase(announcementRepo)
+	announcementService := service.NewAnnouncementService(announcementUsecase)
+	emailRepo := data.NewEmailRepo(adminBackend)
+	emailUsecase := biz.NewEmailUsecase(emailRepo)
+	emailService := service.NewEmailService(emailUsecase)
+	taskScheduler := server.NewTaskScheduler(taskService, logger)
+	ginServer := server.NewGinServer(confServer, adminBackend, systemService, accessService, settingsService, versionService, exportService, auditService, taskService, mediaService, announcementService, emailService, taskScheduler, logger)
+	app := newApp(logger, grpcServer, ginServer, taskScheduler)
+	return app, func() {
+		cleanup()
+	}, nil
+}