支持多微服务和多数据库测试通过

main.go

@@ -3,69 +3,221 @@ package main
 import (
 	"log"
 
-	"git.x2erp.com/qdy/go-base/bootstraps"
+	"git.x2erp.com/qdy/go-db/dbstart"
 	"git.x2erp.com/qdy/go-db/factory/database"
-	"git.x2erp.com/qdy/go-db/sqldef"
-	"go-micro.dev/v4/web"
 
-	_ "git.x2erp.com/qdy/go-svc-mercury/tables" // 导入表定义包，触发 init() 函数
+	"git.x2erp.com/qdy/go-base/bootstraps"
+	"git.x2erp.com/qdy/go-base/ctx"
+	"git.x2erp.com/qdy/go-base/logger"
+	"git.x2erp.com/qdy/go-base/middleware"
+	"git.x2erp.com/qdy/go-base/webx"
 )
 
 var (
-	serviceName    = "svc-configure"
+	serviceName    = "svc-ai"
 	serviceVersion = "1"
 )
 
+// 定义结构体
+type CreateUserRequest struct {
+	Name  string `json:"name"`
+	Email string `json:"email"`
+	Age   int    `json:"age"`
+}
+
+type UserResponse struct {
+	ID    string `json:"id"`
+	Name  string `json:"name"`
+	Email string `json:"email"`
+	Data  []map[string]interface{}
+}
+
 func main() {
+	// 1. 初始化基础启动器
+	boot := bootstraps.NewBootstrapper(serviceName, serviceVersion)
 
-	// 创建服务启动器
-	bootstrapper := bootstraps.NewServiceBootstrapper(serviceName, serviceVersion)
+	// 2. 初始化数据库（都在bootstraps包中）
+	dbBoot := dbstart.NewDBBootstrapper(boot.GetConfig())
+	dbBoot.Init()
 
-	//加载配置
-	bootstrapper.InitConfig()
+	agentDB := dbBoot.GetDBFactory("agent")
+	// 赋值认证中间件参数
+	middleware.JWTAuthMiddlewareInit(boot.GetConfig())
 
-	//构建数据库工厂
-	bootstrapper.InitDatabase()
+	// 3. 启动服务
+	//boot.StartService("default")
 
-	//创建表到数据库
-	creteTabel(bootstrapper.DbFactory)
+	router := boot.GetRouter("default")
+	// 3. 创建 WebService 并传入数据库接口
+	ws := webx.NewWebService(router)
 
-	// 启动服务，传入路由注册函数
-	bootstrapper.Run(registerRoutes)
+	//使用默认数据库
+	registerDefaultRouter(ws, dbBoot.GetDefaultDBFactory())
 
+	//启动第2个服务
+	routerDoris := boot.GetRouter("doris")
+	wsDoris := webx.NewWebService(routerDoris)
+	registerDorisRouter(wsDoris, agentDB)
+
+	// 6. 运行服务
+	boot.Run(dbBoot)
 }
 
