go-svc-configure/main.go

package main

import (
	"context"
	"log"

	"git.x2erp.com/qdy/go-base/authbase"
	"git.x2erp.com/qdy/go-base/config"
	"git.x2erp.com/qdy/go-base/container"
	"git.x2erp.com/qdy/go-base/ctx"
	"git.x2erp.com/qdy/go-base/graceful"
	"git.x2erp.com/qdy/go-base/logger"
	"git.x2erp.com/qdy/go-base/model/request/configreq"
	"git.x2erp.com/qdy/go-base/model/response"
	"git.x2erp.com/qdy/go-base/webx"
	"git.x2erp.com/qdy/go-base/webx/router"
	"git.x2erp.com/qdy/go-db/factory/database"

	"git.x2erp.com/qdy/go-db/sqldef"

	"git.x2erp.com/qdy/go-svc-configure/internal/service/project"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/register"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/role"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/startup"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/svcstartup"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/tenant"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/token"
	"git.x2erp.com/qdy/go-svc-configure/internal/tables"
	_ "git.x2erp.com/qdy/go-svc-configure/internal/tables" // 导入表定义包，触发 init() 函数
)

var (
	appName    = "svc-configure"
	appVersion = "1"
)

func main() {

	// 0. 初始化日志
	logBootFactory := logger.InitBootLog()

	// 1. 获取配置文件
	cfg := config.GetConfig()
	cfg.SetAppName(appName)
	cfg.SetAppVersion(appVersion)

	// 2. 创建关闭容器
	ctr := container.NewContainer(cfg)

	// 注册日志，实现自动关闭
	container.Reg(ctr, logBootFactory)

	// 3. 创建数据库工厂
	dbFactory := container.Create(ctr, database.CreateDBFactory)
	dbFactory.TestConnection()
	// 创建表
	creteTabel(dbFactory)

	//启用运行日志
	container.Create(ctr, logger.InitRuntimeLogger)

	//建立httpservice
	//得到webservice服务工厂
	webxFactory := webx.GetWebServiceFactory()

	//建立hhtpService服务
	webServcie, _ := webxFactory.CreateService(cfg.GetServiceConfig())

	//建立路由-api
	routerService := router.NewWebService(webServcie.GetRouter())
	//注册路由--api
	registerRoutes(routerService, dbFactory)

	//启动服务
	webServcie.Run()

	//等待关闭
	graceful.WaitForShutdown(cfg.GetServiceConfig().ServiceName, ctr, webServcie.GetServer())

}

// 注册所有路由
func registerRoutes(ws *router.RouterService, dbFactory *database.DBFactory) {
	// 建立访问配置中心Token
	ws.POST("/api/create/config/token",
		func(tokenRequest *configreq.ConfigTokenRequest, reqCtx *ctx.RequestContext) (*response.QueryResult[string], error) {
			return token.CreateConfigureToken(tokenRequest, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("建立访问配置中心的token").Register()

	// svc
	ws.POST("/api/delete/config/startup/{servicename}",
		func(servicename string, req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return svcstartup.DeleteSVCtartupConfig(servicename, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Register()

	ws.POST("/api/create/config/startup/{servicename}",
		func(servicename string, req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return svcstartup.CreateStartupSVCConfig(servicename, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Register()

	// startup
	ws.POST("/api/delete/config/startup",
		func(req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return startup.DeleteStartupConfig(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Register()

	ws.POST("/api/create/config/startup",
		func(req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return startup.CreateStartupConfig(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Register()

	//init
	ws.POST("/api/init/config/meta",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[map[string]interface{}], error) {
			return register.RegisterConfigMeta(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Register()

	// project
	ws.POST("/api/create/config/project",
		func(req *configreq.ProjectRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return project.CreateProject(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建项目").Register()

	ws.POST("/api/update/config/project/{project_id}",
		func(projectID string, req *configreq.ProjectRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return project.UpdateProject(projectID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新项目").Register()

	ws.POST("/api/delete/config/project/{project_id}",
		func(projectID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return project.DeleteProject(projectID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除项目").Register()

	ws.POST("/api/query/config/project/{project_id}",
		func(projectID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.ProjectDB], error) {
			return project.GetProject(projectID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目").Register()

	ws.POST("/api/query/config/projects",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.ProjectDB], error) {
			return project.ListProjects(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目列表").Register()

	// tenant
	ws.POST("/api/create/config/tenant",
		func(req *configreq.TenantRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return tenant.CreateTenant(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建租户").Register()

	ws.POST("/api/update/config/tenant/{tenant_id}",
		func(tenantID string, req *configreq.TenantRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return tenant.UpdateTenant(tenantID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新租户").Register()

	ws.POST("/api/delete/config/tenant/{tenant_id}",
		func(tenantID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return tenant.DeleteTenant(tenantID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除租户").Register()

	ws.POST("/api/query/config/tenant/{tenant_id}",
		func(tenantID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.TenantDB], error) {
			return tenant.GetTenant(tenantID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询租户").Register()

	ws.POST("/api/query/config/tenants",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.TenantDB], error) {
			return tenant.ListTenants(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询租户列表").Register()

	// role
	ws.POST("/api/create/config/role",
		func(req *configreq.RoleRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return role.CreateRole(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建角色").Register()

	ws.POST("/api/update/config/role/{role_id}",
		func(roleID string, req *configreq.RoleRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return role.UpdateRole(roleID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新角色").Register()

	ws.POST("/api/delete/config/role/{role_id}",
		func(roleID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return role.DeleteRole(roleID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除角色").Register()

	ws.POST("/api/query/config/role/{role_id}",
		func(roleID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.RoleDB], error) {
			return role.GetRole(roleID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询角色").Register()

	ws.POST("/api/query/config/roles",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.RoleDB], error) {
			return role.ListRoles(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询角色列表").Register()

}

func creteTabel(factory *database.DBFactory) {

	// 获取数据库连接和类型
	db := factory.GetDB()
	dbType := factory.GetDBType()

	// 创建表同步器
	syncer, err := sqldef.NewTableSyncer(db, dbType)
	if err != nil {
		log.Printf("创建 - 建立器失败: %v", err)
	}

	// 创建表
	if err := syncer.CreateTables(); err != nil {
		log.Printf("建表失败: %v", err)
	}

	log.Println("数据库表建立完成！")
}