-// 注册所有路由
-func registerRoutes(webService web.Service, dbFactory *database.DBFactory) {
+func registerDorisRouter(ws *webx.WebService, dbFactory *database.DBFactory) {
+	// GET示例：路径参数绑定
+	ws.GET("/app/users/{id}",
+		func(id string, reqCtx *ctx.RequestContext) (UserResponse, error) {
+
+			log.Print("ctx TenantID:", reqCtx.TenantID)
+			// id 自动从路径绑定
+			// 注意：webx版本没有自动注入dbFactory
+			return getUserOracle(id, dbFactory) // 需要修改getUser函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
+
+	// POST示例：Body参数绑定
+	ws.POST("/app/users",
+		func(req CreateUserRequest, reqCtx *ctx.RequestContext) (UserResponse, error) {
+			log.Print("ctx TenantID:", reqCtx.TenantID)
+			// req 自动从JSON Body绑定
+			return createUser(req, nil) // 需要修改createUser函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
+
+	// 您的业务路由
+	ws.POST("/app/query/yaml",
+		func() (interface{}, error) {
+			return queryYaml(nil) // 需要修改queryYaml函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
+
+	ws.POST("/app/init/config/template",
+		func() (interface{}, error) {
+			return initConfigTemplate(nil) // 需要修改initConfigTemplate函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
 
-	// // 查询yaml配置文件
-	// webService.Handle("/api/query/yaml", middleware.JWTAuthMiddleware(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-	// 	myhandle.QueryHandlerMap(w, r, dbFactory, service.QueryYamlConfigure)
-	// })))
+}
 
-	// // 初始化配置模版
-	// webService.Handle("/api/init/config/template", middleware.JWTAuthMiddleware(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-	// 	myhandle.QueryHandlerMap(w, r, dbFactory, service.InitConfigTemplates)
-	// })))
+func registerDefaultRouter(ws *webx.WebService, dbFactory *database.DBFactory) {
+	// GET示例：路径参数绑定
+	ws.GET("/api/users/{id}",
+		func(id string, reqCtx *ctx.RequestContext) (UserResponse, error) {
+
+			log.Print("ctx TenantID:", reqCtx.TenantID)
+			// id 自动从路径绑定
+			// 注意：webx版本没有自动注入dbFactory
+			return getUser(id, dbFactory) // 需要修改getUser函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
+
+	// POST示例：Body参数绑定
+	ws.POST("/api/users",
+		func(req CreateUserRequest, reqCtx *ctx.RequestContext) (UserResponse, error) {
+			log.Print("ctx TenantID:", reqCtx.TenantID)
+			// req 自动从JSON Body绑定
+			return createUser(req, nil) // 需要修改createUser函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
+
+	// 您的业务路由
+	ws.POST("/api/query/yaml",
+		func() (interface{}, error) {
+			return queryYaml(nil) // 需要修改queryYaml函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
+
+	ws.POST("/api/init/config/template",
+		func() (interface{}, error) {
+			return initConfigTemplate(nil) // 需要修改initConfigTemplate函数以获取dbFactory
+		},
+	).Use(middleware.JWTAuthMiddleware).Register()
 
 }
 
-func creteTabel(factory *database.DBFactory) {
+// 4. 创建路由器
+// 首先需要创建标准库的ServeMux
+//mux := http.NewServeMux()
+//router := webx.NewRouter(boot.GetRouter())
+//router := webx.NewWebService(mux)
+
+// 5. 注册路由（Spring风格！）
 
-	// 获取数据库连接和类型
-	db := factory.GetDB()
-	dbType := factory.GetDBType()
+func getUser(id string, dbFactory *database.DBFactory) (UserResponse, error) {
+	if dbFactory != nil {
+		dbFactory.TestConnection(dbFactory.GetDBType())
+	}
+
+	// 查询数据库...
+	return UserResponse{
+		ID:    id,
+		Name:  "掌扇过去-AAAAA",
+		Email: "zhangsan@example.com",
+	}, nil
+}
 
-	// 创建表同步器
-	syncer, err := sqldef.NewTableSyncer(db, dbType)
-	if err != nil {
-		log.Printf("创建 - 建立器失败: %v", err)
+// 业务函数示例
+func getUserOracle(id string, dbFactory *database.DBFactory) (UserResponse, error) {
+	if dbFactory != nil {
+		dbFactory.TestConnection(dbFactory.GetDBType())
 	}
 
-	// 创建表
-	if err := syncer.CreateTables(); err != nil {
-		log.Printf("建表失败: %v", err)
+	sql, queryParams := getSQLWithPaginationSQL(0, 10)
+
+	jsonData := dbFactory.QueryPositionalToJSON(sql, queryParams, nil)
+	// 查询数据库...
+	return UserResponse{
+		ID:    id,
+		Name:  "掌扇过去",
+		Email: "zhangsan@example.com",
+		Data:  jsonData.Data,
+	}, nil
+}
+
+func createUser(req CreateUserRequest, dbFactory *database.DBFactory) (UserResponse, error) {
+	logger.Debug("CreateUserRequest:%v", req)
+	return UserResponse{
+		ID:    "1",
+		Name:  req.Name,
+		Email: req.Email,
+	}, nil
+}
+
+func queryYaml(dbFactory *database.DBFactory) (interface{}, error) {
+	// 您的业务逻辑
+	return map[string]interface{}{"message": "query yaml success"}, nil
+}
+
+func initConfigTemplate(dbFactory *database.DBFactory) (interface{}, error) {
+	// 您的业务逻辑
+	return map[string]interface{}{"message": "init config success"}, nil
+}
+
+// getSQLWithPagination 生成带分页的SQL语句（参数模式）
+// 返回SQL语句和参数映射
+func getSQLWithPaginationSQL(startRow, endRow int) (string, []interface{}) {
+	sql := `SELECT
+    CLOTHING_ID,
+    CLOTHING_YEAR,
+    CLOTHING_NAME
+
+   FROM (
+    SELECT a.*, ROWNUM as rn
+    FROM (
+        SELECT *
+        FROM X6_STOCK_DEV.A3_CLOTHING 
+     
+        ORDER BY CLOTHING_ID
+    ) a
+    WHERE ROWNUM <= :1
+)
+WHERE rn > :2`
+
+	// 创建参数映射
+	params := []interface{}{
+		endRow,
+		startRow - 1, // WHERE rn > :start_row 所以是startRow-1
 	}
 
-	log.Println("数据库表建立完成！")
+	return sql, params
 }

tables/dim_store.go

@@ -15,7 +15,8 @@ func init() {
 			String("store_group_id", 32).Comment("所属店铺组ID").End().
 			String("region_code", 32).Comment("区域编码").End().
 			String("creator", 32).NotNull().Comment("创建人").End().
-			DateTime("created_at").NotNull().Default("CURRENT_TIMESTAMP").Comment("创建时间").End()
+			DateTime("created_at").NotNull().Default("CURRENT_TIMESTAMP").Comment("创建时间").End().
+			DateTime("updated_at").NotNull().Default("CURRENT_TIMESTAMP").Comment("最后修改时间").End()
 		r.RegisterTable(tb.Build())
 	})
 }
@@ -28,6 +29,7 @@ type DimStore struct {
 	RegionCode   string    `gorm:"column:region_code;type:varchar(32);comment:区域编码"`
 	Creator      string    `gorm:"column:creator;type:varchar(32);not null;comment:创建人"`
 	CreatedAt    time.Time `gorm:"column:created_at;not null;default:CURRENT_TIMESTAMP;comment:创建时间"`
+	UpdatedAt    time.Time `gorm:"column:updated_at;not null;default:CURRENT_TIMESTAMP;comment:最后修改时间"`
 	TenantId     string    `gorm:"column:tenant_id;type:varchar(8);comment:租户ID"`
 }